Update app.py

app.py

@@ -12,22 +12,22 @@ def generate_argument(topic, stance):
     prompt = f"""Generate a compelling argument for the following debate topic.
 Topic: {topic}
 Stance: {stance}
-Argument:"""
-    response = generate_text(prompt)
-    return response.split("Argument:")[1].strip()
+
+Your response should be a well-structured argument supporting the given stance on the topic."""
+    return generate_text(prompt)
 
 def generate_counterargument(topic, original_argument):
     prompt = f"""Generate a strong counterargument for the following debate topic and argument.
 Topic: {topic}
 Original Argument: {original_argument}
-Counterargument:"""
-    response = generate_text(prompt)
-    return response.split("Counterargument:")[1].strip()
+
+Your response should be a well-structured counterargument addressing the points made in the original argument."""
+    return generate_text(prompt)
 
 def debate_assistant(topic, stance):
     argument = generate_argument(topic, stance)
     counterargument = generate_counterargument(topic, argument)
-    return f"Argument: {argument}\n\nCounterargument: {counterargument}"
+    return f"Argument:\n{argument}\n\nCounterargument:\n{counterargument}"
 
 # Create the Gradio interface
 iface = gr.Interface(