Update Gradio_UI.py

Gradio_UI.py

@@ -176,18 +176,25 @@ def stream_to_gradio(
 
 
 class GradioUI:
-    """A one-line interface to launch your agent in Gradio"""
-
-    def __init__(self, agent: MultiStepAgent, file_upload_folder: str | None = None):
-        if not _is_package_available("gradio"):
-            raise ModuleNotFoundError(
-                "Please install 'gradio' extra to use the GradioUI: `pip install 'smolagents[gradio]'`"
-            )
+    def __init__(self, agent):
         self.agent = agent
-        self.file_upload_folder = file_upload_folder
-        if self.file_upload_folder is not None:
-            if not os.path.exists(file_upload_folder):
-                os.mkdir(file_upload_folder)
+
+    def launch(self, share=False, **kwargs):
+        import gradio as gr
+
+        # Your interface setup here
+        with gr.Blocks() as demo:
+            gr.Markdown("### Welcome to the AI Agent UI")
+            inp = gr.Textbox(label="Input")
+            out = gr.Textbox(label="Output")
+
+            def respond(message):
+                return self.agent.run(message)
+
+            btn = gr.Button("Submit")
+            btn.click(fn=respond, inputs=inp, outputs=out)
+
+        demo.launch(debug=True, share=share, **kwargs)
 
     def interact_with_agent(self, prompt, messages):
         import gradio as gr