Update app.py

app.py

@@ -1,11 +1,9 @@
 from huggingface_hub import InferenceClient
 import gradio as gr
 import random
-import os
 
-# Set up the client for Meta-Llama-3.1-8B model inference
-# Make sure you have set the HUGGING_FACE_HUB_TOKEN environment variable with your access token
-client = InferenceClient("meta-llama/Meta-Llama-3.1-8B", token=os.environ.get("HUGGING_FACE_HUB_TOKEN"))
+# Set up the client for FLAN-T5-XL model inference
+client = InferenceClient("google/flan-t5-xl")
 
 def generate_text(prompt, max_length=500):
     response = client.text_generation(prompt, max_new_tokens=max_length, temperature=0.7)
@@ -13,50 +11,42 @@ def generate_text(prompt, max_length=500):
 
 def debate_assistant(topic, stance):
     # Generate the main argument
-    argument_prompt = f"""Generate a comprehensive and well-structured argument for the following debate topic:
+    argument_prompt = f"""Generate a comprehensive argument for the following debate topic:
 Topic: {topic}
 Stance: {stance}
 
-Your response should include:
-1. A clear main claim
-2. Three strong supporting points with evidence
-3. A potential counterargument
-4. A rebuttal to that counterargument
-5. A compelling conclusion
+Include:
+1. Main claim
+2. Three supporting points
+3. Potential counterargument
+4. Rebuttal
+5. Conclusion"""
 
-Please ensure your argument is logical, well-reasoned, and persuasive."""
-
-    argument = generate_text(argument_prompt, max_length=800)
+    argument = generate_text(argument_prompt, max_length=600)
 
     # Generate a counterargument
-    counter_prompt = f"""Generate a strong counterargument for the following debate topic:
+    counter_prompt = f"""Generate a counterargument for the following debate topic:
 Topic: {topic}
 Original Stance: {stance}
 
-Your response should include:
-1. A clear counter-claim
-2. Three rebuttals to the original stance's likely arguments
-3. A new supporting point for your counter-claim
-4. A conclusion that reinforces your position
-
-Ensure your counterargument is logical, well-supported, and addresses key points of the original stance."""
+Include:
+1. Counter-claim
+2. Three rebuttals
+3. New supporting point
+4. Conclusion"""
 
-    counterargument = generate_text(counter_prompt, max_length=800)
+    counterargument = generate_text(counter_prompt, max_length=600)
 
-    # Generate analysis and additional content
-    analysis_prompt = f"""Provide a comprehensive analysis of the debate on the following topic:
+    # Generate analysis
+    analysis_prompt = f"""Analyze the debate on the following topic:
 Topic: {topic}
 
-Your analysis should include:
-1. Topic categorization (e.g., Politics, Economics, Technology, Ethics)
-2. Three key ethical considerations related to this topic
-3. Four thought-provoking discussion questions to further explore the topic
-4. A brief overview of the historical context or background of this debate
-5. Two potential real-world implications or consequences of this debate
-
-Please provide a balanced and insightful analysis."""
+Provide:
+1. Topic categorization
+2. Two ethical considerations
+3. Three discussion questions"""
 
-    analysis = generate_text(analysis_prompt, max_length=800)
+    analysis = generate_text(analysis_prompt, max_length=400)
 
     return f"""Argument ({stance}):
 {argument}
@@ -69,16 +59,16 @@ Analysis:
 
 def suggest_topic():
     topics = [
-        "Should artificial general intelligence development be regulated globally?",
-        "Is universal basic income a viable solution to technological unemployment?",
-        "Should human genetic enhancement be allowed for non-medical purposes?",
-        "Is space colonization necessary for the long-term survival of humanity?",
-        "Should social media platforms be held legally responsible for user-generated content?",
-        "Is nuclear fusion the ultimate solution to the global energy crisis?",
-        "Should autonomous weapons systems be completely banned in warfare?",
-        "Is a carbon tax the most effective way to combat climate change?",
-        "Should governments implement a four-day work week to improve work-life balance?",
-        "Is the current patent system hindering rather than promoting innovation?"
+        "Should artificial intelligence be regulated?",
+        "Is universal basic income a viable economic policy?",
+        "Should voting be mandatory?",
+        "Is space exploration a worthwhile investment?",
+        "Should gene editing in humans be allowed?",
+        "Is nuclear energy the solution to climate change?",
+        "Should social media platforms be held responsible for user content?",
+        "Is a four-day work week beneficial for society?",
+        "Should animal testing be banned?",
+        "Is globalization overall positive or negative for developing countries?"
     ]
     return random.choice(topics)
 
@@ -90,12 +80,12 @@ iface = gr.Interface(
         gr.Radio(["For", "Against"], label="Stance")
     ],
     outputs=gr.Textbox(label="Generated Debate Content"),
-    title="Advanced AI-powered Debate Assistant (Meta-Llama-3.1-8B)",
-    description="Enter a debate topic and choose a stance to generate comprehensive arguments, counterarguments, and analysis.",
+    title="AI-powered Debate Assistant (FLAN-T5-XL)",
+    description="Enter a debate topic and choose a stance to generate arguments, counterarguments, and analysis.",
     examples=[
-        ["Should artificial general intelligence development be regulated globally?", "For"],
-        ["Is universal basic income a viable solution to technological unemployment?", "Against"],
-        ["Should human genetic enhancement be allowed for non-medical purposes?", "For"]
+        ["Should artificial intelligence be regulated?", "For"],
+        ["Is universal basic income a viable economic policy?", "Against"],
+        ["Should gene editing in humans be allowed?", "For"]
     ]
 )