Clone a new agent per session to isolate the session conversation from others

Gradio_UI.py

@@ -178,6 +178,33 @@ def stream_to_gradio(
         yield gr.ChatMessage(role="assistant", content=f"**Final answer:** {str(final_answer)}")
 
 
+class AgentCloner:
+    def __init__(self, agent: CodeAgent) -> None:
+        self.agent = agent
+        self.model_client = agent.model
+
+    def clone():
+        new_model_client = HfApiModel(
+            max_tokens=self.model.max_tokens,
+            temperature=self.model.temperature,
+            model_id=self.model.model_id,
+            custom_role_conversions=self.model.custom_role_conversions,
+        )
+        new_agent = CodeAgent(
+            model=new_model_client,
+            tools=self.agent.tools,
+            max_steps=self.agent.max_steps,
+            verbosity_level=self.agent.verbosity_level,
+            grammar=self.agent.,
+            planning_interval=self.agent.,
+            name=self.agent.,
+            description=self.agent.,
+            prompt_templates=self.agent.prompt_templates
+        )
+        return new_agent
+
+
+
 class GradioUI:
     """A one-line interface to launch your agent in Gradio"""
 
@@ -186,7 +213,7 @@ class GradioUI:
             raise ModuleNotFoundError(
                 "Please install 'gradio' extra to use the GradioUI: `pip install 'smolagents[gradio]'`"
             )
-        self.agent = agent
+        self.agent_cloner = AgentCloner(agent, model)
         self.file_upload_folder = file_upload_folder
         if self.file_upload_folder is not None:
             if not os.path.exists(file_upload_folder):
@@ -197,8 +224,7 @@ class GradioUI:
 
         # Get the agent type from the template agent
         if "agent" not in session_state:
-            from copy import deepcopy
-            session_state["agent"] = deepcopy(self.agent)
+            session_state["agent"] = self.agent_cloner.clone()
 
         messages.append(gr.ChatMessage(role="user", content=prompt))
         yield messages
@@ -300,8 +326,6 @@ class GradioUI:
                 [text_input, file_uploads_log],
                 [stored_messages, text_input],
             ).then(self.interact_with_agent, [stored_messages, chatbot, session_state], [chatbot])
-            # Initialize instance when page loads
-            demo.load(random.random, inputs=None, outputs=None)
 
         demo.launch(debug=True, share=True, **kwargs)