Update GRADIO_UI.py

GRADIO_UI.py

@@ -175,15 +175,16 @@ class GradioUI:
         if not _is_package_available("gradio"):
             raise ModuleNotFoundError("Please install 'gradio' extra to use the GradioUI: `pip install 'smolagents[gradio]'`")
         self.initialize_agent = initialize_agent
+        self.messages = []  # Initialize messages as a class attribute
 
-    def interact_with_agent(self, prompt, api_key, messages):
+    def interact_with_agent(self, prompt, api_key):
         import gradio as gr
-        messages.append(gr.ChatMessage(role="user", content=prompt))
-        yield messages
+        self.messages.append(gr.ChatMessage(role="user", content=prompt))
+        yield self.messages
         for msg in stream_to_gradio(self.initialize_agent, task=prompt, api_key=api_key, reset_agent_memory=False):
-            messages.append(msg)
-            yield messages
-        yield messages
+            self.messages.append(msg)
+            yield self.messages
+        yield self.messages
 
     def launch(self, **kwargs):
         import gradio as gr
@@ -217,8 +218,7 @@ class GradioUI:
             text_input = gr.Textbox(
                 lines=1, label="Enter URL and request (e.g., navigate to https://en.wikipedia.org/wiki/Nvidia, and provide me info on its history)"
             )
-            messages = []  
-            text_input.submit(self.interact_with_agent, [text_input, api_key_input, messages], [chatbot])
+            text_input.submit(self.interact_with_agent, [text_input, api_key_input], [chatbot])
 
         demo.launch(debug=True, **kwargs)