Update app.py

app.py

@@ -1,39 +1,334 @@
 import os
 import gradio as gr
 from google import genai
+import json
+import time
+import re
 
 # --- 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. Set it in Hugging Face Secrets.")
+    raise ValueError("❌ GEMINI_API_KEY not found. Please set it in your Hugging Face Space Secrets.")
 
-# --- Initialize the client (no configure method!) ---
-client = genai.Client()
+# --- Initialize the client ---
+# The new genai.Client() is the recommended way.
+client = genai.Client(api_key=GEMINI_API_KEY)
 
-# --- Create chat session ---
-chat = client.chats.create(model="gemini-2.5-flash")
+# --- UI Customization: Bright Neon Trippy Blending ---
+cyberpunk_css = """
+@import url('https://fonts.googleapis.com/css2?family=Orbitron:wght@400;700&display=swap');
 
-# --- Define chat function ---
-def chat_with_gemini(user_input, history):
+: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);
+    --text-glow: 0 0 3px var(--neon-green);
+}
+
+.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);
+}
+
+.user-message, .bot-message {
+    border-radius: 5px;
+    padding: 10px;
+    box-shadow: 0 0 5px var(--neon-pink) inset;
+}
+
+.user-message { border-left: 5px solid var(--neon-cyan); background-color: rgba(0, 255, 255, 0.05); }
+.bot-message { border-left: 5px solid var(--neon-pink); background-color: rgba(255, 0, 255, 0.05); }
+
+.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; }
+"""
+
+# --- Helper function to parse LLM lists ---
+def parse_list_from_llm(text_response):
+    """Cleans up numbered/bulleted lists from Gemini response."""
+    lines = text_response.strip().split('\n')
+    # Remove numbering like "1. ", "1) ", or bullet points "* "
+    cleaned_lines = [re.sub(r'^\d+[\.\)]\s*|^\*\s*', '', line).strip() for line in lines]
+    return [line for line in cleaned_lines if line]
+
+# --- Core Application Logic ---
+
+def generate_topics():
+    """Calls Gemini to generate 20 debate topics."""
+    prompt = "Generate a list of exactly 20 specific, debatable topics related to the future of coding, banking, and artificial intelligence. The topics should be concise and provocative. Format as a numbered list."
     try:
