Update main.py

main.py

@@ -167,7 +167,7 @@ def extract_page_text(reader, page_num):
         return reader.pages[page_num].extract_text() or ""
     return ""
 
-def process_with_gemini(model, text):
+def process_with_gemini(text: str) -> str::
     prompt = """Analyze this bank statement and extract transactions in JSON format with these fields:
     - Date (format DD/MM/YYYY)
     - Description
@@ -243,7 +243,7 @@ def process_pdf_pages(pdf_file):
         if isinstance(txns, list):
             all_txns.extend(txns)
 
-    return all_txns, raw
+    return all_txns
 
 # --------- Chat Endpoint ---------
 @app.route("/chat", methods=["POST"])
@@ -312,7 +312,7 @@ def upload_statements():
             fs.collection("transactions").add(record)
             stored_count += 1
 
-    return jsonify({"message": f"Stored {stored_count} transactions", "file": str(raw)}), 200
+    return jsonify({"message": f"Stored {stored_count} transactions"}), 200
 
 # --------- Endpoint: Retrieve or Generate Financial Statement ---------