Upload folder using huggingface_hub

home.py

@@ -294,8 +294,8 @@ def format_final_reports(finals: List[Dict[str, Any]]) -> pd.DataFrame:
             {
                 "final_report_id": str(fr.get("_id", "")),
                 "generated_at": _fmt(fr.get("generated_at")),
-                "summary": fr.get("summary", ""),
-                "recommendations": "; ".join(fr.get("recommendations", [])),
+                # "summary": fr.get("summary", ""),
+                # "recommendations": "; ".join(fr.get("recommendations", [])),
             }
         )
     return pd.DataFrame(
@@ -304,8 +304,8 @@ def format_final_reports(finals: List[Dict[str, Any]]) -> pd.DataFrame:
             {
                 "final_report_id": "",
                 "generated_at": "",
-                "summary": "",
-                "recommendations": "",
+                # "summary": "",
+                # "recommendations": "",
             }
         ]
     )
@@ -465,8 +465,8 @@ with gr.Blocks(
                         headers=[
                             "final_report_id",
                             "generated_at",
-                            "summary",
-                            "recommendations",
+                            # "summary",
+                            # "recommendations",
                         ],
                         row_count=(0, "dynamic"),
                         wrap=True,
@@ -474,18 +474,20 @@ with gr.Blocks(
                     )
 
                     with Modal(visible=False) as final_report_download_modal:
-                        gr.Markdown("# download report")
                         with gr.Row():
+                            gr.Column()
                             report_id_in = gr.Textbox(
                                 label="Report ID",
                                 placeholder="Paste Report ID to download",
+                                visible=False,
                             )
-                            download_btn = gr.Button(
-                                "⬇️ Download PDF", variant="primary"
-                            )
-                            download_file = gr.File(
-                                label="Your PDF", type="binary", file_types=[".pdf"], file_count="single"
+
+                            download_btn = gr.DownloadButton(
+                                "⬇️ Download PDF", variant="primary", scale=0
                             )
+                            gr.Column()
+
+                        html_report_component = gr.HTML()
 
                     # Function triggered when a row is selected
                     def on_row_select(df_row):
@@ -493,23 +495,34 @@ with gr.Blocks(
                         if df_row is None:
                             return gr.update(visible=False), None
                         report_id = df_row["final_report_id"][0]
+                        report_file_name = download_final_report(report_id)
+                        report_html_content = get_db().get_final_report_html(report_id)
                         # Return modal visibility + the file to bind to download button
-                        return gr.update(visible=True), str(report_id)
+                        return (
+                            gr.update(visible=True),
+                            str(report_id),
+                            report_file_name,
+                            report_html_content,
+                        )  # final_report_download_modal, report_id_in, download_btn, html_report
 
                     final_df.select(close_side_bar, outputs=[sheami_sidebar]).then(
                         on_row_select,
                         inputs=[final_df],
-                        outputs=[final_report_download_modal, report_id_in],
+                        outputs=[
+                            final_report_download_modal,
+                            report_id_in,
+                            download_btn,
+                            html_report_component,
+                        ],
                     )
                     download_btn.click(
-                            fn=download_final_report,
-                            inputs=[report_id_in], 
-                            outputs=download_file
-                        )
+                        fn=download_final_report,
+                        inputs=[report_id_in],
+                        outputs=download_btn,
+                    )
 
-                    gr.Markdown("Tip: Select another patient to refresh final reports.")
                 with gr.Tab("📄 Upload History"):
-                    gr.Markdown("## here is your upload history")
+                    gr.Markdown("### here is your upload history")
 
     # ---- Events ----
     def on_load(email):
@@ -531,7 +544,7 @@ with gr.Blocks(
             selected_patient_info,
             reports_df_val,
             gr.update(choices=test_names),
-            format_final_reports(finals),
+            gr.update(headers=["final_report_id","generated_at"],value=format_final_reports(finals)),
         )
 
     patient_list.change(

modules/db.py

@@ -435,4 +435,13 @@ class SheamiDB:
         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()
+        return self.fs.get(doc["pdf_file_id"]).read()
+    
+    def get_final_report_html(self, report_id: str) -> bytes | None:
+        """
+        Fetch the HTML for a given final_report (if stored).
+        """
+        doc = self.final_reports.find_one({"_id": ObjectId(report_id)})
+        if not doc or not doc.get("summary"):
+            return "<html></html>"#empty tag
+        return doc.get("summary")