Update app.py

app.py

@@ -2,13 +2,15 @@ import gradio as gr
 import PyPDF2
 import google.generativeai as genai
 import re
-import io
+import tempfile
+import os
 
 # 🔐 Gemini API Key
 GEMINI_API_KEY = "AIzaSyDnx_qUjGTFG1pv1otPUhNt_bGGv14aMDI"
 genai.configure(api_key=GEMINI_API_KEY)
 
-# 📄 Extract text from PDFdef extract_text_from_pdf(file):
+# 📄 Extract text from PDF
+def extract_text_from_pdf(file):
     try:
         reader = PyPDF2.PdfReader(file)
         text = ""
@@ -17,18 +19,25 @@ genai.configure(api_key=GEMINI_API_KEY)
             if content:
                 text += content + "\n"
         return text.strip()
-    except:
+    except Exception as e:
+        print("PDF Extraction Error:", e)
         return ""
 
+# ✂️ Extract sections from full text using regex
 def extract_section(full_text, label):
     pattern = rf"\*\*\- {re.escape(label)}:\*\*\s*(.*?)(?=\n\*\*|\Z)"
     match = re.search(pattern, full_text, re.DOTALL)
     return match.group(1).strip() if match else "❓ Not found"
 
+# 🧠 Main function to analyze financial data
 def analyze_financial_data(file):
     text = extract_text_from_pdf(file)
+
     if not text:
-        return ("⚠️ Failed to extract text. Ensure it’s not a scanned image PDF.", "", "", "", "", "", "", "", None)
+        return (
+            "⚠️ Failed to extract text. Ensure it’s a text-based PDF.",
+            "", "", "", "", "", "", "", None
+        )
 
     prompt = f"""
     Analyze the following Paytm transaction history and generate financial insights in the following structure:
@@ -48,64 +57,67 @@ def analyze_financial_data(file):
         response = model.generate_content(prompt)
         full_text = response.text.strip()
 
-        # Generate downloadable report
-        report_io = io.StringIO()
-        report_io.write(full_text)
-        report_bytes = io.BytesIO(report_io.getvalue().encode())
-        report_bytes.name = "financial_analysis.txt"
+        # Save report to temporary .txt file
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode="w", encoding="utf-8") as tmp:
+            tmp.write(full_text)
+            report_path = tmp.name
 
         return (
-            "✅ Analysis Complete",
-            text[:2000] + "..." if len(text) > 2000 else text,  # Full text preview
+            "✅ Analysis Complete!",
+            text[:2000] + "..." if len(text) > 2000 else text,
             extract_section(full_text, "Monthly Income & Expenses"),
             extract_section(full_text, "Unnecessary Expense Categories"),
             extract_section(full_text, "Estimated Savings %"),
             extract_section(full_text, "Spending Trends"),
             extract_section(full_text, "Category-wise Expense Breakdown (Partial)"),
             extract_section(full_text, "Cost Control Suggestions"),
-            report_bytes
+            report_path
         )
 
     except Exception as e:
         return (f"❌ Gemini Error: {e}", "", "", "", "", "", "", "", None)
 
-# 🎨 Gradio Interface
-with gr.Blocks(title="AI Finance Assistant") as demo:
-    gr.Markdown("## 💰 AI-Powered Personal Finance Assistant")
-    gr.Markdown("Upload your **Paytm transaction PDF** (text-based only) and get structured financial insights using **Gemini AI**.")
+# 🎨 Gradio UI
+with gr.Blocks(title="AI Financial Analyzer", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("""
+    # 💸 AI-Powered Personal Finance Analyzer
+    Upload your **Paytm/YouTube Income PDF** and get structured financial insights using **Gemini AI**.
+    """)
 
     with gr.Row():
-        with gr.Column():
+        with gr.Column(scale=1):
             pdf_input = gr.File(label="📂 Upload PDF", file_types=[".pdf"])
-            analyze_btn = gr.Button("🔍 Analyze My Finances")
+            analyze_btn = gr.Button("🔍 Analyze")
 
-        with gr.Column():
+        with gr.Column(scale=1):
             status = gr.Textbox(label="✅ Status", interactive=False)
-            download_btn = gr.File(label="📥 Download Report", interactive=False)
+            download_btn = gr.File(label="📥 Download AI Report", interactive=False)
 
-    with gr.Accordion("📜 Extracted PDF Text (preview)", open=False):
-        pdf_text = gr.Textbox(lines=10, label="Extracted Text", interactive=False)
+    with gr.Accordion("📜 View Extracted PDF Text (Optional)", open=False):
+        extracted_text = gr.Textbox(label="📝 Extracted Text", lines=10, interactive=False)
 
     with gr.Row():
-        income_box = gr.Textbox(label="💵 Monthly Income & Expenses", lines=5, interactive=False)
-        unnecessary_box = gr.Textbox(label="🛒 Unnecessary Expenses", lines=5, interactive=False)
+        income_expense = gr.Textbox(label="💵 Monthly Income & Expenses", lines=4, interactive=False)
+        unnecessary = gr.Textbox(label="🛒 Unnecessary Expenses", lines=4, interactive=False)
 
     with gr.Row():
-        savings_box = gr.Textbox(label="💰 Estimated Savings %", lines=2, interactive=False)
-        trends_box = gr.Textbox(label="📈 Spending Trends", lines=5, interactive=False)
+        savings = gr.Textbox(label="💰 Estimated Savings %", lines=2, interactive=False)
+        trends = gr.Textbox(label="📈 Spending Trends", lines=4, interactive=False)
 
     with gr.Row():
-        breakdown_box = gr.Textbox(label="📊 Category-wise Breakdown", lines=6, interactive=False)
-        suggestion_box = gr.Textbox(label="🧠 Cost Control Suggestions", lines=6, interactive=False)
+        category_breakdown = gr.Textbox(label="📊 Category-wise Breakdown", lines=6, interactive=False)
+        suggestions = gr.Textbox(label="🧠 Cost Control Suggestions", lines=6, interactive=False)
 
     analyze_btn.click(
         fn=analyze_financial_data,
         inputs=pdf_input,
         outputs=[
-            status, pdf_text,
-            income_box, unnecessary_box, savings_box,
-            trends_box, breakdown_box, suggestion_box, download_btn
+            status, extracted_text,
+            income_expense, unnecessary,
+            savings, trends,
+            category_breakdown, suggestions,
+            download_btn
         ]
     )
 
-demo.launch()
+demo.launch(share=True)