Update app.py

app.py

@@ -1,9 +1,9 @@
 import os
 import gradio as gr
 from google import genai
+import time
 
 # --- Load Gemini API key securely ---
-# Make sure to set the GEMINI_API_KEY in your Hugging Face Space Secrets
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
 if not GEMINI_API_KEY:
     raise ValueError("❌ GEMINI_API_KEY not found. Please set it in your Hugging Face Space Secrets.")
@@ -34,13 +34,11 @@ h1, .gr-label { color: var(--neon-cyan); text-shadow: var(--text-shadow); text-a
 .gr-button:disabled { background: #333 !important; color: #555 !important; border-color: #555 !important; box-shadow: none !important; cursor: not-allowed; }
 .chatbot { background-color: rgba(13, 0, 26, 0.8); border: 1px solid var(--neon-pink); box-shadow: var(--border-glow); }
 .message-bubble { background: none !important; border: none !important; } /* Clean up message bubbles */
-footer { display: none !important; }
+footer { display: none !import
 """
 
-# --- Core Logic Functions ---
-
-def stream_content_to_chat(model, prompt, history_list, role_name, avatar_url):
-    """Helper to stream a single LLM response into the chat history."""
+def stream_llm_response(model, prompt, history_list, role_name, avatar_url):
+    """A dedicated generator to stream a response into the history and yield updates."""
     history_list.append({"role": "assistant", "name": role_name, "content": "", "avatar_url": avatar_url})
     full_response = ""
     try:
@@ -50,143 +48,107 @@ def stream_content_to_chat(model, prompt, history_list, role_name, avatar_url):
             history_list[-1]["content"] = full_response + "▌"
             yield history_list
         history_list[-1]["content"] = full_response.strip()
-        yield history_list
     except Exception as e:
-        error_msg = f"⚠️ An error occurred: {e}"
-        history_list[-1]["content"] = error_msg
-        yield history_list
+        history_list[-1]["content"] = f"⚠️ An error occurred: {e}"
+    
+    yield history_list # Yield the final, clean response
     return full_response.strip()
 
 def run_full_debate_flow(chat_history):
-    """
-    This single, powerful function handles the entire user flow:
-    1. Clears the chat.
-    2. Disables the button.
-    3. Streams the generation of a topic.
-    4. Streams the generation of a question.
-    5. Streams the full debate between two agents.
-    6. Re-enables the button at the end.
-    """
+    """Handles the entire automated debate, narrating each step to the screen."""
+    
+    def add_system_message(text):
+        """Helper to append a system message and yield the UI update."""
+        chat_history.append({"role": "user", "name": "SYSTEM", "content": text, "avatar_url": AVATAR_IMAGES["SYSTEM"]})
+        time.sleep(0.5) # A brief, dramatic pause
+        return {chatbot: chat_history}
+    
     # 1. Clear chat and disable button for a fresh start
     chat_history.clear()
     yield {
         start_button: gr.update(interactive=False, value="🔄 Initializing..."),
-        chatbot: chat_history
+        chatbot: chat_history,
+        download_button: gr.update(visible=False), # Hide download button
+        download_file_output: gr.update(value=None, visible=False)
     }
+
+    yield add_system_message("Receiving transmission... Booting up debate protocol.")
     
-    # --- STEP 1: GENERATE TOPIC ---
     try:
         gen_model = genai.GenerativeModel(MODEL_NAME)
+        
+        # --- NARRATE & GENERATE TOPIC ---
+        yield add_system_message("Contacting orbital AI to generate a fresh topic...")
         topic_prompt = "Generate a single, specific, and highly debatable topic about the future of technology, AI, or finance. Output only the topic string, nothing else."
         topic = ""
-        for updated_history in stream_content_to_chat(gen_model, topic_prompt, chat_history, "SYSTEM", AVATAR_IMAGES["SYSTEM"]):
-            topic = updated_history[-1]['content'].replace('▌', '').strip()
+        for updated_history in stream_llm_response(gen_model, topic_prompt, chat_history, "SYSTEM", AVATAR_IMAGES["SYSTEM"]):
+            topic = updated_history[-1]['content'].replace('▌', '')
             yield { chatbot: updated_history }
-        
-        if "⚠️" in topic: raise Exception("Topic generation failed.")
+        if "⚠️" in topic or not topic: raise ValueError("Topic generation failed.")
 
-    except Exception as e:
-        gr.Error(f"Could not generate a debate topic. Please check your API key and network. Details: {e}")
-        yield { start_button: gr.update(interactive=True, value="▷ GENERATE & START NEW DEBATE") }
-        return
-
-    # --- STEP 2: GENERATE QUESTION ---
-    try:
+        # --- NARRATE & GENERATE QUESTION ---
+        yield add_system_message("Formulating a provocative starting question...")
         question_prompt = f"For the debate topic '{topic}', generate one provocative, open-ended starting question. Output only the question string, nothing else."
         question = ""
-        for updated_history in stream_content_to_chat(gen_model, question_prompt, chat_history, "SYSTEM", AVATAR_IMAGES["SYSTEM"]):
-            question = updated_history[-1]['content'].replace('▌', '').strip()
+        for updated_history in stream_llm_response(gen_model, question_prompt, chat_history, "SYSTEM", AVATAR_IMAGES["SYSTEM"]):
+            question = updated_history[-1]['content'].replace('▌', '')
             yield { chatbot: updated_history }
-
-        if "⚠️" in question: raise Exception("Question generation failed.")
+        if "⚠️" in question or not question: raise ValueError("Question generation failed.")
 
     except Exception as e:
-        gr.Error(f"Could not generate a starting question. Please check your API key and network. Details: {e}")
-        yield { start_button: gr.update(interactive=True, value="▷ GENERATE & START NEW DEBATE") }
+        yield add_system_message(f"❌ FATAL ERROR during setup: {e}. Protocol terminated. Please try again.")
+        yield { start_button: gr.update(interactive=True, value="▷ RE-INITIATE DEBATE PROTOCOL") }
         return
         
-    # --- STEP 3: RUN THE DEBATE ---
+    # --- NARRATE DEBATE START & RUN THE DEBATE ---
+    yield add_system_message("Initializing agents... Let the debate begin!")
     yield { start_button: gr.update(value="⚔️ DEBATE IN PROGRESS...") }
     
-    # Initialize agents with their personalities
-    agent_a_chat = gen_model.start_chat(history=[
-        {'role': 'user', 'parts': ["You are Agent A, a tech optimist and futurist. Argue passionately for the positive aspects."]},
-        {'role': 'model', 'parts': ["I am Agent A, ready to envision a brighter future."]}
-    ])
-    agent_b_chat = gen_model.start_chat(history=[
-        {'role': 'user', 'parts': ["You are Agent B, a cautious skeptic and ethicist. Argue the cautionary side, raising ethical concerns."]},
-        {'role': 'model', 'parts': ["I am Agent B, here to ensure we consider the risks."]}
-    ])
+    agent_a_chat = gen_model.start_chat(history=[{'role': 'user', 'parts': ["You are Agent A, a tech optimist. Argue positively."]}, {'role': 'model', 'parts': ["Understood."]}])
+    agent_b_chat = gen_model.start_chat(history=[{'role': 'user', 'parts': ["You are Agent B, a skeptic. Argue cautiously."]}, {'role': 'model', 'parts': ["Acknowledged."]}])
 
     last_response = question
-    num_turns = 5
-
-    for i in range(num_turns):
-        # Agent A's Turn (Optimist)
-        chat_history.append({"role": "assistant", "name": "Agent A (Optimist)", "content": "", "avatar_url": AVATAR_IMAGES["Agent A (Optimist)"]})
-        try:
-            stream_a = agent_a_chat.send_message(f"Debate Topic: '{topic}'. The previous statement was: '{last_response}'. Provide your optimistic argument.", stream=True)
-            full_response_a = ""
-            for chunk in stream_a:
-                full_response_a += chunk.text
-                chat_history[-1]["content"] = full_response_a + "▌"
-                yield { chatbot: chat_history }
-            chat_history[-1]["content"] = full_response_a
-            last_response = full_response_a
-            yield { chatbot: chat_history }
-        except Exception as e:
-            chat_history[-1]['content'] = f"⚠️ Agent A Error: {e}"
-            yield { chatbot: chat_history, start_button: gr.update(interactive=True, value="▷ GENERATE & START NEW DEBATE") }
-            return
+    for _ in range(5):
+        # Agent A
+        prompt_a = f"Debate Topic: '{topic}'. The previous statement was: '{last_response}'. Provide your optimistic argument."
+        for updated_history in stream_llm_response(agent_a_chat, prompt_a, chat_history, "Agent A (Optimist)", AVATAR_IMAGES["Agent A (Optimist)"]):
+             last_response = updated_history[-1]['content'].replace('▌', '')
+             yield { chatbot: updated_history }
+        
+        # Agent B
+        prompt_b = f"Debate Topic: '{topic}'. Agent A said: '{last_response}'. Provide your skeptical counter-argument."
+        for updated_history in stream_llm_response(agent_b_chat, prompt_b, chat_history, "Agent B (Skeptic)", AVATAR_IMAGES["Agent B (Skeptic)"]):
+            last_response = updated_history[-1]['content'].replace('▌', '')
+            yield { chatbot: updated_history }
             
-        # Agent B's Turn (Skeptic)
-        chat_history.append({"role": "assistant", "name": "Agent B (Skeptic)", "content": "", "avatar_url": AVATAR_IMAGES["Agent B (Skeptic)"]})
-        try:
-            stream_b = agent_b_chat.send_message(f"Debate Topic: '{topic}'. Agent A said: '{last_response}'. Provide your skeptical counter-argument.", stream=True)
-            full_response_b = ""
-            for chunk in stream_b:
-                full_response_b += chunk.text
-                chat_history[-1]["content"] = full_response_b + "▌"
-                yield { chatbot: chat_history }
-            chat_history[-1]["content"] = full_response_b
-            last_response = full_response_b
-            yield { chatbot: chat_history }
-        except Exception as e:
-            chat_history[-1]['content'] = f"⚠️ Agent B Error: {e}"
-            yield { chatbot: chat_history, start_button: gr.update(interactive=True, value="▷ GENERATE & START NEW DEBATE") }
-            return
-
-    # --- DEBATE CONCLUDED ---
-    chat_history.append({"role": "user", "name": "SYSTEM", "content": "--- DEBATE CONCLUDED ---", "avatar_url": AVATAR_IMAGES["SYSTEM"]})
+    # --- CONCLUDE DEBATE ---
+    yield add_system_message("--- DEBATE CONCLUDED ---")
     yield {
         chatbot: chat_history,
-        start_button: gr.update(interactive=True, value="▷ GENERATE & START NEW DEBATE"),
+        start_button: gr.update(interactive=True, value="▷ START NEW DEBATE"),
         download_button: gr.update(visible=True)
     }
 
 def create_transcript(chat_history):
-    """Formats the chat history into a downloadable text file."""
-    if not chat_history:
-        return None
+    """Formats chat history into a downloadable text file."""
+    if not chat_history: return None
     transcript = "AI DEBATE SIMULATOR TRANSCRIPT\n=================================\n\n"
     for message in chat_history:
         content = message.get("content", "").replace("▌", "")
         transcript += f"[{message.get('name', 'SYSTEM')}]\n{content}\n\n---------------------------------\n\n"
-    
     filepath = "debate_transcript.txt"
-    with open(filepath, "w", encoding="utf-8") as f:
-        f.write(transcript)
-    return filepath
+    with open(filepath, "w", encoding="utf-8") as f: f.write(transcript)
+    return gr.update(value=filepath, visible=True)
 
 # --- Gradio UI Layout ---
-with gr.Blocks(css=cyberpunk_css, theme=gr.themes.Base(), title="AI Debate Simulator v4.0") as demo:
-    # State to hold the chat history list. This is the single source of truth.
+with gr.Blocks(css=cyberpunk_css, theme=gr.themes.Base(), title="AI Debate Simulator v4.2") as demo:
     chat_history_state = gr.State([])
 
-    gr.Markdown("# 👾 AI DEBATE SIMULATOR v4.0 👾")
-    gr.Markdown("### A Fully Automated War of Wits, Streamed in Real-Time")
+    gr.Markdown("# 👾 AI DEBATE SIMULATOR v4.2 👾")
+    gr.Markdown("### An Automated War of Wits, Narrated in Real-Time")
 
     chatbot = gr.Chatbot(
+        type="messages",
         label="Debate Arena",
         height=700,
         avatar_images=list(AVATAR_IMAGES.values()),
@@ -194,28 +156,22 @@ with gr.Blocks(css=cyberpunk_css, theme=gr.themes.Base(), title="AI Debate Simul
     )
 
     with gr.Row():
-        start_button = gr.Button("▷ GENERATE & START NEW DEBATE", variant="primary", scale=3)
+        start_button = gr.Button("▷ START NEW DEBATE", variant="primary", scale=3)
         download_button = gr.Button("💾 Download Transcript", visible=False, scale=1)
         
     download_file_output = gr.File(label="Download", visible=False)
 
-    # --- Event Listeners ---
     start_button.click(
         fn=run_full_debate_flow,
         inputs=[chat_history_state],
-        outputs=[chatbot, start_button, download_button]
+        outputs=[chatbot, start_button, download_button, download_file_output]
     )
 
     download_button.click(
         fn=create_transcript,
         inputs=[chat_history_state],
-        outputs=[download_file_output]
-    ).then(
-        fn=lambda: gr.update(visible=True), # Show the file component
-        outputs=[download_file_output]
+        outputs=download_file_output
     )
 
-
-# --- Launch the App ---
 if __name__ == "__main__":
     demo.queue().launch(debug=True)