Update app.py

app.py

@@ -2,262 +2,125 @@ import os
 import gradio as gr
 from google import genai
 import json
+import time
 import re
 
 # --- Load Gemini API key securely ---
 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.")
+    raise ValueError("\u274c GEMINI_API_KEY not found. Please set it in your Hugging Face Space Secrets.")
 
+# --- Initialize the client ---
 client = genai.Client()
 
-# --- Constants ---
-MODEL_NAME = "gemini-1.5-flash-latest"
-
+# --- Avatar Images ---
 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"
 }
 
 # --- 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; --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);
+  --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);
 }
-.gradio-container { background: linear-gradient(45deg, var(--bg-color-1), var(--bg-color-2)); font-family: 'Orbitron', sans-serif; }
-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); 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; }
-.chatbot { background-color: rgba(13, 0, 26, 0.8); border: 1px solid var(--neon-pink); box-shadow: var(--border-glow); }
-.gr-input, .gr-dropdown, .gr-radio { border: 1px solid var(--neon-cyan) !important; background-color: var(--bg-color-2) !important; color: var(--neon-cyan) !important; }
-textarea { background-color: rgba(13, 0, 26, 0.8) !important; color: var(--neon-cyan) !important; border: 1px solid var(--neon-pink) !important; }
-.accordion-header { background-color: var(--bg-color-3) !important; }
-footer { display: none !important; }
 """
 
-# --- Core Logic Functions ---
-
-def generate_debate_data_on_load():
-    """
-    Called once on app load. Generates a full set of topics and questions
-    in a single API call, structured as JSON for robust parsing.
-    """
-    yield {
-        topic_dropdown: gr.update(interactive=False, placeholder="⏳ Generating fresh topics from the digital ether..."),
-        question_dropdown: gr.update(interactive=False, placeholder=""),
-        start_debate_btn: gr.update(interactive=False)
-    }
-
-    prompt = """
-    Generate a JSON object for an AI debate simulator.
-    The root of the object should be a single key, "debates".
-    The value of "debates" should be an array of 5 distinct, high-level, debatable topics related to AI, finance, and the future of technology.
-    Each element in the array must be an object with two keys:
-    1. "topic": A string containing the debatable topic.
-    2. "questions": An array of exactly 4 provocative, concise starting questions related to that topic.
-
-    Ensure the output is only the raw JSON object, without any markdown formatting like ```json.
-    """
-    try:
-        model = genai.GenerativeModel(MODEL_NAME)
-        response = model.generate_content(prompt)
-
-        # Clean the response to ensure it's valid JSON
-        json_text = re.search(r'\{.*\}', response.text, re.DOTALL).group(0)
-        data = json.loads(json_text)
-
-        debate_data = {item['topic']: item['questions'] for item in data['debates']}
-        topics = list(debate_data.keys())
-
-        yield {
-            debate_data_state: debate_data,
-            topic_dropdown: gr.update(choices=topics, value=None, interactive=True, placeholder="Step 1: Select a Generated Topic"),
-        }
-    except Exception as e:
-        gr.Error(f"Failed to generate debate data. The AI may be busy. Please try refreshing. Error: {e}")
-        yield {
-            debate_data_state: {},
-            topic_dropdown: gr.update(choices=[], placeholder="❌ Error: Could not generate topics.", interactive=False)
-        }
-
-
-def update_question_choices(topic, debate_data):
-    """Populates the question dropdown from the pre-fetched state. This is instant."""
-    questions = debate_data.get(topic, [])
-    return gr.update(choices=questions, value=None, interactive=True, placeholder="Step 2: Select a Starting Question")
-
-def lock_controls_for_debate():
-    """Disables UI controls when the debate starts."""
-    return {
-        topic_dropdown: gr.update(interactive=False),
-        question_dropdown: gr.update(interactive=False),
-        start_debate_btn: gr.update(interactive=False, visible=False),
-        reset_btn: gr.update(interactive=False)
-    }
-
-def run_debate(topic, question, history):
-    """Main function to run the streamed debate between two AI agents."""
-    if not topic or not question:
-        gr.Error("Topic and Question must be selected before starting.")
-        return
-
-    # Initialize models and chat sessions
-    try:
-        model = genai.GenerativeModel(MODEL_NAME)
-        agent_a_chat = model.start_chat(history=[
-            {'role': 'user', 'parts': ["You are Agent A, a tech optimist and futurist. Your goal is to argue for the positive, transformative aspects of any debate topic."]},
-            {'role': 'model', 'parts': ["Understood. I am Agent A, ready to champion the future."]}
-        ])
-        agent_b_chat = model.start_chat(history=[
-            {'role': 'user', 'parts': ["You are Agent B, a cautious skeptic and ethicist. Your goal is to argue the negative, risky, or cautionary side of any debate topic."]},
-            {'role': 'model', 'parts': ["Acknowledged. I am Agent B, prepared to voice caution and examine the risks."]}
-        ])
-    except Exception as e:
-        gr.Error(f"Failed to initialize AI models. Details: {e}")
-        return
-
-    # --- Start the debate ---
-    system_message = f"**DEBATE TOPIC:** {topic}\n\n**STARTING QUESTION:** {question}"
-    history.append({"role": "user", "name": "SYSTEM", "content": system_message, "avatar_url": AVATAR_IMAGES["SYSTEM"]})
+# --- Utils ---
+def parse_list(text):
+    lines = text.strip().split('\n')
+    return [re.sub(r'^\d+[.)]\s*', '', line).strip() for line in lines if line.strip()]
+
+# --- 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 ---
+def run_debate(topic, question):
+    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_response = question
-    num_turns = 5
-
-    for i in range(num_turns):
-        # --- Agent A's Turn (Optimist) ---
-        history.append({"role": "assistant", "name": "Agent A (Optimist)", "content": "", "avatar_url": AVATAR_IMAGES["Agent A (Optimist)"]})
-        try:
-            prompt_a = f"Based on this: '{last_response}', provide your optimistic argument in a concise paragraph."
-            stream_a = agent_a_chat.send_message(prompt_a, stream=True)
-            full_response = ""
-            for chunk in stream_a:
-                full_response += chunk.text
-                history[-1]["content"] = full_response + "▌"
-                yield history
-            history[-1]["content"] = full_response
-            last_response = full_response
-            yield history
-        except Exception as e:
-            history[-1]["content"] = f"⚠️ Agent A encountered an error: {e}"
-            yield history
-            break
-
-        # --- Agent B's Turn (Skeptic) ---
-        history.append({"role": "assistant", "name": "Agent B (Skeptic)", "content": "", "avatar_url": AVATAR_IMAGES["Agent B (Skeptic)"]})
-        try:
-            prompt_b = f"Respond to Agent A: '{last_response}' with a skeptical counter-argument in a concise paragraph."
-            stream_b = agent_b_chat.send_message(prompt_b, stream=True)
-            full_response = ""
-            for chunk in stream_b:
-                full_response += chunk.text
-                history[-1]["content"] = full_response + "▌"
-                yield history
-            history[-1]["content"] = full_response
-            last_response = full_response
-            yield history
-        except Exception as e:
-            history[-1]["content"] = f"⚠️ Agent B encountered an error: {e}"
-            yield history
-            break
-
-    history.append({"role": "user", "name": "SYSTEM", "content": "--- DEBATE CONCLUDED ---", "avatar_url": AVATAR_IMAGES["SYSTEM"]})
-    yield {
-        chatbot: history,
-        debate_log_state: history,
-        save_btn: gr.update(visible=True),
-        reset_btn: gr.update(interactive=True)
-    }
-
-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:
-        transcript += f"[{message.get('name', 'SYSTEM')}]\n{message.get('content', '')}\n\n---------------------------------\n\n"
-    filepath = "debate_transcript.txt"
-    with open(filepath, "w", encoding="utf-8") as f:
-        f.write(transcript)
-    return filepath
-
-def reset_all():
-    """Resets the UI. Note: This will not re-generate topics. A page refresh is needed for that."""
-    return {
-        chatbot: [],
-        debate_log_state: [],
-        topic_dropdown: gr.update(value=None, interactive=True),
-        question_dropdown: gr.update(value=None, interactive=False, choices=[], placeholder="Select a topic first"),
-        start_debate_btn: gr.update(interactive=False, visible=True),
-        save_btn: gr.update(visible=False),
-        download_file: gr.update(value=None, visible=False),
-        reset_btn: gr.update(interactive=True)
-    }
-
-# --- Gradio UI Layout ---
-with gr.Blocks(css=cyberpunk_css, theme=gr.themes.Base(), title="AI Debate Simulator v3.1") as demo:
-    debate_data_state = gr.State({}) # Holds {"topic": ["q1", "q2"]}
-    debate_log_state = gr.State([])
-
-    gr.Markdown("# 👾 AI DEBATE SIMULATOR v3.1 👾")
-    gr.Markdown("## On-the-Fly Topic Generation Meets Real-Time Streaming")
-    chatbot = gr.Chatbot(label="Debate Arena", height=600, avatar_images=list(AVATAR_IMAGES.values()))
-
-    with gr.Accordion("Debate Setup", open=True):
-        with gr.Row():
-            topic_dropdown = gr.Dropdown(label="Topic", interactive=False, placeholder="⏳ Loading initial data...")
-            question_dropdown = gr.Dropdown(label="Starting Question", interactive=False)
-    with gr.Row():
-        start_debate_btn = gr.Button("🚀 START DEBATE", interactive=False)
-        reset_btn = gr.Button("🔄 NEW DEBATE")
-    with gr.Row():
-        save_btn = gr.Button("💾 Download Transcript", visible=False)
-    download_file = gr.File(label="Download Link", visible=False)
-
-    # --- Event Listeners ---
-    demo.load(
-        fn=generate_debate_data_on_load,
-        outputs=[debate_data_state, topic_dropdown, question_dropdown, start_debate_btn]
-    )
+    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
 
-    topic_dropdown.change(
-        fn=update_question_choices,
-        inputs=[topic_dropdown, debate_data_state],
-        outputs=question_dropdown
-    ).then(fn=lambda: gr.update(interactive=False), outputs=start_debate_btn)
+        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
 
-    question_dropdown.change(
-        fn=lambda x: gr.update(interactive=bool(x)),
-        inputs=question_dropdown,
-        outputs=start_debate_btn
-    )
+# --- 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")
 
-    start_debate_btn.click(
-        fn=lock_controls_for_debate,
-        outputs=[topic_dropdown, question_dropdown, start_debate_btn, reset_btn]
-    ).then(
-        fn=run_debate,
-        inputs=[topic_dropdown, question_dropdown, chatbot],
-        outputs=[chatbot, debate_log_state, save_btn, reset_btn]
-    )
+    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)
 
-    save_btn.click(
-        fn=create_transcript,
-        inputs=debate_log_state,
-        outputs=download_file
-    ).then(fn=lambda: gr.update(visible=True), outputs=download_file)
+    state = gr.State()
 
-    reset_btn.click(
-        fn=reset_all,
-        outputs=[chatbot, debate_log_state, topic_dropdown, question_dropdown, start_debate_btn, save_btn, download_file, reset_btn]
-    )
+    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)
 
-# --- Launch the App ---
 if __name__ == "__main__":
-    demo.queue().launch(debug=True)
+    demo.launch(debug=True, ssr_mode=False)