Update app.py

app.py

@@ -231,10 +231,11 @@ def generate_pdf_report_with_charts(summary: str, report_path: str, detailed_bat
     chart_dir = os.path.join(os.path.dirname(report_path), "charts")
     os.makedirs(chart_dir, exist_ok=True)
 
-    # Bar chart
-    bar_chart_path = os.path.join(chart_dir, "summary_chart.png")
+    # Generate 3 charts
     categories = ['Diagnostics', 'Medications', 'Missed', 'Inconsistencies', 'Follow-up']
     values = [4, 2, 3, 1, 5]
+
+    bar_chart_path = os.path.join(chart_dir, "bar_chart.png")
     plt.figure(figsize=(6, 4))
     plt.bar(categories, values)
     plt.title('Clinical Issues Overview')
@@ -242,55 +243,81 @@ def generate_pdf_report_with_charts(summary: str, report_path: str, detailed_bat
     plt.savefig(bar_chart_path)
     plt.close()
 
-    # Pie chart
-    pie_chart_path = os.path.join(chart_dir, "summary_pie_chart.png")
+    pie_chart_path = os.path.join(chart_dir, "pie_chart.png")
     plt.figure(figsize=(6, 6))
-    plt.pie(values, labels=categories, autopct='%1.1f%%', startangle=140)
-    plt.title('Distribution of Clinical Issues')
+    plt.pie(values, labels=categories, autopct='%1.1f%%')
+    plt.title('Issue Distribution')
     plt.tight_layout()
     plt.savefig(pie_chart_path)
     plt.close()
 
-    # Line graph
-    line_chart_path = os.path.join(chart_dir, "trend_chart.png")
+    trend_chart_path = os.path.join(chart_dir, "trend_chart.png")
     plt.figure(figsize=(6, 4))
     plt.plot(categories, values, marker='o')
-    plt.title('Clinical Concern Trend')
-    plt.xlabel('Category')
-    plt.ylabel('Concern Level')
+    plt.title('Trend Analysis')
     plt.tight_layout()
-    plt.savefig(line_chart_path)
+    plt.savefig(trend_chart_path)
     plt.close()
 
-    # PDF Creation
     pdf_path = report_path.replace('.md', '.pdf')
     pdf = FPDF()
+    pdf.set_auto_page_break(auto=True, margin=15)
+
+    # Title Page
     pdf.add_page()
-    pdf.set_font("Arial", size=12)
-    pdf.multi_cell(0, 10, txt="Final Medical Report", align="C")
+    pdf.set_font("Arial", 'B', 24)
+    pdf.cell(0, 20, "🩺 Final Medical Report", ln=True, align='C')
+    pdf.set_font("Arial", '', 14)
+    pdf.cell(0, 10, datetime.now().strftime("Generated on %B %d, %Y at %H:%M"), ln=True, align='C')
+    pdf.ln(20)
+    pdf.set_font("Arial", 'I', 12)
+    pdf.multi_cell(0, 10, "This report contains a professional summary of clinical observations, potential inconsistencies, and follow-up recommendations based on the uploaded medical document.", align="C")
+
+    # Summary Page
+    pdf.add_page()
+    pdf.set_font("Arial", 'B', 16)
+    pdf.cell(0, 10, "📄 Final Summary", ln=True)
+    pdf.set_draw_color(200, 200, 200)
+    pdf.line(10, pdf.get_y(), 200, pdf.get_y())
     pdf.ln(5)
+    pdf.set_font("Arial", '', 12)
     for line in summary.split("\n"):
-        clean_line = remove_non_ascii(line)
-        pdf.multi_cell(0, 10, txt=clean_line)
+        clean_line = remove_non_ascii(line.strip())
+        pdf.multi_cell(0, 8, txt=clean_line)
 
-    pdf.ln(10)
-    pdf.image(bar_chart_path, w=150)
+    # Charts Page
+    pdf.add_page()
+    pdf.set_font("Arial", 'B', 16)
+    pdf.cell(0, 10, "📊 Statistical Overview", ln=True)
+    pdf.line(10, pdf.get_y(), 200, pdf.get_y())
     pdf.ln(5)
+    pdf.set_font("Arial", 'B', 12)
+    pdf.cell(0, 10, "1. Clinical Issues Overview", ln=True)
+    pdf.image(bar_chart_path, w=180)
+    pdf.ln(5)
+
+    pdf.cell(0, 10, "2. Issue Distribution", ln=True)
     pdf.image(pie_chart_path, w=150)
     pdf.ln(5)
-    pdf.image(line_chart_path, w=150)
 
-    # Detailed batch results
+    pdf.cell(0, 10, "3. Trend Analysis", ln=True)
+    pdf.image(trend_chart_path, w=180)
+
+    # Detailed Batches
     if detailed_batches:
         pdf.add_page()
-        pdf.set_font("Arial", size=12)
-        pdf.multi_cell(0, 10, txt="Detailed Analysis Per Tool", align="L")
+        pdf.set_font("Arial", 'B', 16)
+        pdf.cell(0, 10, "🔎 Detailed Tool Insights", ln=True)
+        pdf.line(10, pdf.get_y(), 200, pdf.get_y())
         pdf.ln(5)
+
         for idx, detail in enumerate(detailed_batches):
-            pdf.multi_cell(0, 10, txt=f"Tool Result #{idx + 1}", align="L")
+            pdf.set_font("Arial", 'B', 13)
+            pdf.cell(0, 10, f"Tool Output #{idx + 1}", ln=True)
+            pdf.set_font("Arial", '', 11)
             for line in remove_non_ascii(detail).split("\n"):
-                pdf.multi_cell(0, 10, txt=line)
-            pdf.ln(5)
+                pdf.multi_cell(0, 8, txt=line.strip())
+            pdf.ln(3)
 
     pdf.output(pdf_path)
     return pdf_path