Update app.py

app.py

@@ -3,6 +3,7 @@ import requests
 import json
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
+from textwrap import wrap
 
 # Hugging Face Inference API details
 HF_API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3"
@@ -10,7 +11,6 @@ HF_API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-I
 import os
 HF_API_KEY = os.getenv("HF_API_KEY")  # Read from environment variable
 
-
 HEADERS = {"Authorization": f"Bearer {HF_API_KEY}"}
 
 # Function to call Hugging Face API
@@ -38,20 +38,38 @@ def conduct_debate(topic):
 
     return response_1, response_2, conclusion
 
-# Function to generate a PDF
+# Function to generate a properly formatted PDF
 def generate_pdf(topic, response_1, response_2, conclusion):
     pdf_filename = "debate_result.pdf"
-    
     c = canvas.Canvas(pdf_filename, pagesize=letter)
-    c.drawString(100, 750, f"Debate Topic: {topic}")
-    c.drawString(100, 730, "Argument in Favor:")
-    c.drawString(100, 710, response_1[:1000])  # Limiting text for PDF readability
-    c.drawString(100, 690, "Argument Against:")
-    c.drawString(100, 670, response_2[:1000])
-    c.drawString(100, 650, "Judge's Conclusion:")
-    c.drawString(100, 630, conclusion[:1000])
+    c.setFont("Helvetica", 12)
+    y_position = 750
+    
+    def draw_wrapped_text(text, x, y, max_width=450):
+        lines = wrap(text, width=80)  # Wrap text for better formatting
+        for line in lines:
+            c.drawString(x, y, line)
+            y -= 20
+        return y
+    
+    c.drawString(100, y_position, f"Debate Topic: {topic}")
+    y_position -= 30
+    
+    c.drawString(100, y_position, "Argument in Favor:")
+    y_position -= 20
+    y_position = draw_wrapped_text(response_1, 100, y_position)
+    
+    y_position -= 30  # Space between sections
+    c.drawString(100, y_position, "Argument Against:")
+    y_position -= 20
+    y_position = draw_wrapped_text(response_2, 100, y_position)
+    
+    y_position -= 30  # Space between sections
+    c.drawString(100, y_position, "Judge's Conclusion:")
+    y_position -= 20
+    y_position = draw_wrapped_text(conclusion, 100, y_position)
+    
     c.save()
-
     return pdf_filename
 
 # Gradio UI