Update app.py

app.py

@@ -1,47 +1,45 @@
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 import gradio as gr
 
-# Load stronger open LLM (Mistral-7B-Instruct)
-model_name = "mistralai/Mistral-7B-Instruct-v0.1"
+# Load Falcon-7B-Instruct (open and free)
+model_name = "tiiuae/falcon-7b-instruct"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-# Define agent styles
+# Define personas
 personas = {
-    "🟢 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."
+    "🟢 Optimist": "Someone who sees the good and hopeful side of any issue.",
+    "🔴 Pessimist": "Someone who focuses on the risks, negatives, or problems in any issue.",
+    "🟡 Neutral": "Someone who presents a balanced, unbiased view based on logic."
 }
 
 # 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:"
-    )
+def build_prompt(topic, style):
+    return f"""You are a debater. Take this persona: {style}.
+Debate Topic: "{topic}"
+Provide a thoughtful and opinionated response with reasoning.
+Answer:"""
 
-# Generate debate
+# Generate responses
 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
+    for label, persona in personas.items():
+        prompt = build_prompt(topic, persona)
+        output = pipe(prompt, max_new_tokens=200, temperature=0.9)[0]["generated_text"]
+        answer = output.split("Answer:")[-1].strip()
+        results[label] = answer
     return results
 
-# Gradio UI
+# Gradio interface
 def run_debate(topic):
-    response = debate(topic)
-    return "\n\n".join([f"**{k}**:\n{v}" for k, v in response.items()])
+    responses = debate(topic)
+    return "\n\n".join([f"**{k}**:\n{v}" for k, v in responses.items()])
 
 gr.Interface(
     fn=run_debate,
     inputs=gr.Textbox(label="Enter a Debate Topic"),
     outputs=gr.Markdown(),
     title="🎙️ Multi-Agent Debate Simulator",
-    description="Debates from multiple perspectives using Mistral-7B on Hugging Face 🤗."
+    description="Simulates multi-perspective debates using Falcon-7B on Hugging Face 🤗."
 ).launch()