-        response = chat.send_message(user_input)
-        history.append((user_input, response.text))
-        return history, history
+        response = client.generate_text(model="models/gemini-pro", prompt=prompt)
+        topics = parse_list_from_llm(response.result)
+        # UI Updates: Show next step, hide current
+        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:
-        history.append((user_input, f"⚠️ Error: {str(e)}"))
-        return history, history
+        gr.Warning(f"Error generating topics: {e}")
+        return {topics_state: [], topic_dropdown: gr.Dropdown(visible=False), generate_topics_btn: gr.Button(visible=True)}
+
+def generate_questions(selected_topic):
+    """Calls Gemini to generate 20 questions for the selected topic."""
+    if not selected_topic:
+        return {question_dropdown: gr.Dropdown(value=None, visible=False)}
+
+    prompt = f"For the debate topic '{selected_topic}', generate a list of exactly 20 thought-provoking starting questions or controversial statements to kick off the debate. Format as a numbered list."
+    try:
+        response = client.generate_text(model="models/gemini-pro", prompt=prompt)
+        questions = parse_list_from_llm(response.result)
+        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)}
+
+def enable_debate_button(selected_question):
+    """Makes the debate button visible once a question is chosen."""
+    return {start_debate_btn: gr.Button(visible=bool(selected_question))}
+
+def run_debate(topic, question, history):
+    """Main function to run the 20-turn debate between two AI agents."""
+    if not topic or not question:
+        gr.Warning("Topic and Question must be selected before starting.")
+        return [], history
+
+    # --- Initialize two separate agent chat sessions for this debate ---
+    # This maintains their distinct personalities and contexts.
+    agent_a_persona = "You are Agent A, a tech optimist and futurist. You see the best in new technology. Your goal is to argue for the positive aspects of the debate topic. Keep your responses concise and impactful."
+    agent_b_persona = "You are Agent B, a cautious skeptic and ethicist. You critically examine the potential downsides and risks of new technology. Your goal is to argue for the negative or cautionary aspects of the debate topic. Keep your responses concise and impactful."
+
+    agent_a_chat = client.start_chat(context=agent_a_persona, history=[])
+    agent_b_chat = client.start_chat(context=agent_b_persona, history=[])
+
+    current_history = [(None, f"**DEBATE TOPIC:** {topic}\n**STARTING QUESTION:** {question}")]
+    yield current_history, current_history # Update UI immediately
+
+    last_response = question
+    
+    # The debate loop: 10 turns for each agent = 20 total messages
+    for i in range(10):
+        try:
+            # --- Agent A's Turn (Optimist) ---
+            time.sleep(1) # Small delay for realism
+            prompt_a = f"Based on the last statement ('{last_response}'), provide your optimistic counter-argument or statement."
+            response_a = agent_a_chat.send_message(prompt_a)
+            current_history.append(("Agent A (Optimist)", response_a.text))
+            last_response = response_a.text
+            yield current_history, current_history
+
+            # --- Agent B's Turn (Skeptic) ---
+            time.sleep(1)
+            prompt_b = f"Based on Agent A's last statement ('{last_response}'), provide your skeptical counter-argument or cautionary statement."
+            response_b = agent_b_chat.send_message(prompt_b)
+            current_history.append(("Agent B (Skeptic)", response_b.text))
+            last_response = response_b.text
+            yield current_history, current_history
+
+        except Exception as e:
+            error_message = f"⚠️ An error occurred during the debate: {e}"
+            current_history.append((None, error_message))
+            yield current_history, current_history
+            break # Exit loop on error
+    
+    current_history.append((None, "--- DEBATE CONCLUDED ---"))
+    
+    # Final update showing save buttons
+    yield {
+        chatbot: current_history, 
+        debate_log_state: current_history, 
+        save_row: gr.Row(visible=True)
+    }
+
+
+def save_as_json(log):
+    """Saves the debate log to a JSON file."""
+    filepath = "debate_log.json"
+    # Format for clarity
+    formatted_log = [{"speaker": speaker if speaker else "SYSTEM", "message": message} for speaker, message in log]
+    with open(filepath, "w") as f:
+        json.dump(formatted_log, f, indent=2)
+    return gr.File(value=filepath, label="Download JSON")
+
+
+def save_as_html(log):
+    """Saves the debate log to a styled HTML file."""
+    filepath = "debate_log.html"
+    
+    # Simple but effective HTML styling
+    html_style = """
+    <style>
+        body { font-family: sans-serif; background-color: #111; color: #eee; }
+        .container { max-width: 800px; margin: auto; padding: 20px; }
+        .message { margin-bottom: 1em; padding: 1em; border-radius: 8px; }
+        .system { background-color: #333; text-align: center; font-weight: bold; }
+        .agent-a { background-color: #023020; border-left: 5px solid #39ff14; }
+        .agent-b { background-color: #30022b; border-left: 5px solid #ff00ff; }
+        .speaker { font-weight: bold; margin-bottom: 0.5em; display: block; }
+    </style>
+    """
+    
+    html_content = f"<html><head><title>Debate Log</title>{html_style}</head><body><div class='container'>"
+    html_content += "<h1>Debate Log</h1>"
+    
+    for speaker, message in log:
+        if speaker == "Agent A (Optimist)":
+            css_class = "agent-a"
+        elif speaker == "Agent B (Skeptic)":
+            css_class = "agent-b"
+        else:
+            css_class = "system"
+            speaker = "SYSTEM"
+
+        # Sanitize message to prevent HTML injection
+        safe_message = gr.Markdown(message).value.replace('\n', '<br>')
+        html_content += f'<div class="message {css_class}"><span class="speaker">{speaker}</span>{safe_message}</div>'
+        
+    html_content += "</div></body></html>"
+    
+    with open(filepath, "w") as f:
+        f.write(html_content)
+        
+    return gr.File(value=filepath, label="Download HTML")
+
+
+def reset_all():
+    """Resets the entire UI to its initial state."""
+    return {
+        chatbot: [],
+        debate_log_state: [],
+        topics_state: [],
+        questions_state: [],
+        generate_topics_btn: gr.Button(visible=True),
+        topic_dropdown: gr.Dropdown(value=None, choices=[], label="Step 2: Select a Topic", visible=False),
+        question_dropdown: gr.Dropdown(value=None, choices=[], label="Step 3: Select a Starting Question", visible=False),
+        start_debate_btn: gr.Button(visible=False),
+        save_row: gr.Row(visible=False),
+        download_file: None,
+    }
+
+
+# --- Gradio UI Layout ---
+with gr.Blocks(theme=gr.themes.Base(), css=cyberpunk_css, title="AI DEBATE SIMULATOR_v2.5") as demo:
+    # State management for data persistence between interactions
+    debate_log_state = gr.State([])
+    topics_state = gr.State([])
+    questions_state = gr.State([])
+
+    gr.Markdown("# 👾 AI DEBATE SIMULATOR_v2.5 👾")
+    gr.Markdown("## A 20-Turn War of Wits Between Two AI Agents")
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            gr.Markdown("### **CONTROL PANEL**")
+            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, interactive=True)
+            question_dropdown = gr.Dropdown(label="Step 3: Select a Starting Question", visible=False, interactive=True)
+            start_debate_btn = gr.Button("Step 4: START THE DEBATE", visible=False)
+            
+            with gr.Row(visible=False) as save_row:
+                save_json_btn = gr.Button("Save as JSON")
+                save_html_btn = gr.Button("Save as HTML")
+            
+            download_file = gr.File(label="Download Your Log")
+            
+            reset_btn = gr.Button("Start New Debate (Reset All)", variant="stop")
+
+        with gr.Column(scale=3):
+            chatbot = gr.Chatbot(
+                label="Debate Arena",
+                bubble_full_width=False,
+                avatar_images=(None, "https://i.imgur.com/Q51aD2S.png"), # Simple bot avatar
+                height=700
+            )
+
+    # --- Event Handling 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]
+    )
 
