Create a new agent per session to isolate session conversation with others session

Gradio_UI.py

@@ -192,12 +192,17 @@ class GradioUI:
             if not os.path.exists(file_upload_folder):
                 os.mkdir(file_upload_folder)
     
-    def interact_with_agent(self, prompt, messages):
+    def interact_with_agent(self, prompt, messages, session_state):
         import gradio as gr
 
+        # Get the agent type from the template agent
+        if "agent" not in session_state:
+            from copy import deepcopy
+            session_state["agent"] = deepcopy(self.agent)
+
         messages.append(gr.ChatMessage(role="user", content=prompt))
         yield messages
-        for msg in stream_to_gradio(self.agent, task=prompt, reset_agent_memory=False):
+        for msg in stream_to_gradio(session_state["agent"], task=prompt, reset_agent_memory=False):
             messages.append(msg)
             yield messages
         yield messages
@@ -267,6 +272,7 @@ class GradioUI:
         with gr.Blocks(fill_height=True) as demo:
             # user_state = gr.State(random.random)  # to avoid sharing gr.State across users (that uses all the users conversations for any user prompts)
             gr.Label("Welcome to my first smolagent! You can use tools to get the time in a location, visit a URL page, search with DuckDuckGo.")
+            session_state = gr.State({})
             stored_messages = gr.State([])
             file_uploads_log = gr.State([])
             chatbot = gr.Chatbot(
@@ -293,7 +299,7 @@ class GradioUI:
                 self.log_user_message,
                 [text_input, file_uploads_log],
                 [stored_messages, text_input],
-            ).then(self.interact_with_agent, [stored_messages, chatbot], [chatbot])
+            ).then(self.interact_with_agent, [stored_messages, chatbot, session_state], [chatbot])
             # Initialize instance when page loads
             demo.load(random.random, inputs=None, outputs=None)