Update app.py

app.py

@@ -1,45 +1,132 @@
 from huggingface_hub import InferenceClient
 import gradio as gr
+import random
 
 # Set up the client for Mistral model inference
 client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.3")
 
 def generate_text(prompt):
-    response = client.text_generation(prompt, max_new_tokens=200, temperature=0.7)
+    response = client.text_generation(prompt, max_new_tokens=300, temperature=0.7)
     return response
 
-def generate_argument(topic, stance):
-    prompt = f"""Generate a compelling argument for the following debate topic.
+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.
 Topic: {topic}
 Stance: {stance}
 
-Your response should be a well-structured argument supporting the given stance on the topic."""
+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)
 
-def generate_counterargument(topic, original_argument):
-    prompt = f"""Generate a strong counterargument for the following debate topic and argument.
+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}
 
-Your response should be a well-structured counterargument addressing the points made in the original argument."""
+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
+
+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.
+
+Argument: {argument}
+
+Provide your assessment in this format:
+Score: [1-10]
+Reasoning: [Brief explanation of the score]"""
+    return generate_text(prompt)
+
+def generate_sources(topic):
+    prompt = f"""Provide a list of 3-5 credible sources that could be used to research the following debate topic:
+
+Topic: {topic}
+
+For each source, provide:
+1. Title
+2. Author or Organization
+3. Brief description (1-2 sentences)
+
+Format the response as a numbered list."""
     return generate_text(prompt)
 
 def debate_assistant(topic, stance):
-    argument = generate_argument(topic, stance)
-    counterargument = generate_counterargument(topic, argument)
-    return f"Argument:\n{argument}\n\nCounterargument:\n{counterargument}"
+    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}
+
+Argument Quality Assessment:
+{arg_quality}
+
+Counterargument:
+{counterargument}
+
+Counterargument Quality Assessment:
+{counter_quality}
+
+Suggested Sources for Further Research:
+{sources}"""
+
+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?"
+    ]
+    return random.choice(topics)
 
 # Create the Gradio interface
 iface = gr.Interface(
     fn=debate_assistant,
     inputs=[
-        gr.Textbox(label="Debate Topic"),
+        gr.Textbox(label="Debate Topic", placeholder="Enter a topic or click 'Suggest Topic'"),
         gr.Radio(["For", "Against"], label="Stance")
     ],
-    outputs=gr.Textbox(label="Generated Debate Arguments"),
-    title="AI-powered Debate Assistant (Mistral-7B-Instruct-v0.3)",
-    description="Enter a debate topic and choose a stance to generate arguments and counterarguments using Mistral-7B-Instruct-v0.3."
+    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.",
+    examples=[
+        ["Should artificial intelligence be regulated?", "For"],
+        ["Is universal basic income a viable economic policy?", "Against"],
+        ["Should gene editing in humans be allowed?", "For"]
+    ]
 )
 
-# Launch the interface
 iface.launch()