Update Gradio_UI.py

Gradio_UI.py

@@ -148,9 +148,7 @@ def stream_to_gradio(
                 step_log.input_token_count = agent.model.last_input_token_count
                 step_log.output_token_count = agent.model.last_output_token_count
 
-        for message in pull_messages_from_step(
-            step_log,
-        ):
+        for message in pull_messages_from_step(step_log):
             yield message
 
     final_answer = step_log  # Last log is the run's final_answer
@@ -227,15 +225,11 @@ class GradioUI:
 
         # Sanitize file name
         original_name = os.path.basename(file.name)
-        sanitized_name = re.sub(
-            r"[^\w\-.]", "_", original_name
-        )  # Replace any non-alphanumeric, non-dash, or non-dot characters with underscores
-
+        sanitized_name = re.sub(r"[^\w\-.]", "_", original_name)
         type_to_ext = {}
         for ext, t in mimetypes.types_map.items():
             if t not in type_to_ext:
                 type_to_ext[t] = ext
-
         # Ensure the extension correlates to the mime type
         sanitized_name = sanitized_name.split(".")[:-1]
         sanitized_name.append("" + type_to_ext[mime_type])
@@ -262,7 +256,8 @@ class GradioUI:
         import gradio as gr
 
         with gr.Blocks(fill_height=True) as demo:
-            stored_messages = gr.State([])
+            # Initialize stored_messages with the agent's conversation (including the initial opener)
+            stored_messages = gr.State(value=self.agent.conversation)
             file_uploads_log = gr.State([])
             chatbot = gr.Chatbot(
                 label="Agent",
@@ -293,4 +288,4 @@ class GradioUI:
         demo.launch(debug=True, share=True, **kwargs)
 
 
-__all__ = ["stream_to_gradio", "GradioUI"]
+__all__ = ["stream_to_gradio", "GradioUI"]