Update app.py

app.py

@@ -1,126 +1,208 @@
 import os
-import gradio as gr
-from google import genai
-import json
 import time
 import re
+import gradio as gr
+from google import genai
 
-# --- Load Gemini API key securely ---
+# Load API key
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
 if not GEMINI_API_KEY:
-    raise ValueError("\u274c GEMINI_API_KEY not found. Please set it in your Hugging Face Space Secrets.")
+    raise ValueError("❌ GEMINI_API_KEY not found. Please set it in Hugging Face Space Secrets.")
 
-# --- Initialize the client ---
+# Initialize Gemini client
 client = genai.Client()
 
-# --- Avatar Images ---
+# Avatars
 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"
 }
 
-# --- UI CSS ---
+# Custom 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);
-}
-.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;
-}
-.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);
-}
-.chatbot {
-  background-color: rgba(13, 0, 26, 0.8);
-  border: 1px solid var(--neon-pink);
-  box-shadow: var(--border-glow);
+--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);
 }
+.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; }
+.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); }
+.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; }
+footer { display: none !important; }
 """
 
-# --- Utils ---
-def parse_list(text):
-    lines = text.strip().split('\n')
+# 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()]
 
-# --- Topic Generator ---
+# Topic Generator
 def generate_topics():
-    prompt = "Generate 20 specific, debatable topics about coding, banking, and AI. Format as a numbered list."
-    chat = client.chats.create(model="gemini-2.5-flash")
-    response = chat.send_message(prompt)
-    topics = parse_list(response.text)
-    return topics
-
-# --- Question Generator ---
-def generate_questions(topic):
-    prompt = f"For the debate topic '{topic}', generate 20 controversial starting questions. Format as a numbered list."
-    chat = client.chats.create(model="gemini-2.5-flash")
-    response = chat.send_message(prompt)
-    return parse_list(response.text)
-
-# --- Debate Logic ---
+    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."
+    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),
+        }
+    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."
+    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)
+        }
+    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. Argue positively.")
-    agent_b.send_message("You are Agent B, a skeptic and ethicist. Argue cautiously.")
-
-    history = [
-        {"role": "user", "name": "SYSTEM", "content": f"**DEBATE TOPIC:** {topic}\n**STARTING QUESTION:** {question}", "avatar_url": AVATAR_IMAGES["SYSTEM"]}
-    ]
-    yield history
-
-    last_msg = question
-    for _ in range(5):
-        time.sleep(0.5)
-        res_a = agent_a.send_message(f"{last_msg}")
-        history.append({"role": "assistant", "name": "Agent A (Optimist)", "content": res_a.text, "avatar_url": AVATAR_IMAGES["Agent A (Optimist)"]})
-        yield history
-
-        res_b = agent_b.send_message(f"Respond to: {res_a.text}")
-        history.append({"role": "assistant", "name": "Agent B (Skeptic)", "content": res_b.text, "avatar_url": AVATAR_IMAGES["Agent B (Skeptic)"]})
-        last_msg = res_b.text
-        yield history
-
-# --- UI Setup ---
-with gr.Blocks(css=cyberpunk_css, title="AI DEBATE SIMULATOR v2.5") as demo:
-    gr.Markdown("# AI DEBATE SIMULATOR v2.5")
-    gr.Markdown("## 10-Round AI Debate with Gemini 2.5 Agents")
-
-    topic_btn = gr.Button("🎯 Generate Topics")
-    topic_dropdown = gr.Dropdown(label="Choose Topic")
-    question_dropdown = gr.Dropdown(label="Choose Starting Question")
-    start_btn = gr.Button("🚀 Start Debate")
-    chatbox = gr.Chatbot(label="Debate Arena", height=700, avatar_images=AVATAR_IMAGES)
-
-    state = gr.State()
-
-    topic_btn.click(fn=generate_topics, outputs=topic_dropdown)
-    topic_dropdown.change(fn=generate_questions, inputs=topic_dropdown, outputs=question_dropdown)
-    start_btn.click(fn=run_debate, inputs=[topic_dropdown, question_dropdown], outputs=chatbox)
-
+    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
+
+        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,
+    }
+
+# 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")
+
+    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
 if __name__ == "__main__":
-    demo.launch(debug=True, ssr_mode=False)
+    demo.launch(debug=True, ssr_mode=False)