Update app.py

app.py

@@ -1,6 +1,7 @@
 import re
 import json
 import os
+import time
 from datetime import datetime
 from io import BytesIO
 
@@ -19,6 +20,7 @@ def configure_gemini(api_key):
 
 # Read PDF content from a file-like object (from Streamlit uploader)
 def read_pdf(file_obj):
+    file_obj.seek(0)  # Ensure the file pointer is at the start
     text_content = []
     pdf_reader = pypdf.PdfReader(file_obj)
     for page in pdf_reader.pages:
@@ -51,6 +53,7 @@ Return ONLY valid JSON with this structure:
     ]
 }"""
     response = model.generate_content([prompt, text])
+    time.sleep(7)  # Sleep for 7 seconds to work around rate limit
     return response.text
 
 # Generate financial report from aggregated JSON transactions and chosen sections
@@ -62,6 +65,7 @@ Generate a detailed financial report that includes the following sections: {', '
 Ensure that each section is clearly formatted with headings and includes insights and summaries.
 Return the complete report as plain text."""
     response = model.generate_content([prompt])
+    time.sleep(7)  # Sleep for 7 seconds to work around rate limit
     return response.text
 
 # Create a PDF file from the report text
@@ -80,11 +84,16 @@ def create_pdf_report(report_text):
     return pdf_buffer
 
 def main():
-    st.title("Quantitlytix AI ")
+    st.title("Quantitlytix AI")
     st.markdown("*Bank Statement Parser & Financial Report Generator*")
+    
     # Allow multiple PDF uploads
     uploaded_files = st.file_uploader("Upload PDF bank statements", type="pdf", accept_multiple_files=True)
     
+    # Debug: Show number of files uploaded
+    if uploaded_files:
+        st.write(f"{len(uploaded_files)} file(s) uploaded.")
+    
     if uploaded_files:
         try:
             # Initialize the Gemini model
@@ -94,10 +103,19 @@ def main():
             for uploaded_file in uploaded_files:
                 # Read PDF text directly from the uploaded file
                 pdf_text = read_pdf(uploaded_file)
+                if not pdf_text:
+                    st.warning(f"No text found in {uploaded_file.name}.")
+                    continue
+                
                 with st.spinner(f"Processing {uploaded_file.name}..."):
                     json_response = process_with_gemini(model, pdf_text)
                     # Extract valid JSON from the response
-                    json_str = json_response[json_response.find('{'):json_response.rfind('}')+1]
+                    start_idx = json_response.find('{')
+                    end_idx = json_response.rfind('}') + 1
+                    if start_idx == -1 or end_idx == -1:
+                        st.warning(f"Invalid JSON response for {uploaded_file.name}.")
+                        continue
+                    json_str = json_response[start_idx:end_idx]
                     json_str = json_str.replace('```json', '').replace('```', '')
                     data = json.loads(json_str)
                     transactions = data.get('transactions', [])