Update app.py

app.py

@@ -3,6 +3,8 @@ import streamlit as st
 import pandas as pd
 import matplotlib.pyplot as plt
 import seaborn as sns
+import time
+import random
 
 # Updated LangChain Imports
 from langchain_google_genai import ChatGoogleGenerativeAI
@@ -48,13 +50,14 @@ def main():
         query = st.text_area("What analysis would you like to perform?", placeholder="e.g., 'Plot the distribution of prices.'")
 
         if st.button("Run Agent") and query:
+            # Initialize LLM with max_retries at the model level
             llm = ChatGoogleGenerativeAI(
                 model="gemini-2.5-flash-preview-09-2025", 
                 google_api_key=GEMINI_API_KEY,
                 temperature=0,
+                max_retries=6, # LangChain's built-in retry logic
             )
 
-            # CUSTOM PROMPT PREFIX: Helps Gemini follow the ReAct format strictly to avoid parsing errors
             custom_prefix = """
             You are working with a pandas dataframe in Python. The name of the dataframe is `df`.
             You should use the tools below to answer the question posed of you.
@@ -62,9 +65,6 @@ def main():
             Do not wrap the code in markdown code blocks within the Action Input.
             """
 
-            # Create the Pandas Agent with enhanced error handling
-            # Fix: Pass 'handle_parsing_errors' directly and avoid passing 'max_iterations' 
-            # inside agent_executor_kwargs if it conflicts with the toolkit's defaults.
             agent = create_pandas_dataframe_agent(
                 llm,
                 df,
@@ -73,8 +73,6 @@ def main():
                 allow_dangerous_code=True,
                 prefix=custom_prefix,
                 handle_parsing_errors=True,
-                # We omit agent_executor_kwargs here to resolve the TypeError
-                # as the function already handles execution parameters internally.
             )
 
             # --- 4. EXECUTION ---
@@ -83,19 +81,30 @@ def main():
             st_callback = StreamlitCallbackHandler(thought_container)
 
             with st.spinner("Agent is analyzing..."):
-                try:
-                    # Run the loop
-                    response = agent.run(query, callbacks=[st_callback])
+                # Manual retry logic wrapper for the whole agentic loop
+                max_loop_retries = 3
+                for attempt in range(max_loop_retries):
+                    try:
+                        response = agent.run(query, callbacks=[st_callback])
+                        st.markdown("---")
+                        st.subheader("✅ Final Analysis Result")
+                        st.success(response)
+                        break # Exit loop if successful
                     
-                    st.markdown("---")
-                    st.subheader("✅ Final Analysis Result")
-                    st.success(response)
-                    
-                except Exception as e:
-                    st.error(f"Agent failed to complete the task.")
-                    with st.expander("Show Technical Error"):
-                        st.code(str(e))
-                    st.info("💡 Tip: The agent had trouble formatting its 'Action'. Try asking the question again or rephrasing it.")
+                    except Exception as e:
+                        if "429" in str(e) or "RESOURCE_EXHAUSTED" in str(e):
+                            if attempt < max_loop_retries - 1:
+                                wait_time = (2 ** attempt) + random.random()
+                                st.warning(f"Quota reached. Retrying in {wait_time:.2f} seconds...")
+                                time.sleep(wait_time)
+                                continue
+                            else:
+                                st.error("Rate limit exceeded consistently. Please try again in a minute.")
+                        else:
+                            st.error(f"Agent failed to complete the task.")
+                            with st.expander("Show Technical Error"):
+                                st.code(str(e))
+                            break
     else:
         st.info("👆 Upload a CSV to begin.")