Update app.py

app.py

@@ -1,7 +1,10 @@
-import os
+
 import gradio as gr
-from google import genai
+import google as genai
+import os
+import random
 import time
+from textwrap import dedent
 
 # --- Load Gemini API key securely ---
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
@@ -10,161 +13,324 @@ if not GEMINI_API_KEY:
 
 client = genai.Client()
 
-# --- Constants ---
-MODEL_NAME = "gemini-2.5-flash"
-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",
+
+# --- 1. CONSTANTS ---
+# Replicates constants from the original React app
+GEMINI_TEXT_MODEL = 'gemini-2.5-flash'
+# Note: imagen-3.0-generate-002 is not available in the Python SDK yet.
+# We will handle this limitation gracefully in the app logic.
+GEMINI_IMAGE_MODEL = 'imagen-3.0-generate-002'
+
+class Agent:
+    SYSTEM = 'SYSTEM'
+    OPTIMIST = 'OPTIMIST'
+    SKEPTIC = 'SKEPTIC'
+
+SYSTEM_AVATAR = 'https://i.imgur.com/z3uxLzT.png'
+AGENT_AVATARS = {
+    Agent.OPTIMIST: 'https://i.imgur.com/2GMV9y7.png',
+    Agent.SKEPTIC: 'https://i.imgur.com/KzX3T21.png',
+    Agent.SYSTEM: SYSTEM_AVATAR,
 }
 
-# --- 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);
+AGENT_PROMPTS = {
+    Agent.OPTIMIST: {
+        "personality": "You are Agent A, a tech optimist and futurist. You see the incredible potential and benefits in every new technology. Your arguments are passionate, forward-thinking, and inspiring. Respond in 1-2 concise paragraphs.",
+        "prompt": lambda topic, last_response: f"Debate Topic: '{topic}'. The last statement was: '{last_response}'. Provide your powerful, optimistic argument.",
+    },
+    Agent.SKEPTIC: {
+        "personality": "You are Agent B, a cautious skeptic and ethicist. You ground the conversation by highlighting risks, societal impact, and unforeseen consequences. Your arguments are critical, grounded, and raise important questions. Respond in 1-2 concise paragraphs.",
+        "prompt": lambda topic, last_response: f"Debate Topic: '{topic}'. Agent A said: '{last_response}'. Provide your insightful, skeptical counter-argument.",
+    },
 }
-.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) !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); }
-.message-bubble { background: none !important; border: none !important; }
-footer { display: none !important; }
-"""
-
-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:
-        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()
-    except Exception as e:
-        history_list[-1]["content"] = f"⚠️ An error occurred: {e}"
-    
-    yield history_list
-    return full_response.strip()
 
-def run_full_debate_flow(chat_history):
-    """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}
-    
-    chat_history.clear()
-    yield {
-        start_button: gr.update(interactive=False, value="🔄 Initializing..."),
-        chatbot: chat_history,
-        download_button: gr.update(visible=False),
-        download_file_output: gr.update(value=None, visible=False)
-    }
+GENERATION_PROMPTS = {
+    "topic": "Generate one specific, debatable topic about the future of AI. The topic should be a short phrase. Output only the topic itself.",
+    "question": lambda topic: f"For the debate topic '{topic}', create one provocative question to start the debate. Output only the question itself.",
+}
 
