test

Gradio_UI.py

@@ -23,6 +23,7 @@ from smolagents.agent_types import AgentAudio, AgentImage, AgentText, handle_age
 from smolagents.agents import ActionStep, MultiStepAgent
 from smolagents.memory import MemoryStep
 from smolagents.utils import _is_package_available
+import gradio as gr
 
 
 def pull_messages_from_step(
@@ -258,6 +259,17 @@ class GradioUI:
             "",
         )
 
+    def append_example_message(x: gr.SelectData, history):
+        if x.value["text"] is not None:
+            history.append((x.value["text"], None))
+        if "files" in x.value:
+            if isinstance(x.value["files"], list):
+                for file in x.value["files"]:
+                    history.append((file, None))
+            else:
+                history.append((x.value["files"], None))
+        return history
+
     def launch(self, **kwargs):
         import gradio as gr
         examples = [
@@ -292,7 +304,7 @@ class GradioUI:
                 # Description
                 examples=examples,  # Example inputs
             )
-            chatbot.example_select(self.log_user_message, [chatbot], [chatbot]).then(self.interact_with_agent, [stored_messages, chatbot], [chatbot])
+            chatbot.example_select(self.append_example_message, [chatbot], [chatbot]).then(self.interact_with_agent, [stored_messages, chatbot], [chatbot])
             # If an upload folder is provided, enable the upload feature
             if self.file_upload_folder is not None:
                 upload_file = gr.File(label="Upload a file")