Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 import gradio as gr
 
-# Load model
+# Load FLAN-T5 model
 tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
 model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
 pipe = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
@@ -14,24 +14,31 @@ personas = {
     "🟡 Neutral": "Respond with a balanced and unbiased view"
 }
 
-# Generate debate response
+# Generate responses
 def generate_debate(topic):
+    if not topic.strip():
+        return "⚠️ Please enter a valid debate topic."
+    
     responses = {}
     for label, style in personas.items():
-        prompt = f"You are a debater. {style}. The topic is: '{topic}'. What is your opinion?"
-        out = pipe(prompt, max_new_tokens=100, temperature=0.7)[0]["generated_text"]
-        responses[label] = out
-    return responses
+        try:
+            prompt = f"You are a debater. {style}. The topic is: '{topic}'. What is your opinion?"
+            out = pipe(prompt, max_new_tokens=120, temperature=0.7)[0]["generated_text"]
+            responses[label] = out.strip()
+        except Exception as e:
+            responses[label] = f"❌ Error generating response: {str(e)}"
+    
+    result = "\n\n".join([f"### {label}\n{resp}" for label, resp in responses.items()])
+    return result
 
 # Gradio Interface
-def interface_fn(topic):
-    responses = generate_debate(topic)
-    return "\n\n".join([f"**{k}**: {v}" for k, v in responses.items()])
-
-demo = gr.Interface(fn=interface_fn,
-                    inputs=gr.Textbox(label="Enter a Debate Topic"),
-                    outputs=gr.Markdown(),
-                    title="🎙️ Multi-Agent Debate Simulator",
-                    description="This app simulates a debate from different viewpoints using FLAN-T5 on Hugging Face.")
+demo = gr.Interface(
+    fn=generate_debate,
+    inputs=gr.Textbox(lines=2, placeholder="e.g. Should AI replace teachers?", label="Enter a Debate Topic"),
+    outputs=gr.Markdown(),
+    title="🎙️ Multi-Agent Debate Simulator",
+    description="This app simulates a debate from different viewpoints using FLAN-T5 (Google) on Hugging Face 🤗 Transformers. Just enter a topic and see how Optimist, Pessimist, and Neutral personas respond!"
+)
 
+# Launch the app
 demo.launch()