Update app.py

app.py

@@ -1,44 +1,47 @@
-import os
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 import gradio as gr
 
-# 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)
+# Load stronger open LLM (Mistral-7B-Instruct)
+model_name = "mistralai/Mistral-7B-Instruct-v0.1"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-# Agent styles
+# Define agent styles
 personas = {
-    "🟢 Optimist": "Take a positive and hopeful viewpoint",
-    "🔴 Pessimist": "Take a critical or negative viewpoint",
-    "🟡 Neutral": "Respond with a balanced and unbiased view"
+    "🟢 Optimist": "An individual who sees the positive side and hopeful outcomes of any topic.",
+    "🔴 Pessimist": "A critic who focuses on the flaws, risks, and downsides of the topic.",
+    "🟡 Neutral": "An unbiased observer who presents a balanced and fair viewpoint."
 }
 
-# Generate responses
-def generate_debate(topic):
-    if not topic.strip():
-        return "⚠️ Please enter a valid debate topic."
-    
-    responses = {}
-    for label, style in personas.items():
-        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
+# Prompt template
+def build_prompt(topic, persona_desc):
+    return (
+        f"You are a skilled debater. Take on this persona: {persona_desc}\n"
+        f"Debate Topic: \"{topic}\"\n"
+        f"Respond in detail and back your opinion with reasoning.\n"
+        f"Your argument:"
+    )
 
-# Gradio Interface
-demo = gr.Interface(
-    fn=generate_debate,
-    inputs=gr.Textbox(lines=2, placeholder="e.g. Should AI replace teachers?", label="Enter a Debate Topic"),
+# Generate debate
+def debate(topic):
+    results = {}
+    for label, desc in personas.items():
+        prompt = build_prompt(topic, desc)
+        output = pipe(prompt, max_new_tokens=200, temperature=0.9, top_p=0.95)[0]["generated_text"]
+        argument = output.split("Your argument:")[-1].strip()
+        results[label] = argument
+    return results
+
+# Gradio UI
+def run_debate(topic):
+    response = debate(topic)
+    return "\n\n".join([f"**{k}**:\n{v}" for k, v in response.items()])
+
+gr.Interface(
+    fn=run_debate,
+    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 (Google) on Hugging Face 🤗 Transformers. Just enter a topic and see how Optimist, Pessimist, and Neutral personas respond!"
-)
-
-# Launch the app
-demo.launch()
+    description="Debates from multiple perspectives using Mistral-7B on Hugging Face 🤗."
+).launch()