Update app.py

app.py

@@ -11,6 +11,8 @@ import pandas as pd
 import pdfplumber
 import gradio as gr
 import torch
+from reportlab.lib.pagesizes import letter
+from reportlab.pdfgen import canvas
 
 # === Configuration ===
 persistent_dir = "/data/hf_cache"
@@ -59,6 +61,21 @@ def remove_duplicate_paragraphs(text: str) -> str:
             seen.add(clean_p)
     return "\n\n".join(unique_paragraphs)
 
+def save_report_as_pdf(text: str, path: str):
+    c = canvas.Canvas(path, pagesize=letter)
+    width, height = letter
+    y = height - 50
+
+    lines = text.split('\n')
+    for line in lines:
+        if y < 50:
+            c.showPage()
+            y = height - 50
+        c.drawString(50, y, line.strip())
+        y -= 15
+
+    c.save()
+
 def extract_text_from_excel(path: str) -> str:
     all_text = []
     xls = pd.ExcelFile(path)
@@ -249,12 +266,11 @@ def process_report(agent, file, messages: List[Dict[str, str]]) -> Tuple[List[Di
             return messages, None
 
         summary = generate_final_summary(agent, "\n\n".join(valid))
-        report_path = os.path.join(report_dir, f"report_{datetime.now().strftime('%Y%m%d_%H%M%S')}.md")
-        with open(report_path, 'w', encoding='utf-8') as f:
-            f.write(f"# 🧠 Final Medical Report\n\n{summary}")
 
-        messages.append({"role": "assistant", "content": f"📊 Final Report:\n\n{summary}"})
-        messages.append({"role": "assistant", "content": f"✅ Report saved: {os.path.basename(report_path)}"})
+        report_path = os.path.join(report_dir, f"report_{datetime.now().strftime('%Y%m%d_%H%M%S')}.pdf")
+        save_report_as_pdf(summary, report_path)
+
+        messages.append({"role": "assistant", "content": "✅ Report generated successfully. Click below to download it."})
         return messages, report_path
 
     except Exception as e:
@@ -277,7 +293,7 @@ def create_ui(agent):
             chatbot = gr.Chatbot(label="🧠 CPS Assistant", height=480, type="messages")
             upload = gr.File(label="📂 Upload Medical File", file_types=[".xlsx", ".csv", ".pdf"])
             analyze = gr.Button("🧠 Analyze")
-            download = gr.File(label="📥 Download Report", visible=False, interactive=False)
+            download = gr.File(label="📥 Download Report", visible=False, interactive=True)
 
         state = gr.State(value=[])
 
@@ -293,4 +309,4 @@ def create_ui(agent):
 if __name__ == "__main__":
     agent = init_agent()
     ui = create_ui(agent)
-    ui.launch(server_name="0.0.0.0", server_port=7860, allowed_paths=["/data/hf_cache/reports"], share=False)
+    ui.launch(server_name="0.0.0.0", server_port=7860, allowed_paths=["/data/hf_cache/reports"], share=False)