Update app.py

app.py

@@ -316,7 +316,7 @@ class BasicAgent:
 
     def _perform_search(self, state: AgentState) -> AgentState:
         try:
-            results = simple_search(state["search_query"], max_results=10)
+            results = simple_search(state["search_query"], max_results=6)  # Reduced to 6 results
             print("\nSearch Results:")
             for i, s in enumerate(results, 1):
                 print(f"[{i}] {s[:120]}…")
@@ -328,17 +328,11 @@ class BasicAgent:
                 state["needs_search"] = False
                 
             state["history"].append({"step": "search", "results": results})
-            state["logs"]["search"] = {
-                "query": state["search_query"],
-                "results_count": len(results),
-                "results": results
-            }
             
         except Exception as e:
             print(f"Search error: {str(e)}")
             state["needs_search"] = True
             state["history"].append({"step": "search", "error": str(e)})
-            state["logs"]["search_error"] = str(e)
             
         state["current_step"] = "recheck"
         return state
@@ -401,7 +395,6 @@ Think step-by-step. Write ANSWER: <answer> on its own line.
         
         state["final_answer"] = answer
         state["history"].append({"step": "answer", "output": raw})
-        state["logs"]["final_answer"] = {"prompt": prompt, "response": raw}
         state["current_step"] = "done"
         return state
 
@@ -561,16 +554,12 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             final_state = agent.workflow.invoke(state)
             answer = final_state["final_answer"]
             
-            # Format logs for display
-            logs_text = json.dumps(final_state["logs"], indent=2)
-            
             # Add to results
             answers_payload.append({"task_id": task_id, "submitted_answer": answer})
             results_log.append({
                 "Task ID": task_id,
                 "Question": question_text,
-                "Submitted Answer": answer,
-                "Processing Logs": logs_text
+                "Submitted Answer": answer
             })
             
             print(f"Completed question {task_id}")
@@ -580,8 +569,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             results_log.append({
                 "Task ID": task_id,
                 "Question": question_text,
-                "Submitted Answer": f"ERROR: {e}",
-                "Processing Logs": f"Error occurred: {str(e)}"
+                "Submitted Answer": f"ERROR: {e}"
             })
 
     if not answers_payload: