Update app.py

app.py

@@ -194,11 +194,21 @@ def retrieve_relevant_memory(query, k=3):
 # - safe_chat_completion (already replaced above)
 # - get_embedding (already replaced above)
 
+# ... [Keep all imports, config, and function definitions above] ...
+
 # === GRADIO UI ===
 with gr.Blocks(theme=gr.themes.Soft(), title="DeepSeek Discussion Platform") as demo:
     gr.Markdown("# 🧠 Hexa-Agent Discussion System (Free Version)")
     gr.Markdown("### Powered by Open-Source Models")
     
+    # State variables
+    conversation_state = gr.State([])
+    turn_count_state = gr.State(0)
+    current_topic_state = gr.State("")
+    last_ruling_turn_state = gr.State(0)
+    auto_mode_state = gr.State(False)
+    agent_params_state = gr.State(agent_params)
+    
     # Status panel
     with gr.Row():
         turn_counter = gr.Number(label="Turn Count", value=0, interactive=False)
@@ -228,83 +238,8 @@ with gr.Blocks(theme=gr.themes.Soft(), title="DeepSeek Discussion Platform") as
                 overseer_out = gr.Textbox(label="Depth Guardian Response", interactive=False)
                 judge_out = gr.Textbox(label="Judge's Response", interactive=False)
     
-    with gr.Tab("Agent Perspectives"):
-        with gr.Row():
-            with gr.Column(scale=1):
-                gr.Markdown("### ⚖️ Depth Guardian")
-                intervention_display = gr.Textbox(label="", interactive=False)
-            with gr.Column(scale=1):
-                gr.Markdown("### 🌐 Cross-Disciplinary")
-                outsider_display = gr.Textbox(label="", interactive=False)
-            with gr.Column(scale=1):
-                gr.Markdown("### 🌍 Cultural Lens")
-                cultural_display = gr.Textbox(label="", interactive=False)
-        
-        with gr.Row():
-            with gr.Column(scale=3):
-                gr.Markdown("### ⚖️ Final Judgment")
-                judge_display = gr.Textbox(label="", interactive=False, lines=4)
-    
-    with gr.Tab("Analysis Dashboard"):
-        gr.Markdown("### Conversation Insights")
-        with gr.Row():
-            sentiment_display = gr.Textbox(label="Discussion Sentiment", interactive=False)
-            topics_display = gr.JSON(label="Key Topics")
-        
-        with gr.Row():
-            agent_plot = gr.Plot(label="Agent Participation")
-            analysis_btn = gr.Button("Generate Insights", variant="primary")
-        
-        with gr.Row():
-            gr.Markdown("### Knowledge Graph")
-            graph_btn = gr.Button("Generate Knowledge Graph", variant="secondary")
-            graph_display = gr.Image(label="Concept Relationships", interactive=False)
-    
-    with gr.Tab("Collaboration"):
-        gr.Markdown("### Real-Time Collaboration")
-        with gr.Row():
-            user_input = gr.Textbox(label="Your Contribution", placeholder="Add your perspective...")
-            submit_btn = gr.Button("Add to Discussion")
-        
-        with gr.Row():
-            voting_btn = gr.Button("👍 Vote for Current Direction")
-            flag_btn = gr.Button("🚩 Flag Issue")
-        
-        with gr.Row():
-            user_feedback = gr.Textbox(label="Community Feedback", interactive=False)
-    
-    with gr.Tab("Export & Integrate"):
-        with gr.Row():
-            format_radio = gr.Radio(
-                ["PDF Report", "JSON Data", "Text Transcript"], 
-                label="Export Format",
-                value="PDF Report"
-            )
-            export_btn = gr.Button("Export Discussion", variant="primary")
-            export_result = gr.File(label="Download File")
-        
-        with gr.Row():
-            gr.Markdown("### API Integration")
-            webhook_url = gr.Textbox(label="Webhook URL", placeholder="https://your-platform.com/webhook")
-            integrate_btn = gr.Button("Connect to External Platform", variant="secondary")
-            integration_status = gr.Textbox(label="Status", interactive=False)
-    
-    with gr.Tab("Agent Configuration"):
-        gr.Markdown("### Customize Agent Behavior")
-        with gr.Row():
-            agent_sliders = {}
-            for agent in ["Initiator", "Responder", "Guardian", "Provocateur", "Cultural", "Judge"]:
-                with gr.Column():
-                    gr.Markdown(f"#### {agent}")
-                    agent_sliders[f"{agent}_creativity"] = gr.Slider(
-                        0.0, 1.0, value=0.7, 
-                        label="Creativity", interactive=True
-                    )
-                    agent_sliders[f"{agent}_critical"] = gr.Slider(
-                        0.0, 1.0, value=0.5, 
-                        label="Criticality", interactive=True
-                    )
-    
+    # ... [Keep all other tabs the same as before] ...
+
     # Custom CSS
     demo.css = """
     .convo-container {
@@ -341,7 +276,111 @@ with gr.Blocks(theme=gr.themes.Soft(), title="DeepSeek Discussion Platform") as
     </div>
     """)
     
-    # ... [Keep all the event handlers from the previous implementation] ...
-    # The event handlers remain unchanged from the last working version
+    # Event handlers with proper state management
+    qbox.submit(
+        overseer_respond, 
+        inputs=[qbox, conversation_state, current_topic_state],
+        outputs=[overseer_out]
+    )
+    
+    ruling_qbox.submit(
+        ask_judge, 
+        inputs=[ruling_qbox, conversation_state, current_topic_state],
+        outputs=[judge_out]
+    )
+    
+    step_btn.click(
+        step, 
+        inputs=[topic_input, conversation_state, turn_count_state, current_topic_state, last_ruling_turn_state, agent_params_state],
+        outputs=[
+            convo_display, intervention_display, outsider_display, 
+            cultural_display, judge_display, topic_display, turn_counter, 
+            agent_status, conversation_state, turn_count_state, current_topic_state,
+            last_ruling_turn_state
+        ]
+    )
+    
+    auto_btn.click(
+        toggle_auto, 
+        inputs=[auto_mode_state],
+        outputs=[auto_btn, auto_mode_state]
+    )
+    
+    clear_btn.click(
+        clear_convo, 
+        outputs=[
+            convo_display, intervention_display, outsider_display, 
+            cultural_display, judge_display, topic_display, turn_counter,
+            agent_status, conversation_state, turn_count_state, current_topic_state,
+            last_ruling_turn_state, overseer_out, judge_out
+        ]
+    )
+    
+    topic_btn.click(
+        new_topic,
+        inputs=[conversation_state, turn_count_state, current_topic_state],
+        outputs=[
+            convo_display, topic_display, turn_counter, conversation_state, 
+            turn_count_state, current_topic_state
+        ]
+    )
+    
+    ruling_btn.click(
+        request_ruling,
+        inputs=[conversation_state, current_topic_state, turn_count_state, last_ruling_turn_state],
+        outputs=[judge_display, conversation_state, last_ruling_turn_state]
+    )
+    
+    analysis_btn.click(
+        run_analysis,
+        inputs=[conversation_state],
+        outputs=[sentiment_display, topics_display, agent_plot]
+    )
+    
+    graph_btn.click(
+        generate_knowledge_graph,
+        inputs=[conversation_state],
+        outputs=[graph_display]
+    )
+    
+    export_btn.click(
+        export_handler,
+        inputs=[format_radio, conversation_state, current_topic_state, turn_count_state],
+        outputs=[export_result]
+    )
+    
+    integrate_btn.click(
+        send_to_webhook,
+        inputs=[webhook_url, conversation_state, current_topic_state, turn_count_state],
+        outputs=[integration_status]
+    )
+    
+    submit_btn.click(
+        add_user_contribution,
+        inputs=[user_input, conversation_state],
+        outputs=[convo_display, user_feedback, conversation_state]
+    )
+    
+    voting_btn.click(
+        lambda: "✅ Your vote has been recorded!",
+        outputs=[user_feedback]
+    )
+    
+    flag_btn.click(
+        lambda: "🚩 Issue flagged for moderator review",
+        outputs=[user_feedback]
+    )
+    
+    # Create input list for slider change events
+    slider_inputs = [agent_sliders[f"{agent}_{param}"] 
+                    for agent in ["Initiator", "Responder", "Guardian", "Provocateur", "Cultural", "Judge"]
+                    for param in ["creativity", "critical"]]
+    
+    for slider in slider_inputs:
+        slider.change(
+            update_agent_params,
+            inputs=slider_inputs + [agent_params_state],
+            outputs=[agent_params_state]
+        )
 
 demo.launch()