Update app.py

app.py

@@ -225,16 +225,17 @@ Avoid repeating the same points multiple times.
     return final_response
 
 def remove_non_ascii(text):
-    return unicodedata.normalize('NFKD', text).encode('ascii', 'ignore').decode('ascii')
+    return ''.join(c for c in text if ord(c) < 256)
 
 def generate_pdf_report_with_charts(summary: str, report_path: str, detailed_batches: List[str] = None):
     chart_dir = os.path.join(os.path.dirname(report_path), "charts")
     os.makedirs(chart_dir, exist_ok=True)
 
-    # Generate 3 charts
+    # Prepare data
     categories = ['Diagnostics', 'Medications', 'Missed', 'Inconsistencies', 'Follow-up']
     values = [4, 2, 3, 1, 5]
 
+    # Chart 1: Bar
     bar_chart_path = os.path.join(chart_dir, "bar_chart.png")
     plt.figure(figsize=(6, 4))
     plt.bar(categories, values)
@@ -243,6 +244,7 @@ def generate_pdf_report_with_charts(summary: str, report_path: str, detailed_bat
     plt.savefig(bar_chart_path)
     plt.close()
 
+    # Chart 2: Pie
     pie_chart_path = os.path.join(chart_dir, "pie_chart.png")
     plt.figure(figsize=(6, 6))
     plt.pie(values, labels=categories, autopct='%1.1f%%')
@@ -251,6 +253,7 @@ def generate_pdf_report_with_charts(summary: str, report_path: str, detailed_bat
     plt.savefig(pie_chart_path)
     plt.close()
 
+    # Chart 3: Line
     trend_chart_path = os.path.join(chart_dir, "trend_chart.png")
     plt.figure(figsize=(6, 4))
     plt.plot(categories, values, marker='o')
@@ -259,61 +262,66 @@ def generate_pdf_report_with_charts(summary: str, report_path: str, detailed_bat
     plt.savefig(trend_chart_path)
     plt.close()
 
+    # PDF init
     pdf_path = report_path.replace('.md', '.pdf')
     pdf = FPDF()
     pdf.set_auto_page_break(auto=True, margin=15)
 
-    # Title Page
+    # === Title Page ===
     pdf.add_page()
     pdf.set_font("Arial", 'B', 24)
-    pdf.cell(0, 20, "🩺 Final Medical Report", ln=True, align='C')
+    pdf.cell(0, 20, remove_non_ascii("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")
+    pdf.multi_cell(0, 10, remove_non_ascii(
+        "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
+    # === Summary Section ===
     pdf.add_page()
     pdf.set_font("Arial", 'B', 16)
-    pdf.cell(0, 10, "📄 Final Summary", ln=True)
+    pdf.cell(0, 10, remove_non_ascii("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.strip())
-        pdf.multi_cell(0, 8, txt=clean_line)
+        if clean_line:
+            pdf.multi_cell(0, 8, txt=clean_line)
 
-    # Charts Page
+    # === Charts Section ===
     pdf.add_page()
     pdf.set_font("Arial", 'B', 16)
-    pdf.cell(0, 10, "📊 Statistical Overview", ln=True)
+    pdf.cell(0, 10, remove_non_ascii("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.cell(0, 10, remove_non_ascii("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.cell(0, 10, remove_non_ascii("2. Issue Distribution"), ln=True)
     pdf.image(pie_chart_path, w=150)
     pdf.ln(5)
 
-    pdf.cell(0, 10, "3. Trend Analysis", ln=True)
+    pdf.cell(0, 10, remove_non_ascii("3. Trend Analysis"), ln=True)
     pdf.image(trend_chart_path, w=180)
 
-    # Detailed Batches
+    # === Detailed Tool Outputs ===
     if detailed_batches:
         pdf.add_page()
         pdf.set_font("Arial", 'B', 16)
-        pdf.cell(0, 10, "🔎 Detailed Tool Insights", ln=True)
+        pdf.cell(0, 10, remove_non_ascii("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.set_font("Arial", 'B', 13)
-            pdf.cell(0, 10, f"Tool Output #{idx + 1}", ln=True)
+            pdf.cell(0, 10, remove_non_ascii(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, 8, txt=line.strip())