Update app.py

app.py

@@ -1,208 +1,221 @@
 import os
-import time
-import re
 import gradio as gr
-from google import genai
+from google import generativeai as genai
 
-# Load API key
+# --- 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 Hugging Face Space Secrets.")
+    raise ValueError("❌ GEMINI_API_KEY not found. Please set it in your Hugging Face Space Secrets.")
 
-# Initialize Gemini client
 client = genai.Client()
 
-# Avatars
+# --- Constants ---
+MODEL_NAME = "gemini-1.5-flash-latest"
 AVATAR_IMAGES = {
     "Agent A (Optimist)": "https://i.imgur.com/2GMV9y7.png",
     "Agent B (Skeptic)": "https://i.imgur.com/KzX3T21.png",
-    "SYSTEM": "https://i.imgur.com/z3uxLzT.png"
+    "SYSTEM": "https://i.imgur.com/z3uxLzT.png",
 }
 
-# Custom CSS
+# --- UI CSS ---
 cyberpunk_css = """
 @import url('https://fonts.googleapis.com/css2?family=Orbitron:wght@400;700&display=swap');
 :root {
---neon-pink: #ff00ff; --neon-cyan: #00ffff; --neon-green: #39ff14;
---bg-color-1: #1a001a; --bg-color-2: #0d001a;
---border-glow: 0 0 5px var(--neon-cyan), 0 0 10px var(--neon-cyan), 0 0 20px var(--neon-pink), 0 0 30px var(--neon-pink);
+    --neon-pink: #ff00ff; --neon-cyan: #00ffff; --neon-green: #39ff14;
+    --bg-color-1: #1a001a; --bg-color-2: #0d001a; --bg-color-3: #2a003a;
+    --border-glow: 0 0 5px var(--neon-cyan), 0 0 10px var(--neon-cyan), 0 0 20px var(--neon-pink), 0 0 30px var(--neon-pink);
+    --text-shadow: 0 0 3px var(--neon-cyan), 0 0 5px var(--neon-cyan);
 }
 .gradio-container { background: linear-gradient(45deg, var(--bg-color-1), var(--bg-color-2)); font-family: 'Orbitron', sans-serif; }
-h1, h2, .gr-label { color: var(--neon-cyan); text-shadow: 0 0 5px var(--neon-cyan), 0 0 10px var(--neon-cyan); text-align: center; }
-.gr-button { background: transparent; color: var(--neon-green); border: 2px solid var(--neon-green); border-radius: 5px; box-shadow: 0 0 5px var(--neon-green), 0 0 10px var(--neon-green) inset; transition: all 0.3s ease; }
+h1, .gr-label { color: var(--neon-cyan); text-shadow: var(--text-shadow); text-align: center; }
+.gr-button { background: transparent; color: var(--neon-green); border: 2px solid var(--neon-green) !important; border-radius: 5px; box-shadow: 0 0 5px var(--neon-green), 0 0 10px var(--neon-green) inset; transition: all 0.3s ease; font-family: 'Orbitron', sans-serif; }
 .gr-button:hover { background: var(--neon-green); color: var(--bg-color-2); box-shadow: 0 0 10px var(--neon-green), 0 0 20px var(--neon-green), 0 0 30px var(--neon-green); }
+.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); }
-.gr-dropdown, .gr-radio { border: 1px solid var(--neon-cyan) !important; background-color: var(--bg-color-2) !important; color: var(--neon-cyan) !important; }
+.message-bubble { background: none !important; border: none !important; } /* Clean up message bubbles */
 footer { display: none !important; }
 """
 
-# Utils
-def parse_list(text_response):
-    lines = text_response.strip().split('\n')
-    return [re.sub(r'^\d+[.)]\s*', '', line).strip() for line in lines if line.strip()]
+# --- Core Logic Functions ---
 
-# Topic Generator
-def generate_topics():
-    prompt = "Generate a list of exactly 20 specific, debatable topics related to the future of coding, banking, and artificial intelligence. Format as a numbered list."
+def stream_content_to_chat(model, prompt, history_list, role_name, avatar_url):
+    """Helper to stream a single LLM response into the chat history."""
+    history_list.append({"role": "assistant", "name": role_name, "content": "", "avatar_url": avatar_url})
+    full_response = ""
     try:
