Update app.py

fix list error

app.py

@@ -8,7 +8,7 @@ import shutil
 import asyncio
 
 # ==========================================
-# 0. ASYNC FIX (CRITICAL)
+# 0. ASYNC FIX
 # ==========================================
 try:
     asyncio.get_running_loop()
@@ -30,7 +30,7 @@ REQUIRED_PDFS = ["Bank Loan Overall Risk Policy.pdf", "Bank Loan Interest Rate P
 
 try:
     from langchain_groq import ChatGroq
-    from langchain_google_genai import ChatGoogleGenerativeAI
+    from langchain_google_genai import ChatGoogleGenerativeAI, HarmBlockThreshold, HarmCategory
     import google.generativeai as genai
     
     from langchain_huggingface import HuggingFaceEmbeddings
@@ -38,11 +38,11 @@ try:
     from langchain_community.callbacks import StreamlitCallbackHandler
     from langchain_community.document_loaders import PyPDFLoader
     from langchain_text_splitters import CharacterTextSplitter
-    from langchain_core.prompts import PromptTemplate  # Changed for ReAct
+    from langchain_core.prompts import PromptTemplate
     from langchain_core.runnables import RunnablePassthrough
     from langchain_core.output_parsers import StrOutputParser
     from langchain_core.tools import tool
-    from langchain.agents import AgentExecutor, create_react_agent # Changed Agent Type
+    from langchain.agents import AgentExecutor, create_react_agent
 except ImportError as e:
     st.error(f"❌ Import Error: {e}")
     st.stop()
@@ -78,7 +78,7 @@ def run_query(query, params=()):
 # ==========================================
 @tool
 def get_credit_score(user_id: str) -> str:
-    """Queries SQL DB for Credit Score. Input is just the numeric ID string."""
+    """Queries SQL DB for Credit Score. Input is ID string."""
     clean_id = ''.join(filter(str.isdigit, str(user_id)))
     row = run_query("SELECT Credit_Score FROM credit_score WHERE ID = ?", (clean_id,))
     return f"Credit Score: {row[0]}" if (row and not isinstance(row, str)) else "User ID not found."
@@ -201,12 +201,21 @@ if st.session_state.get('auth_status', False):
             model_name="llama-3.3-70b-versatile"
         )
     else:
-        # Use Gemini 1.5 Flash
+        # CRITICAL FIX: DISABLE SAFETY FILTERS
+        # This prevents Gemini from returning an empty list or error when it gets confused
+        safety = {
+            HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
+            HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
+            HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
+            HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
+        }
+        
         llm = ChatGoogleGenerativeAI(
             google_api_key=current_key, 
             temperature=0, 
             model="gemini-1.5-flash",
-            transport="rest"
+            transport="rest",
+            safety_settings=safety # <--- APPLY FIX
         )
 
     # --- RAG CHAIN ---
@@ -224,7 +233,7 @@ if st.session_state.get('auth_status', False):
 
     tools = [get_credit_score, get_account_status, check_pr_status, consult_policy_doc]
     
-    # --- REACT AGENT TEMPLATE (Universal Compatibility) ---
+    # --- REACT PROMPT ---
     template = '''Answer the following questions as best you can. You have access to the following tools:
 
 {tools}
@@ -247,7 +256,7 @@ Thought:{agent_scratchpad}'''
 
     prompt = PromptTemplate.from_template(template)
     
-    # Switch to create_react_agent (More Robust than tool_calling_agent)
+    # --- AGENT CREATION ---
     agent = create_react_agent(llm, tools, prompt)
     
     agent_executor = AgentExecutor(
@@ -255,9 +264,10 @@ Thought:{agent_scratchpad}'''
         tools=tools, 
         verbose=True, 
         return_intermediate_steps=True,
-        handle_parsing_errors=True
+        handle_parsing_errors=True # Auto-fix formatting errors
     )
 
+    # --- UI ---
     col1, col2 = st.columns([1, 2])
     with col1:
         uid = st.text_input("Customer ID", "1111")
@@ -293,7 +303,6 @@ Thought:{agent_scratchpad}'''
             st.markdown(final_output)
             
             with st.expander("Trace"):
-                # Handle ReAct agent steps format
                 steps = res.get("intermediate_steps", [])
                 for action, obs in steps:
                     st.markdown(f"**Tool:** `{action.tool}`\n**Result:** `{obs}`")