Upload app.py

app.py

@@ -21,6 +21,10 @@ from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
 from reportlab.lib.utils import ImageReader
 from PIL import Image
+
+REPORTS_DIR = "reports"
+os.makedirs(REPORTS_DIR, exist_ok=True)
+
 # ===============================
 # 3️⃣ Groq Client
 # ===============================
@@ -167,11 +171,29 @@ Explain the trends in stress, mood, and sleep in simple, policymaker-friendly la
     explanation = response.choices[0].message.content
 
     # ---- PDF generation ----
-    PDF_DIR = "reports"
-    os.makedirs(PDF_DIR, exist_ok=True)
+    import time
+
+    timestamp = int(time.time())
+    pdf_path = f"{REPORTS_DIR}/weekly_report_user_{user_id}_{timestamp}.pdf"
 
-    pdf_path = f"{PDF_DIR}/weekly_report_user_{user_id}.pdf"
     c = canvas.Canvas(pdf_path, pagesize=letter)
+    width, height = letter
+
+    c.setFont("Helvetica-Bold", 14)
+    c.drawString(40, height - 40, "Weekly Mental Health Trend Report")
+
+    c.setFont("Helvetica", 11)
+    y = height - 80
+    for line in explanation.split("\n"):
+        c.drawString(40, y, line)
+        y -= 14
+        if y < 100:
+            c.showPage()
+            y = height - 40
+
+    c.showPage()
+    c.drawImage(ImageReader(chart_buf), 50, 200, width=500, height=400)
+    c.save()
 
     return explanation, chart_image, pdf_path
 
@@ -202,14 +224,19 @@ with gr.Blocks() as demo:
     with gr.Tab("Medical QA"):
         gr.Markdown("Ask questions about stress, mood, sleep, or general wellness.")
 
-        chatbot = gr.Chatbot()
+        chatbot = gr.Chatbot(label="Medical QA")
         msg = gr.Textbox(label="Your Question")
         clear = gr.Button("Clear Chat")
 
         def respond(message, history):
-            history = history or []
+            if history is None:
+                history = []
+
             answer = rag_answer(message)
+
+            # MUST be tuple format
             history.append((message, answer))
+
             return "", history
 
         msg.submit(respond, [msg, chatbot], [msg, chatbot])