-        chat = client.chats.create(model="gemini-2.5-flash")
-        response = chat.send_message(prompt)
-        topics = parse_list(response.text)
-        return {
-            topics_state: topics,
-            topic_dropdown: gr.Dropdown(choices=topics, value=None, label="Step 2: Select a Topic", visible=True),
-            generate_topics_btn: gr.Button(visible=False),
-        }
+        stream = model.generate_content(prompt, stream=True)
+        for chunk in stream:
+            full_response += chunk.text
+            history_list[-1]["content"] = full_response + "▌"
+            yield history_list
+        history_list[-1]["content"] = full_response.strip()
+        yield history_list
     except Exception as e:
-        gr.Warning(f"Error generating topics: {e}")
-        return {
-            topics_state: [],
-            topic_dropdown: gr.Dropdown(visible=False),
-            generate_topics_btn: gr.Button(visible=True)
-        }
-
-# Question Generator
-def generate_questions(selected_topic):
-    if not selected_topic:
-        return {question_dropdown: gr.Dropdown(visible=False)}
-    prompt = f"For the debate topic '{selected_topic}', generate a list of exactly 20 provocative questions to start the debate. Format as a numbered list."
+        error_msg = f"⚠️ An error occurred: {e}"
+        history_list[-1]["content"] = error_msg
+        yield history_list
+    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.
+    """
+    # 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
+    }
+    
+    # --- STEP 1: GENERATE TOPIC ---
     try:
-        chat = client.chats.create(model="gemini-2.5-flash")
-        response = chat.send_message(prompt)
-        questions = parse_list(response.text)
-        return {
-            questions_state: questions,
-            question_dropdown: gr.Dropdown(choices=questions, value=None, label="Step 3: Select a Starting Question", visible=True)
-        }
+        gen_model = genai.GenerativeModel(MODEL_NAME)
+        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()
+            yield { chatbot: updated_history }
+        
+        if "⚠️" in topic: raise Exception("Topic generation failed.")
+
     except Exception as e:
-        gr.Warning(f"Error generating questions: {e}")
-        return {
-            questions_state: [],
-            question_dropdown: gr.Dropdown(visible=False)
-        }
-
-# Debate Enabler
-def enable_debate_button(selected_question):
-    return {start_debate_btn: gr.Button(visible=bool(selected_question))}
-
-# Debate Runner
-def run_debate(topic, question):
-    if not topic or not question:
-        gr.Warning("Select a topic and question first.")
-        return [], []
-
-    agent_a = client.chats.create(model="gemini-2.5-flash")
-    agent_b = client.chats.create(model="gemini-2.5-flash")
-
-    agent_a.send_message("You are Agent A, a tech optimist and futurist. Defend the topic positively.")
-    agent_b.send_message("You are Agent B, a skeptic and ethicist. Argue against the topic cautiously.")
-
-    history = [{
-        "role": "user", "name": "SYSTEM",
-        "content": f"**DEBATE TOPIC:** {topic}\n\n**STARTING QUESTION:** {question}",
-        "avatar_url": AVATAR_IMAGES["SYSTEM"]
-    }]
-    yield history, history
-
-    last_message = question
-
-    for i in range(5):  # 5 rounds = 10 exchanges
-        try:
-            time.sleep(0.7)
-            res_a = agent_a.send_message(f"Based on: '{last_message}', make an optimistic argument.")
-            history.append({
-                "role": "assistant", "name": "Agent A (Optimist)",
-                "content": res_a.text,
-                "avatar_url": AVATAR_IMAGES["Agent A (Optimist)"]
-            })
-            last_message = res_a.text
-            yield history, history
-
-            time.sleep(0.7)
-            res_b = agent_b.send_message(f"Respond to Agent A: '{last_message}' with a skeptical argument.")
-            history.append({
-                "role": "assistant", "name": "Agent B (Skeptic)",
-                "content": res_b.text,
-                "avatar_url": AVATAR_IMAGES["Agent B (Skeptic)"]
-            })
-            last_message = res_b.text
-            yield history, history
+        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:
+        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()
+            yield { chatbot: updated_history }
+
+        if "⚠️" in question: raise Exception("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") }
+        return
+        
+    # --- STEP 3: RUN THE DEBATE ---
+    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."]}
+    ])
+
+    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:
-            history.append({
-                "role": "user", "name": "SYSTEM",
-                "content": f"⚠️ An error occurred: {e}",
-                "avatar_url": AVATAR_IMAGES["SYSTEM"]
-            })
-            yield history, history
-            break
-
-    history.append({
-        "role": "user", "name": "SYSTEM",
-        "content": "--- DEBATE CONCLUDED ---",
-        "avatar_url": AVATAR_IMAGES["SYSTEM"]
-    })
-    yield history, history, gr.Row(visible=True)
-
-# Reset
-def reset_all():
-    return {
-        chatbox: [], debate_log_state: [], topics_state: [], questions_state: [],
-        generate_topics_btn: gr.Button(visible=True),
-        topic_dropdown: gr.Dropdown(value=None, choices=[], visible=False),
-        question_dropdown: gr.Dropdown(value=None, choices=[], visible=False),
-        start_debate_btn: gr.Button(visible=False),
-        save_row: gr.Row(visible=False),
-        download_file: None,
+            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
+            
+        # 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"]})
+    yield {
+        chatbot: chat_history,
+        start_button: gr.update(interactive=True, value="▷ GENERATE & START NEW DEBATE"),
+        download_button: gr.update(visible=True)
     }
 
-# UI
-with gr.Blocks(css=cyberpunk_css, title="AI DEBATE SIMULATOR") as demo:
-    debate_log_state = gr.State([])
-    topics_state = gr.State([])
-    questions_state = gr.State([])
-
-    gr.Markdown("# AI DEBATE SIMULATOR")
-    gr.Markdown("## A War of Wits Between Two AI Agents")
+def create_transcript(chat_history):
+    """Formats the 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
+
+# --- 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.
+    chat_history_state = gr.State([])
+
+    gr.Markdown("# 👾 AI DEBATE SIMULATOR v4.0 👾")
+    gr.Markdown("### A Fully Automated War of Wits, Streamed in Real-Time")
+
+    chatbot = gr.Chatbot(
+        label="Debate Arena",
+        height=700,
+        avatar_images=list(AVATAR_IMAGES.values()),
+        show_copy_button=True
+    )
 
     with gr.Row():
