Upload folder using huggingface_hub

home.py

@@ -157,6 +157,29 @@ def flatten_reports(reports: List[Dict[str, Any]]) -> pd.DataFrame:
         ]
     return pd.DataFrame(rows)
 
+import tempfile
+import os
+
+def download_final_report(report_id: str) -> str | None:
+    """
+    Fetch the PDF for a given final_report and return a temporary file path.
+    """
+    if not report_id:
+        return None
+
+    db = get_db()
+    pdf_bytes = db.get_final_report_pdf(report_id)
+    if not pdf_bytes:
+        return None
+
+    # Write to a temporary file
+    tmpdir = tempfile.mkdtemp()
+    file_path = os.path.join(tmpdir, f"{report_id}.pdf")
+    with open(file_path, "wb") as f:
+        f.write(pdf_bytes)
+
+    return file_path
+
 
 def trends_index(trends: List[Dict[str, Any]]) -> List[str]:
     names = sorted({t.get("test_name", "") for t in trends if t.get("test_name")})
@@ -435,7 +458,7 @@ with gr.Blocks(
                         choices=[], label="Select Test", interactive=True
                     )
                     trend_plot = gr.Plot(label="Trend Chart")
-                with gr.Tab("✅ Final Reports"):
+                with gr.TabItem("✅ Final Reports"):
                     final_df = gr.DataFrame(
                         headers=[
                             "final_report_id",
@@ -447,6 +470,13 @@ with gr.Blocks(
                         wrap=True,
                         interactive=False,
                     )
+
+                    with gr.Row():
+                        report_id_in = gr.Textbox(label="Report ID", placeholder="Paste Report ID to download")
+                        download_btn = gr.Button("⬇️ Download PDF", variant="primary")
+                        download_file = gr.File(label="Your PDF", type="filepath", file_types=[".pdf"])
+
+                    gr.Markdown("Tip: Select another patient to refresh final reports.")
                 with gr.Tab("📄 Upload History"):
                     gr.Markdown("## here is your upload history")
 

modules/db.py

@@ -6,7 +6,7 @@ from bson import ObjectId
 from dotenv import load_dotenv
 
 from modules.models import StandardizedReport
-
+import gridfs
 
 class SheamiDB:
     def __init__(self, uri: str, db_name: str = "sheami"):
@@ -24,6 +24,7 @@ class SheamiDB:
         self.trends = self.db["trends"]
         self.final_reports = self.db["final_reports"]
         self.run_stats = self.db["run_stats"]
+        self.fs = gridfs.GridFS(self.db)
 
     # ---------------------------
     # USER FUNCTIONS
@@ -372,3 +373,49 @@ class SheamiDB:
 
         # print("updated/inserted", updated, "trends")
         return updated
+
+   # ---------------------------
+    # FINAL REPORT FUNCTIONS
+    # ---------------------------
+    def add_final_report_v2(
+        self,
+        patient_id: str,
+        summary: str,
+        recommendations: list,
+        trend_snapshots: list,
+        pdf_bytes: bytes = None,
+        file_name: str = None,
+    ) -> str:
+        """
+        Insert a final report.
+        If pdf_bytes is provided, stores it in GridFS and saves file_id in metadata.
+        """
+        pdf_file_id = None
+        if pdf_bytes:
+            pdf_file_id = self.fs.put(
+                pdf_bytes,
+                filename=file_name or f"final_report_{patient_id}.pdf",
+                patient_id=ObjectId(patient_id),
+                uploaded_at=datetime.now(timezone.utc),
+            )
+
+        final_report = {
+            "patient_id": ObjectId(patient_id),
+            "generated_at": datetime.now(timezone.utc),
+            "summary": summary,
+            "recommendations": recommendations,
+            "trend_snapshots": trend_snapshots,
+            "pdf_file_id": pdf_file_id,  # Reference to GridFS file
+        }
+
+        result = self.final_reports.insert_one(final_report)
+        return str(result.inserted_id)
+
+    def get_final_report_pdf(self, report_id: str) -> bytes | None:
+        """
+        Fetch the PDF bytes for a given final_report (if stored).
+        """
+        doc = self.final_reports.find_one({"_id": ObjectId(report_id)})
+        if not doc or not doc.get("pdf_file_id"):
+            return None
+        return self.fs.get(doc["pdf_file_id"]).read()