Update app.py

app.py

@@ -4,6 +4,8 @@ from torchvision import transforms
 from PIL import Image
 import joblib
 import base64
+import io
+from contextlib import redirect_stdout
 
 # Load model and metadata
 model = torch.jit.load("resnet_grocery_model_scripted.pt", map_location="cpu")
@@ -36,6 +38,40 @@ def classify_and_track(files, budget):
     total_cost = 0
     receipt_lines = []
 
+    # Initialize LaTeX document for PDF
+    latex_lines = [
+        r"\documentclass[a4paper,12pt]{article}",
+        r"\usepackage[utf8]{inputenc}",
+        r"\usepackage[T1]{fontenc}",
+        r"\usepackage{lmodern}",
+        r"\usepackage{geometry}",
+        r"\geometry{margin=2cm}",
+        r"\usepackage{longtable}",
+        r"\usepackage{array}",
+        r"\usepackage{booktabs}",
+        r"\usepackage{fancyhdr}",
+        r"\pagestyle{fancy}",
+        r"\fancyhf{}",
+        r"\fancyhead[C]{SOBRANG TINDANG INA!}",
+        r"\fancyfoot[C]{\thepage}",
+        r"\usepackage{lastpage}",
+        r"\usepackage{noto}",  # Using Noto font for compatibility
+        r"\begin{document}",
+        r"\begin{center}",
+        r"{\Large \textbf{OFFICIAL RECEIPT}} \\[0.2cm]",
+        r"{\normalsize SOBRANG TINDANG INA!} \\",
+        r"{\normalsize Online Smart Checkout} \\",
+        r"\vspace{0.5cm}",
+        r"\hrulefill \\",
+        r"\end{center}",
+        r"\vspace{0.5cm}",
+        r"\begin{longtable}{p{10cm} r}",
+        r"\hline",
+        r"\textbf{Item} & \textbf{Price (\textsf{₱})} \\",
+        r"\hline",
+        r"\endhead",
+    ]
+
     receipt_lines.append("========================================")
     receipt_lines.append("           🧾 OFFICIAL RECEIPT")
     receipt_lines.append("         🛒 SOBRANG TINDANG INA!")
@@ -64,6 +100,9 @@ def classify_and_track(files, budget):
             if len(name) > 26:
                 name = name[:23] + "..."
             receipt_lines.append(f" {name:<28} ₱{price:>6.2f}")
+            # Sanitize name for LaTeX
+            latex_name = name.replace("_", r"\_").replace("&", r"\&").replace("%", r"\%")
+            latex_lines.append(f"{latex_name} & {price:.2f} \\\\")
 
     remaining = budget - total_cost
     receipt_lines.append("----------------------------------------")
@@ -72,15 +111,25 @@ def classify_and_track(files, budget):
     receipt_lines.append(f" REMAINING{'':<20}₱{remaining:>7.2f}")
     receipt_lines.append("========================================")
 
+    latex_lines.append(r"\hline")
+    latex_lines.append(f"\\textbf{{Total}} & {total_cost:.2f} \\\\")
+    latex_lines.append(f"\\textbf{{Budget}} & {budget:.2f} \\\\")
+    latex_lines.append(f"\\textbf{{Remaining}} & {remaining:.2f} \\\\")
+    latex_lines.append(r"\hline")
+    latex_lines.append(r"\end{longtable}")
+    latex_lines.append(r"\vspace{0.5cm}")
+
     percent = min(100, int((total_cost / budget) * 100)) if budget else 0
     status = f"**Total: ₱{total_cost:.2f} / ₱{budget:.2f}**"
 
     if total_cost <= budget:
         receipt_lines.append(" ✅ You're within budget! 🎉")
+        latex_lines.append(r"{\color{green} \textbf{You're within budget!}} \\")
         status_msg = f"✅ **Within Budget!** 🎉\n{status}"
         sound_path = "success.mp3"
     else:
         receipt_lines.append(" ❌ Over budget! Remove some snacks!")
+        latex_lines.append(r"{\color{red} \textbf{Over budget! Remove some snacks!}} \\")
         status_msg = f"🚨 **Over Budget!** 💸\n{status}"
         sound_path = "fail.mp3"
 
@@ -89,11 +138,26 @@ def classify_and_track(files, budget):
     receipt_lines.append("   Come back for smarter buys! 🧾")
     receipt_lines.append("========================================")
 
+    latex_lines.append(r"\vspace{0.5cm}")
+    latex_lines.append(r"\begin{center}")
+    latex_lines.append(r"THANK YOU FOR SHOPPING WITH US! \\")
+    latex_lines.append(r"Come back for smarter buys! \\")
+    latex_lines.append(r"\hrulefill")
+    latex_lines.append(r"\end{center}")
+    latex_lines.append(r"\end{document}")
+
     receipt_text = "```\n" + "\n".join(receipt_lines) + "\n```"
-    return results, status_msg, receipt_text, percent, sound_path
+    latex_content = "\n".join(latex_lines)
+
+    # Create a file-like object for the LaTeX content
+    latex_file = io.StringIO()
+    with redirect_stdout(latex_file):
+        print(latex_content)
+
+    return results, status_msg, receipt_text, percent, sound_path, latex_file
 
 def reset_ui():
-    return [], "", "", 0, None
+    return [], "", "", 0, None, None
 
 # Load and encode local background image
 with open("background.jpg", "rb") as img_file:
@@ -172,16 +236,17 @@ with gr.Blocks(title="Grocery Classifier") as app:
         progress_bar = gr.Slider(label="Budget Used (%)", minimum=0, maximum=100, interactive=False)
         receipt_display = gr.Markdown(label="🧾 Grocery Receipt")
         audio_output = gr.Audio(interactive=False, autoplay=True)
+        pdf_download = gr.File(label="📄 Download Receipt as PDF", interactive=False)
 
     classify_btn.click(
         fn=classify_and_track,
         inputs=[file_input, budget_input],
-        outputs=[gallery, status_display, receipt_display, progress_bar, audio_output]
+        outputs=[gallery, status_display, receipt_display, progress_bar, audio_output, pdf_download]
     )
 
     reset_btn.click(
         fn=reset_ui,
-        outputs=[gallery, status_display, receipt_display, progress_bar, audio_output]
+        outputs=[gallery, status_display, receipt_display, progress_bar, audio_output, pdf_download]
     )
 
 app.launch()