-# --- Gradio UI ---
-with gr.Blocks(title="Gemini 2.5 Flash Chatbot") as demo:
-    gr.Markdown("## 🤖 Gemini 2.5 Flash Chatbot\nPowered by Google Generative AI")
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox(placeholder="Type your message...", label="Your Message")
-    clear = gr.Button("🧹 Clear")
-    state = gr.State([])
+    start_debate_btn.click(
+        fn=lambda: {
+            # Disable controls during debate
+            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, debate_log_state],
+        outputs=[chatbot, debate_log_state, save_row]
+    )
+    
+    save_json_btn.click(save_as_json, inputs=debate_log_state, outputs=download_file)
+    save_html_btn.click(save_as_html, inputs=debate_log_state, outputs=download_file)
 
-    msg.submit(chat_with_gemini, inputs=[msg, state], outputs=[chatbot, state])
-    clear.click(lambda: ([], []), None, [chatbot, state])
+    reset_btn.click(
+        fn=reset_all,
+        inputs=None,
+        outputs=[
+            chatbot, debate_log_state, topics_state, questions_state,
+            generate_topics_btn, topic_dropdown, question_dropdown,
+            start_debate_btn, save_row, download_file
+        ]
+    )
 
-# --- Launch ---
+# --- Launch the App ---
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(debug=True)