Update app.py

app.py

@@ -1,8 +1,6 @@
-# app.py
 import os
 import gradio as gr
 
-
 from genesis.pipeline import (
     run_literature_review,
     run_molecule_lookup,
@@ -13,7 +11,6 @@ from genesis.pipeline import (
 from genesis.visualization import generate_pathway_graph, generate_funding_network
 from genesis.utils.pdf_export import export_report_to_pdf
 
-
 # Load custom CSS
 with open("genesis/static/style.css", "r") as f:
     custom_css = f.read()
@@ -32,36 +29,61 @@ Explore molecules, pathways, funding, literature, and images — with real-time
 def literature_tab(query):
     """Run literature review (PubMed, ChEMBL, BioPortal, UMLS)"""
     summary, citations = run_literature_review(query)
-    pdf_path = export_report_to_pdf("Literature Review", summary, citations, filename="literature_report.pdf")
-
+    pdf_path = export_report_to_pdf(
+        filename="literature_report.pdf",
+        title="Literature Review",
+        summary=summary,
+        citations=citations
+    )
     return summary, citations, pdf_path
 
+
 def molecule_tab(molecule_name):
     """Fetch molecule data from ChEMBL/NCBI & visualize"""
     mol_data, mol_img = run_molecule_lookup(molecule_name)
-    pdf_path = export_to_pdf(mol_data, None, filename="molecule_report.pdf")
+    pdf_path = export_report_to_pdf(
+        filename="molecule_report.pdf",
+        title="Molecule Report",
+        summary=mol_data
+    )
     return mol_data, mol_img, pdf_path
 
+
 def pathway_tab(pathway_name):
     """Run pathway analysis & generate graph"""
     analysis = run_pathway_analysis(pathway_name)
     graph_fig = generate_pathway_graph(pathway_name)
-    pdf_path = export_to_pdf(analysis, None, filename="pathway_report.pdf")
+    pdf_path = export_report_to_pdf(
+        filename="pathway_report.pdf",
+        title="Pathway Analysis",
+        summary=analysis
+    )
     return analysis, graph_fig, pdf_path
 
+
 def funding_tab(keyword):
     """Run funding analysis & generate investor graph"""
     funding_info = run_funding_analysis(keyword)
     funding_graph = generate_funding_network(keyword)
-    pdf_path = export_to_pdf(funding_info, None, filename="funding_report.pdf")
+    pdf_path = export_report_to_pdf(
+        filename="funding_report.pdf",
+        title="Funding Analysis",
+        summary=funding_info
+    )
     return funding_info, funding_graph, pdf_path
 
+
 def image_tab(image_file):
     """Run microscopy/image analysis"""
     analysis_text = run_image_analysis(image_file)
-    pdf_path = export_to_pdf(analysis_text, None, filename="image_report.pdf")
+    pdf_path = export_report_to_pdf(
+        filename="image_report.pdf",
+        title="Image Analysis",
+        summary=analysis_text
+    )
     return analysis_text, pdf_path
 
+
 # =========================
 # GRADIO UI
 # =========================