Update app.py

app.py

@@ -7,6 +7,40 @@ import json
 import os
 import shutil
 import requests
+import gradio as gr
+from gradio.components import Button, ClearButton
+from gradio.layouts import Column, Row
+from gradio.data_classes import InterfaceTypes
+import inspect
+
+# CustomInterface class definition
+class CustomInterface(gr.Interface):
+    def render_input_column(self):
+        submit_btn, clear_btn, stop_btn, flag_btns = None, None, None, None
+
+        with Column(variant="panel"):
+            input_component_column = Column()
+            with input_component_column:
+                for component in self.input_components:
+                    component.render()
+            with Row():
+                if self.interface_type in [InterfaceTypes.STANDARD, InterfaceTypes.INPUT_ONLY]:
+                    clear_btn = ClearButton(value = "XXXXXX")
+                    if not self.live:
+                        submit_btn = Button("Ciaoooooooo", variant="primary")  # Custom label
+                        if inspect.isgeneratorfunction(self.fn) or inspect.isasyncgenfunction(self.fn):
+                            stop_btn = Button("Stop", variant="stop", visible=False)
+                elif self.interface_type == InterfaceTypes.UNIFIED:
+                    clear_btn = ClearButton(value = "XXXXXX")
+                    submit_btn = Button("Ciaoooooooo", variant="primary")  # Custom label
+                    if (inspect.isgeneratorfunction(self.fn) or inspect.isasyncgenfunction(self.fn)) and not self.live:
+                        stop_btn = Button("Stop", variant="stop")
+                    if self.allow_flagging == "manual":
+                        flag_btns = self.render_flag_btns()
+                    elif self.allow_flagging == "auto":
+                        flag_btns = [submit_btn]
+        return submit_btn, clear_btn, stop_btn, flag_btns, input_component_column
+
 
 # Define the device
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -138,7 +172,7 @@ additional_inputs=[
     ),
 ]
 
-iface = gr.Interface(
+iface = CustomInterface(
     fn=mistral_bot.predict,
     title=title,
     description=description,
@@ -148,12 +182,7 @@ iface = gr.Interface(
     ] + additional_inputs,
     outputs=[
         gr.Textbox(label="Output will happear here", type="text", lines=5)],
-    clear_buttom = 'Clear click here',
-    submit_button = 'Clear click here',
     theme="carlosrosash/bellay_test",
-    css= """
-    p {align : center}
-    """
 )
 
 # Launch the Gradio interface for the Mistral model