Update app.py

app.py

@@ -2,8 +2,10 @@ import gradio as gr
 import PyPDF2
 import google.generativeai as genai
 import re
+import os
 
-GEMINI_API_KEY = "AIzaSyDSqhsofWeBd7FOV5YkdZGw0_OLH4UKBPo"  
+# Load API key safely
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")  # set in environment
 genai.configure(api_key=GEMINI_API_KEY)
 
 def extract_text_from_pdf(file):
@@ -27,26 +29,27 @@ def analyze_financial_data(file):
     text = extract_text_from_pdf(file)
 
     if not text:
-        return (
-            "⚠️ Failed to extract text from the PDF. Ensure it’s not scanned.",
-            "", "", "", "", "", ""
-        )
+        return ("⚠️ Failed to extract text", "", "", "", "", "", "")
+
+    text = text[:6000]  # LIMIT TOKENS (IMPORTANT)
 
     prompt = f"""
-    Analyze the following Paytm transaction history and generate financial insights in the following structure:
+    Analyze Paytm transactions and return structured insights.
+
     **Financial Insights**
-    **- Monthly Income & Expenses:** [data]
-    **- Unnecessary Expense Categories:** [data]
-    **- Estimated Savings %:** [data]
-    **- Spending Trends:** [data]
-    **- Category-wise Expense Breakdown (Partial):** [data]
-    **- Cost Control Suggestions:** [data]
+    **- Monthly Income & Expenses:** ...
+    **- Unnecessary Expense Categories:** ...
+    **- Estimated Savings %:** ...
+    **- Spending Trends:** ...
+    **- Category-wise Expense Breakdown (Partial):** ...
+    **- Cost Control Suggestions:** ...
+
     Transaction History:
     {text}
     """
 
     try:
-        model = genai.GenerativeModel("gemini-1.5-flash")
+        model = genai.GenerativeModel("gemini-1.5-flash-latest")
         response = model.generate_content(prompt)
         full_text = response.text.strip()
 
@@ -65,16 +68,14 @@ def analyze_financial_data(file):
 
 gr.Interface(
     fn=analyze_financial_data,
-    inputs=gr.File(label="📂 Upload Paytm PDF", file_types=[".pdf"]),
+    inputs=gr.File(label="Upload Paytm PDF", file_types=[".pdf"]),
     outputs=[
-        gr.Textbox(label="✅ Status", lines=2, interactive=False),
-        gr.Textbox(label="💵 Monthly Income & Expenses", lines=8, interactive=False),
-        gr.Textbox(label="🛒 Unnecessary Expense Categories", lines=8, interactive=False),
-        gr.Textbox(label="💰 Estimated Savings %", lines=4, interactive=False),
-        gr.Textbox(label="📈 Spending Trends", lines=8, interactive=False),
-        gr.Textbox(label="📊 Category-wise Breakdown", lines=10, interactive=False),
-        gr.Textbox(label="🧠 Cost Control Suggestions", lines=8, interactive=False),
+        gr.Textbox(label="Status"),
+        gr.Textbox(label="Monthly Income & Expenses"),
+        gr.Textbox(label="Unnecessary Expense Categories"),
+        gr.Textbox(label="Estimated Savings %"),
+        gr.Textbox(label="Spending Trends"),
+        gr.Textbox(label="Category-wise Breakdown"),
+        gr.Textbox(label="Cost Control Suggestions"),
     ],
-    title="💰 AI-Powered Personal Finance Assistant",
-    description="Upload your Paytm transaction PDF (text-based) and get structured financial insights using Gemini AI.",
-).launch()
+).launch()