-        with gr.Column(scale=1):
-            generate_topics_btn = gr.Button("Step 1: Generate 20 Debate Topics", variant="primary")
-            topic_dropdown = gr.Dropdown(label="Step 2: Select a Topic", visible=False)
-            question_dropdown = gr.Dropdown(label="Step 3: Select a Starting Question", visible=False)
-            start_debate_btn = gr.Button("Step 4: START THE DEBATE", visible=False)
-            with gr.Row(visible=False) as save_row:
-                save_row_button = gr.Button("Download Log")
-            download_file = gr.File(label="Download File")
-            reset_btn = gr.Button("Start New Debate", variant="stop")
-        with gr.Column(scale=3):
-            chatbox = gr.Chatbot(label="Debate Arena", height=700, type="messages")
-
-    # Button logic
-    generate_topics_btn.click(fn=generate_topics, inputs=None,
-                              outputs=[topics_state, topic_dropdown, generate_topics_btn])
-    topic_dropdown.change(fn=generate_questions, inputs=topic_dropdown,
-                          outputs=[questions_state, question_dropdown])
-    question_dropdown.change(fn=enable_debate_button, inputs=question_dropdown,
-                             outputs=[start_debate_btn])
-    start_debate_btn.click(
-        fn=lambda: {
-            generate_topics_btn: gr.Button(visible=False),
-            topic_dropdown: gr.Dropdown(interactive=False),
-            question_dropdown: gr.Dropdown(interactive=False),
-            start_debate_btn: gr.Button(visible=False),
-        },
-        inputs=None,
-        outputs=[generate_topics_btn, topic_dropdown, question_dropdown, start_debate_btn]
-    ).then(fn=run_debate, inputs=[topic_dropdown, question_dropdown],
-           outputs=[chatbox, debate_log_state, save_row])
-
-    reset_btn.click(fn=reset_all, inputs=None,
-                    outputs=[chatbox, debate_log_state, topics_state,
-                             questions_state, generate_topics_btn,
-                             topic_dropdown, question_dropdown,
-                             start_debate_btn, save_row, download_file])
-
-# Launch app
+        start_button = gr.Button("▷ GENERATE & 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]
+    )
+
+    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]
+    )
+
+
+# --- Launch the App ---
 if __name__ == "__main__":
-    demo.launch(debug=True, ssr_mode=False)
+    demo.queue().launch(debug=True)