-    yield add_system_message("Receiving transmission... Booting up debate protocol.")
-    
-    try:
-        gen_model = genai.GenerativeModel(MODEL_NAME)
-        
-        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_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 or not topic: raise ValueError("Topic generation failed.")
-
-        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_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 or not question: raise ValueError("Question generation failed.")
-
-    except Exception as e:
-        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
-        
-    yield add_system_message("Initializing agents... Let the debate begin!")
-    yield { start_button: gr.update(value="⚔️ DEBATE IN PROGRESS...") }
+INITIAL_MESSAGE = [(None, "Awaiting signal to begin debate protocol...")]
+
+# --- 2. JAVASCRIPT FOR TEXT-TO-SPEECH ---
+# This JS code runs in the user's browser to enable speech.
+tts_js = dedent("""
+function() {
+    let optimistVoice = null;
+    let skepticVoice = null;
+    let voices = [];
+
+    const OPTIMIST_VOICE_CANDIDATES = ['Google UK English Female', 'Samantha', 'Microsoft Zira - English (United States)', 'Karen', 'en-GB-female'];
+    const SKEPTIC_VOICE_CANDIDATES = ['Google UK English Male', 'Daniel', 'Microsoft David - English (United States)', 'Rishi', 'en-GB-male'];
+
+    const getVoices = () => {
+        return new Promise((resolve) => {
+            const voiceList = window.speechSynthesis.getVoices();
+            if (voiceList.length) {
+                resolve(voiceList);
+                return;
+            }
+            window.speechSynthesis.onvoiceschanged = () => {
+                resolve(window.speechSynthesis.getVoices());
+            };
+        });
+    };
+
+    const selectVoices = async () => {
+        voices = await getVoices();
+        if (!voices.length) return;
+
+        for (const name of OPTIMIST_VOICE_CANDIDATES) {
+            const found = voices.find(v => v.name === name);
+            if (found) { optimistVoice = found; break; }
+        }
+        if (!optimistVoice) {
+            optimistVoice = voices.find(v => v.lang.startsWith('en') && /female/i.test(v.name)) || voices.find(v => v.lang.startsWith('en')) || null;
+        }
+
+        for (const name of SKEPTIC_VOICE_CANDIDATES) {
+            const found = voices.find(v => v.name === name && v !== optimistVoice);
+            if (found) { skepticVoice = found; break; }
+        }
+        if (!skepticVoice) {
+            skepticVoice = voices.find(v => v.lang.startsWith('en') && /male/i.test(v.name) && v !== optimistVoice) || voices.find(v => v.lang.startsWith('en') && v !== optimistVoice) || null;
+        }
+    };
+
+    window.initSpeech = () => {
+        if (typeof window === 'undefined' || !('speechSynthesis' in window)) {
+            console.warn("Speech Synthesis not supported in this browser.");
+            return;
+        }
+        selectVoices();
+        window.speechSynthesis.onvoiceschanged = selectVoices;
+    };
+
+    window.speakText = (text, agent) => {
+        if (!window.speechSynthesis || !text) return;
+        window.speechSynthesis.cancel();
+        const utterance = new SpeechSynthesisUtterance(text);
+        if (agent === 'OPTIMIST' && optimistVoice) {
+            utterance.voice = optimistVoice;
+        } else if (agent === 'SKEPTIC' && skepticVoice) {
+            utterance.voice = skepticVoice;
+        }
+        utterance.pitch = 1;
+        utterance.rate = 1;
+        utterance.volume = 0.9;
+        window.speechSynthesis.speak(utterance);
+    };
+
+    window.cancelSpeech = () => {
+        if (window.speechSynthesis) {
+            window.speechSynthesis.cancel();
+        }
+    };
     
-    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
-    for _ in range(5):
-        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 }
-        
-        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 }
-            
-    yield add_system_message("--- DEBATE CONCLUDED ---")
-    yield {
-        chatbot: chat_history,
-        start_button: gr.update(interactive=True, value="▷ START NEW DEBATE"),
-        download_button: gr.update(visible=True)
+    return (history, text, agent) => {
+        window.speakText(text, agent);
+        return history;
     }
+}
+""")
 
