Update app.py

app.py

@@ -990,7 +990,6 @@ class TemporalPredictionSystem:
 # Global System Initialization
 # -------------------------------------------------------------------
 system = TemporalPredictionSystem()
-# Run some initial simulation steps to pre-train
 time.sleep(2)
 system.run_simulation_steps(20)
 print("Future Self Conversation System initialized and pre-trained.")
@@ -1003,7 +1002,8 @@ def chat_fn(user_message, history):
         history = []
     response = system.process_user_message(user_message)
     bot_message = f"{response['text']} ({response['timeOffset']} years in future | Confidence: {int(response['confidence']*100)}%)"
-    history.append((user_message, bot_message))
+    history.append({"role": "user", "content": user_message})
+    history.append({"role": "assistant", "content": bot_message})
     system.run_simulation_steps(5)
     return history
 
@@ -1056,13 +1056,23 @@ with gr.Blocks(title="Future Self Conversation System") as demo:
     with main_interface:
         with gr.Row():
             with gr.Column():
-                chatbot = gr.Chatbot(label="Conversation", elem_id="chatbot")
+                # Use type="messages" to get openai-style dictionaries.
+                chatbot = gr.Chatbot(label="Conversation", elem_id="chatbot", type="messages")
                 txt = gr.Textbox(placeholder="Type your message here...", label="Your Message")
                 send_btn = gr.Button("Send")
             with gr.Column():
                 network_image = gr.Image(label="Neural Network Visualization")
-                # Auto-update network visualization every 5 seconds.
-                interval = gr.Interval(5, fn=update_network_visualization, outputs=network_image)
+                # A hidden refresh button used to update network visualization automatically.
+                refresh_btn = gr.Button("Refresh Network", visible=False, elem_id="refresh_network")
+                refresh_btn.click(fn=update_network_visualization, inputs=[], outputs=network_image)
+                # Inject custom JS to click the hidden refresh button every 5 seconds.
+                gr.HTML("""
+                <script>
+                setInterval(function() {
+                    document.getElementById("refresh_network").click();
+                }, 5000);
+                </script>
+                """)
         send_btn.click(fn=chat_fn, inputs=[txt, chatbot], outputs=chatbot).then(lambda: "", None, txt)
     
     # On page load, run pre-training and then reveal the main interface.