Update app.py

app.py

@@ -231,31 +231,56 @@ 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)
 
-    chart_path = os.path.join(chart_dir, "summary_chart.png")
+    # Bar chart
+    bar_chart_path = os.path.join(chart_dir, "summary_chart.png")
     categories = ['Diagnostics', 'Medications', 'Missed', 'Inconsistencies', 'Follow-up']
     values = [4, 2, 3, 1, 5]
     plt.figure(figsize=(6, 4))
     plt.bar(categories, values)
     plt.title('Clinical Issues Overview')
     plt.tight_layout()
-    plt.savefig(chart_path)
+    plt.savefig(bar_chart_path)
     plt.close()
 
+    # Pie chart
+    pie_chart_path = os.path.join(chart_dir, "summary_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.tight_layout()
+    plt.savefig(pie_chart_path)
+    plt.close()
+
+    # Line graph
+    line_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.tight_layout()
+    plt.savefig(line_chart_path)
+    plt.close()
+
+    # PDF Creation
     pdf_path = report_path.replace('.md', '.pdf')
     pdf = FPDF()
     pdf.add_page()
     pdf.set_font("Arial", size=12)
     pdf.multi_cell(0, 10, txt="Final Medical Report", align="C")
     pdf.ln(5)
-
     for line in summary.split("\n"):
         clean_line = remove_non_ascii(line)
         pdf.multi_cell(0, 10, txt=clean_line)
 
     pdf.ln(10)
-    pdf.image(chart_path, w=150)
+    pdf.image(bar_chart_path, w=150)
+    pdf.ln(5)
+    pdf.image(pie_chart_path, w=150)
+    pdf.ln(5)
+    pdf.image(line_chart_path, w=150)
 
-    # Add detailed outputs per batch/tool
+    # Detailed batch results
     if detailed_batches:
         pdf.add_page()
         pdf.set_font("Arial", size=12)
@@ -264,9 +289,9 @@ def generate_pdf_report_with_charts(summary: str, report_path: str, detailed_bat
         for idx, detail in enumerate(detailed_batches):
             pdf.multi_cell(0, 10, txt=f"Tool Result #{idx + 1}", align="L")
             for line in remove_non_ascii(detail).split("\n"):
-
                 pdf.multi_cell(0, 10, txt=line)
             pdf.ln(5)
+
     pdf.output(pdf_path)
     return pdf_path