-def create_transcript(chat_history):
-    """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 gr.update(value=filepath, visible=True)
-
-# --- Gradio UI Layout ---
-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.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()),
-        show_copy_button=True
-    )
-
-    with gr.Row():
-        start_button = gr.Button("▷ START NEW DEBATE", variant="primary", scale=3)
-        download_button = gr.Button("💾 Download Transcript", visible=False, scale=1)
+# --- 3. GEMINI SERVICE ---
+class GeminiService:
+    """A service class to encapsulate all interactions with the Google Gemini API."""
+    def __init__(self):
+        try:
+            api_key = os.environ["API_KEY"]
+            genai.configure(api_key=api_key)
+        except KeyError:
+            raise RuntimeError("FATAL: API_KEY environment variable not set. Please add it to your environment.")
+        self.text_model = genai.GenerativeModel(GEMINI_TEXT_MODEL)
+
+    def generate_topic(self):
+        """Generates a debate topic."""
+        response = self.text_model.generate_content(
+            GENERATION_PROMPTS["topic"],
+            generation_config={"temperature": 1.0}
+        )
+        return response.text.strip()
+
+    def generate_question(self, topic):
+        """Generates a starting question for a given topic."""
+        response = self.text_model.generate_content(
+            GENERATION_PROMPTS["question"](topic),
+            generation_config={"temperature": 1.0}
+        )
+        return response.text.strip()
+
+    def create_agent_session(self, agent_type):
+        """Creates a stateful chat session for an agent with a specific personality."""
+        personality = AGENT_PROMPTS[agent_type]["personality"]
+        return self.text_model.start_chat(
+            history=[
+                {'role': 'user', 'parts': [personality]},
+                {'role': 'model', 'parts': ["Understood. I am ready."]}
+            ]
+        )
+
+# Instantiate the service. This will raise an error on startup if the API key is missing.
+gemini_service = GeminiService()
+
+# --- 4. GRADIO APP ---
+def create_interface():
+    # Gradio CSS for styling
+    css = """
+    @import url('https://fonts.googleapis.com/css2?family=Orbitron:wght@400;700&family=Roboto:wght@400&display=swap');
+    body { font-family: 'Roboto', sans-serif; }
+    .font-orbitron { font-family: 'Orbitron', sans-serif; }
+    .gradio-container { background: linear-gradient(45deg, #0d001a, #1a001a); }
+    .chatbot { background-color: rgba(0,0,0,0.2) !important; border: none !important; }
+    .chatbot .user-message, .chatbot .bot-message { max-width: 100% !important; }
+    .system-message-container { display: flex; align-items: center; justify-content: center; gap: 1rem; margin: 1rem 0; }
+    .system-message-line { flex-grow: 1; border-top: 1px dashed rgba(52, 211, 153, 0.3); }
+    .system-message-text { color: #34d399; font-family: monospace; font-size: 0.8rem; white-space: pre-wrap; text-align: center; }
+    .image-message-container { padding: 0.5rem; border: 1px solid rgba(168, 85, 247, 0.5); background: rgba(168, 85, 247, 0.1); border-radius: 0.5rem; margin: 1.5rem auto; max-width: 80%; }
+    .image-message-container img { border-radius: 0.25rem; }
+    .image-message-caption { color: #c084fc; font-size: 0.75rem; text-align: center; font-family: monospace; margin-top: 0.5rem; }
+    """
+
+    with gr.Blocks(theme=gr.themes.Base(), css=css) as demo:
+        # App State
+        topic_state = gr.State("")
+        question_state = gr.State("")
+
+        # Header
+        with gr.Row():
+            gr.HTML("""
+            <div style="text-align: center; padding: 1rem;">
+                <h1 class="font-orbitron" style="font-size: 2rem; font-weight: bold; color: #67e8f9; text-shadow: 0 0 5px #00ffff;">AI DEBATE SIMULATOR</h1>
+                <h2 class="font-orbitron" style="font-size: 1rem; color: #f472b6; text-shadow: 0 0 5px #ff00ff;">Hardened Edition</h2>
+            </div>
+            """)
+
+        # Chat Window
+        chatbot = gr.Chatbot(
+            label="Debate Arena",
+            value=INITIAL_MESSAGE,
+            elem_classes=["chatbot"],
+            height=600,
+            avatar_images=(AGENT_AVATARS[Agent.SKEPTIC], AGENT_AVATARS[Agent.OPTIMIST]),
+            show_copy_button=False,
+            bubble_full_width=False,
+        )
+
+        # Controls
+        start_btn = gr.Button("▷ START NEW DEBATE", variant="primary", elem_classes="font-orbitron")
         
-    download_file_output = gr.File(label="Download", visible=False)
+        # Hidden components for JS interop
+        text_to_speak = gr.Textbox(visible=False)
+        agent_to_speak = gr.Textbox(visible=False)
+
+        def add_system_message(history, text):
+            container_html = f"""
+            <div class="system-message-container">
+                <div class="system-message-line"></div>
+                <p class="system-message-text">{text}</p>
+                <div class="system-message-line"></div>
+            </div>
+            """
+            history.append((None, container_html))
+            time.sleep(0.5)
+            return history
+
+        def run_full_debate_flow():
+            yield {start_btn: gr.update(value="🔄 Initializing...", interactive=False), chatbot: []}
+
+            history = []
+            yield { chatbot: history, text_to_speak: "", agent_to_speak: "" }
+
+            try:
+                history = add_system_message(history, "SYSTEM: Receiving transmission... Booting up debate protocol.")
+                yield {chatbot: history}
 
-    start_button.click(
-        fn=run_full_debate_flow,
-        inputs=[chat_history_state],
-        outputs=[chatbot, start_button, download_button, download_file_output]
-    )
+                history = add_system_message(history, "SYSTEM: Contacting orbital AI to generate a fresh topic...")
+                yield {chatbot: history}
+                topic = gemini_service.generate_topic()
+
+                history = add_system_message(history, "SYSTEM: Formulating a provocative starting question...")
+                yield {chatbot: history}
+                question = gemini_service.generate_question(topic)
+                
+                topic_and_question_text = f"TOPIC: {topic}\\n\\nQUESTION: {question}"
+                history = add_system_message(history, topic_and_question_text)
+                yield {chatbot: history, topic_state: topic, question_state: question}
+
+                history = add_system_message(history, "SYSTEM: Initializing agents... Let the debate begin!")
+                yield {chatbot: history, start_btn: gr.update(value="⚔️ DEBATE IN PROGRESS...")}
+
+                optimist_session = gemini_service.create_agent_session(Agent.OPTIMIST)
+                skeptic_session = gemini_service.create_agent_session(Agent.SKEPTIC)
+
+                last_response = question
+                debate_turns = 3
+                image_turn = random.randint(2, debate_turns * 2 - 1)
+                
+                turn_counter = 0
+                for _ in range(debate_turns):
+                    turn_counter += 1
+                    history.append(("", ""))
+                    optimist_prompt = AGENT_PROMPTS[Agent.OPTIMIST]['prompt'](topic, last_response)
+                    stream = optimist_session.send_message(optimist_prompt, stream=True)
+                    full_response = ""
+                    for chunk in stream:
+                        full_response += chunk.text
+                        history[-1] = (full_response + "▌", None)
+                        yield {chatbot: history}
+                    history[-1] = (full_response, None)
+                    last_response = full_response
+                    yield {chatbot: history, text_to_speak: full_response, agent_to_speak: Agent.OPTIMIST}
+
+                    if turn_counter == image_turn:
+                        history = add_system_message(history, "SYSTEM: Visual data generation is not supported in this Python version.\\n(Imagen 3 is not yet available in the Python SDK).")
+                        yield {chatbot: history}
+
+                    turn_counter += 1
+                    history.append(("", ""))
+                    skeptic_prompt = AGENT_PROMPTS[Agent.SKEPTIC]['prompt'](topic, last_response)
+                    stream = skeptic_session.send_message(skeptic_prompt, stream=True)
+                    full_response = ""
+                    for chunk in stream:
+                        full_response += chunk.text
+                        history[-1] = (None, full_response + "▌")
+                        yield {chatbot: history}
+                    history[-1] = (None, full_response)
+                    last_response = full_response
+                    yield {chatbot: history, text_to_speak: full_response, agent_to_speak: Agent.SKEPTIC}
+
+                    if turn_counter == image_turn:
+                        history = add_system_message(history, "SYSTEM: Visual data generation is not supported in this Python version.\\n(Imagen 3 is not yet available in the Python SDK).")
+                        yield {chatbot: history}
+
+            except Exception as e:
+                print(f"An error occurred: {e}")
+                error_message = f"❌ FATAL ERROR: {e}. Protocol terminated."
+                history = add_system_message(history, error_message)
+                yield {chatbot: history, start_btn: gr.update(value="▷ RE-INITIATE DEBATE", interactive=True)}
+                return
+
+            history = add_system_message(history, "--- DEBATE CONCLUDED ---")
+            yield {chatbot: history, start_btn: gr.update(value="▷ START NEW DEBATE", interactive=True)}
+
+        start_btn.click(
+            fn=run_full_debate_flow,
+            outputs=[chatbot, start_btn, topic_state, question_state, text_to_speak, agent_to_speak]
+        )
+        
+        agent_to_speak.change(
+            fn=None,
+            js=tts_js,
+            inputs=[chatbot, text_to_speak, agent_to_speak],
+            outputs=[chatbot],
+            api_name=False
+        )
+        
+        demo.load(fn=None, js="() => { window.initSpeech(); }")
 
-    download_button.click(
-        fn=create_transcript,
-        inputs=[chat_history_state],
-        outputs=download_file_output
-    )
+    return demo
 
+# --- 5. MAIN EXECUTION BLOCK ---
 if __name__ == "__main__":
-    demo.queue().launch(debug=True)
+    app_interface = create_interface()
+    app_interface.queue().launch(debug=True, server_name="0.0.0.0", server_port=7860)