Upload app.py

app.py

@@ -167,28 +167,13 @@ Explain the trends in stress, mood, and sleep in simple, policymaker-friendly la
     explanation = response.choices[0].message.content
 
     # ---- PDF generation ----
-    pdf_buf = BytesIO()
-    c = canvas.Canvas(pdf_buf, pagesize=letter)
-    width, height = letter
+    PDF_DIR = "reports"
+    os.makedirs(PDF_DIR, exist_ok=True)
 
-    c.setFont("Helvetica-Bold", 14)
-    c.drawString(40, height - 40, "Weekly Mental Health Trend Report")
+    pdf_path = f"{PDF_DIR}/weekly_report_user_{user_id}.pdf"
+    c = canvas.Canvas(pdf_path, pagesize=letter)
 
-    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()
-    pdf_buf.seek(0)
-
-    return explanation, chart_image, pdf_buf
+    return explanation, chart_image, pdf_path
 
 # ===============================
 # 🔟 Gradio interface
@@ -216,13 +201,17 @@ 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(type="messages")
         msg = gr.Textbox(label="Your Question")
         clear = gr.Button("Clear Chat")
-        def respond(message,history):
+        def respond(message, history):
             answer = rag_answer(message)
-            history.append((message,answer))
-            return "",history
+
+            history = history or []
+            history.append({"role": "user", "content": message})
+            history.append({"role": "assistant", "content": answer})
+
+            return "", history
         msg.submit(respond,[msg,chatbot],[msg,chatbot])
         clear.click(lambda: None,None,chatbot,queue=False)