Update app.py

app.py

@@ -154,9 +154,18 @@ class BasicAgent:
                 state["current_step"] = "sheet"
             return state
 
+        # Check if question is about the file we just processed
+        if state["history"]:
+            last_step = state["history"][-1]["step"]
+            if last_step in ["video", "image", "sheet"]:
+                state["current_step"] = "answer"
+                return state
+
         # Regular text question analysis
         prompt = (
-            "Decide if this question needs web search. Respond with a Python dict:\n"
+            "Decide if this question needs web search. Only use search for factual questions that require external knowledge. "
+            "For questions about files we just processed or simple reasoning, don't use search. "
+            "Respond with a Python dict:\n"
             "{\n  'needs_search': bool,\n  'search_query': str\n}\n\n"
             f"Question: {state['question']}"
         )
@@ -167,7 +176,7 @@ class BasicAgent:
             state["search_query"] = decision.get("search_query", state["question"])
         except Exception as e:
             print(f"\nLLM Error in question analysis: {str(e)}")
-            state["needs_search"] = True
+            state["needs_search"] = False  # Default to no search on error
             state["search_query"] = state["question"]
             
         state["current_step"] = "search" if state["needs_search"] else "answer"