Upload app.py with huggingface_hub

app.py

@@ -617,9 +617,31 @@ def build_app():
             with gr.Tab("⚖️ Court Brief"):
                 court_md = gr.Markdown(label="Court Brief (FRE 702)")
 
-        # Wire up the analysis
+            with gr.Tab("📥 Export"):
+                gr.Markdown("""### Export Forensic Report
+Export the complete analysis in your preferred format. Reports are professionally formatted using **Qwen2.5-72B-Instruct** when available.
+                """)
+                with gr.Row():
+                    export_pdf_btn = gr.Button("📄 Export PDF", variant="primary")
+                    export_docx_btn = gr.Button("📝 Export DOCX", variant="primary")
+                    export_txt_btn = gr.Button("📃 Export TXT", variant="secondary")
+                    export_md_btn = gr.Button("📋 Export Markdown", variant="secondary")
+                export_file = gr.File(label="Download Report", visible=True)
+                export_status = gr.Markdown("")
+
+        # ── Hidden state to store report data for exports ────────────
+        report_state = gr.State("")
+        court_state = gr.State("")
+        tree_state = gr.State("")
+
+        # Wire up the analysis — also store report data in state
+        def analyze_and_store(img):
+            results = analyze_image(img)
+            # results[4] = report_md, results[5] = tree_md, results[6] = court_md
+            return list(results) + [results[4], results[6], results[5]]
+
         analyze_btn.click(
-            fn=analyze_image,
+            fn=analyze_and_store,
             inputs=[image_input],
             outputs=[
                 verdict_html,
@@ -635,9 +657,52 @@ def build_app():
                 benford_plot,
                 metadata_table,
                 agent_details_md,
+                report_state,
+                court_state,
+                tree_state,
             ],
         )
 
+        # ── Export handlers ───────────────────────────────────────────
+        from export import export_pdf, export_docx, export_txt, export_md
+
+        def do_export_pdf(report, court, tree):
+            if not report: return None, "⚠️ Run analysis first"
+            try:
+                path = export_pdf(report, court, tree)
+                return path, "✅ PDF exported successfully"
+            except Exception as e:
+                return None, f"❌ Export failed: {e}"
+
+        def do_export_docx(report, court, tree):
+            if not report: return None, "⚠️ Run analysis first"
+            try:
+                path = export_docx(report, court, tree)
+                return path, "✅ DOCX exported successfully"
+            except Exception as e:
+                return None, f"❌ Export failed: {e}"
+
+        def do_export_txt(report, court, tree):
+            if not report: return None, "⚠️ Run analysis first"
+            try:
+                path = export_txt(report, court, tree)
+                return path, "✅ TXT exported successfully"
+            except Exception as e:
+                return None, f"❌ Export failed: {e}"
+
+        def do_export_md(report, court, tree):
+            if not report: return None, "⚠️ Run analysis first"
+            try:
+                path = export_md(report, court, tree)
+                return path, "✅ Markdown exported successfully"
+            except Exception as e:
+                return None, f"❌ Export failed: {e}"
+
+        export_pdf_btn.click(fn=do_export_pdf, inputs=[report_state, court_state, tree_state], outputs=[export_file, export_status])
+        export_docx_btn.click(fn=do_export_docx, inputs=[report_state, court_state, tree_state], outputs=[export_file, export_status])
+        export_txt_btn.click(fn=do_export_txt, inputs=[report_state, court_state, tree_state], outputs=[export_file, export_status])
+        export_md_btn.click(fn=do_export_md, inputs=[report_state, court_state, tree_state], outputs=[export_file, export_status])
+
         # Footer
         gr.HTML("""
         <div style="text-align:center; padding:20px; color:#aaa; font-size:0.8em; border-top:1px solid #eee; margin-top:20px;">