Update app.py

app.py

@@ -1,114 +1,84 @@
 from huggingface_hub import InferenceClient
 import gradio as gr
 import random
+import os
 
-# Set up the client for Mistral model inference
-client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.3")
+# 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"))
 
-def generate_text(prompt):
-    response = client.text_generation(prompt, max_new_tokens=300, temperature=0.7)
+def generate_text(prompt, max_length=500):
+    response = client.text_generation(prompt, max_new_tokens=max_length, temperature=0.7)
     return response
 
-def categorize_topic(topic):
-    prompt = f"""Categorize the following debate topic into one of these categories: Politics, Economics, Technology, Environment, Social Issues, Ethics, or Other.
-
-Topic: {topic}
-
-Category:"""
-    return generate_text(prompt).strip()
-
-def generate_structured_argument(topic, stance):
-    prompt = f"""Generate a structured argument for the following debate topic.
+def debate_assistant(topic, stance):
+    # Generate the main argument
+    argument_prompt = f"""Generate a comprehensive and well-structured argument for the following debate topic:
 Topic: {topic}
 Stance: {stance}
 
-Your response should follow this structure:
-1. Main claim
-2. Supporting point 1
-3. Supporting point 2
-4. Potential counterargument
-5. Rebuttal to counterargument
-6. Conclusion
-
-Ensure each point is concise and well-reasoned."""
-    return generate_text(prompt)
+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
 
-def generate_structured_counterargument(topic, original_argument):
-    prompt = f"""Generate a structured counterargument for the following debate topic and argument.
-Topic: {topic}
-Original Argument: {original_argument}
+Please ensure your argument is logical, well-reasoned, and persuasive."""
 
-Your response should follow this structure:
-1. Main counter-claim
-2. Rebuttal to original supporting point 1
-3. Rebuttal to original supporting point 2
-4. New supporting point for counter-claim
-5. Addressing the original rebuttal
-6. Conclusion
+    argument = generate_text(argument_prompt, max_length=800)
 
-Ensure each point directly addresses the original argument and provides new insights."""
-    return generate_text(prompt)
-
-def assess_argument_quality(argument):
-    prompt = f"""Assess the quality of the following argument on a scale of 1 to 10, where 1 is very poor and 10 is excellent. Consider factors such as logical coherence, evidence use, and persuasiveness.
+    # Generate a counterargument
+    counter_prompt = f"""Generate a strong counterargument for the following debate topic:
+Topic: {topic}
+Original Stance: {stance}
 
-Argument: {argument}
+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
 
-Provide your assessment in this format:
-Score: [1-10]
-Reasoning: [Brief explanation of the score]"""
-    return generate_text(prompt)
+Ensure your counterargument is logical, well-supported, and addresses key points of the original stance."""
 
-def generate_sources(topic):
-    prompt = f"""Provide a list of 3-5 credible sources that could be used to research the following debate topic:
+    counterargument = generate_text(counter_prompt, max_length=800)
 
+    # Generate analysis and additional content
+    analysis_prompt = f"""Provide a comprehensive analysis of the debate on the following topic:
 Topic: {topic}
 
-For each source, provide:
-1. Title
-2. Author or Organization
-3. Brief description (1-2 sentences)
+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
 
-Format the response as a numbered list."""
-    return generate_text(prompt)
+Please provide a balanced and insightful analysis."""
 
-def debate_assistant(topic, stance):
-    category = categorize_topic(topic)
-    argument = generate_structured_argument(topic, stance)
-    counterargument = generate_structured_counterargument(topic, argument)
-    arg_quality = assess_argument_quality(argument)
-    counter_quality = assess_argument_quality(counterargument)
-    sources = generate_sources(topic)
-    
-    return f"""Topic Category: {category}
-
-Argument:
-{argument}
+    analysis = generate_text(analysis_prompt, max_length=800)
 
-Argument Quality Assessment:
-{arg_quality}
+    return f"""Argument ({stance}):
+{argument}
 
 Counterargument:
 {counterargument}
 
-Counterargument Quality Assessment:
-{counter_quality}
-
-Suggested Sources for Further Research:
-{sources}"""
+Analysis:
+{analysis}"""
 
 def suggest_topic():
     topics = [
-        "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?"
+        "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?"
     ]
     return random.choice(topics)
 
@@ -120,12 +90,12 @@ iface = gr.Interface(
         gr.Radio(["For", "Against"], label="Stance")
     ],
     outputs=gr.Textbox(label="Generated Debate Content"),
-    title="Advanced AI-powered Debate Assistant",
-    description="Enter a debate topic and choose a stance to generate structured arguments, counterarguments, quality assessments, and suggested sources.",
+    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.",
     examples=[
-        ["Should artificial intelligence be regulated?", "For"],
-        ["Is universal basic income a viable economic policy?", "Against"],
-        ["Should gene editing in humans be allowed?", "For"]
+        ["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"]
     ]
 )