Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import gradio as gr
 import json
 from transformers import pipeline
@@ -15,46 +14,37 @@ def load_personas():
 
 personas = load_personas()
 
-# Use GPT-2 for demo purposes
-bot = pipeline("text-generation", model="gpt2")
+# Use distilgpt2 for faster responses
+bot = pipeline("text-generation", model="distilgpt2")
 
-# Debate function
-def start_debate(topic):
+# Debate generator (yields one response at a time)
+def debate_generator(topic):
     history = f"Debate Topic: {topic}\n"
-    outputs = []
 
     for idx, persona in enumerate(personas):
         prompt = f"{persona['prompt_style']}\n{history}\n{persona['name']}:"
-        response = bot(prompt, max_length=100, do_sample=True)[0]['generated_text']
+        response = bot(prompt, max_length=50, do_sample=True)[0]['generated_text']
         response_text = response[len(prompt):].strip().split("\n")[0]
 
-        # Update history
         history += f"{persona['name']}: {response_text}\n"
 
-        # Generate TTS using gTTS (placeholder for 11 Labs)
         tts = gTTS(response_text, lang='en')
         audio_file = f"temp_{idx}.mp3"
         tts.save(audio_file)
 
-        outputs.append((f"{persona['name']}: {response_text}", audio_file))
-
-    return outputs
+        yield f"{persona['name']}: {response_text}", audio_file
 
-# Gradio Interface
-def debate_interface(topic):
-    results = start_debate(topic)
-    texts = [r[0] for r in results]
-    audios = [r[1] for r in results]
-    return texts, audios
+with gr.Blocks() as iface:
+    gr.Markdown("# AI Persona Debate")
+    topic_input = gr.Textbox(label="Enter Debate Question")
+    start_button = gr.Button("Start Debate")
+    output_text = gr.Textbox(label="Current Response")
+    output_audio = gr.Audio(label="Voice", autoplay=True)
 
-text_output = [gr.Textbox(label=f"Turn {i+1}") for i in range(6)]
-audio_output = [gr.Audio(label=f"Voice {i+1}") for i in range(6)]
+    def run_debate(topic):
+        for text, audio in debate_generator(topic):
+            yield text, audio
 
-iface = gr.Interface(
-    fn=debate_interface,
-    inputs=gr.Textbox(label="Enter Debate Question"),
-    outputs=text_output + audio_output,
-    title="AI Persona Debate"
-)
+    start_button.click(run_debate, inputs=topic_input, outputs=[output_text, output_audio])
 
 iface.launch()