Update app.py

app.py

@@ -30,8 +30,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 JINA_API_KEY = os.getenv("JINA_API_KEY")
 
-LOGS_DIR = "question_logs"
-os.makedirs(LOGS_DIR, exist_ok=True)
+# Remove logs directory creation since we're not storing logs anymore
 
 # -------------------------
 # Jina AI search tool (replaces DDG + Reader)
@@ -183,8 +182,6 @@ class BasicAgent:
             "logs": {},
         }
         final_state = self.workflow.invoke(state)
-        if final_state["logs"]:
-            log_to_file(task_id, question, final_state["logs"])
         return final_state["final_answer"]
 
 # ----------------------------------------------------------------------------------
@@ -258,14 +255,33 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
         try:
             print(f"\nProcessing question {task_id}: {question_text[:50]}...")
-            answer = agent(question_text, task_id)
+            
+            # Initialize state for this question
+            state: AgentState = {
+                "question": question_text,
+                "current_step": "analyze",
+                "final_answer": "",
+                "history": [],
+                "needs_search": False,
+                "search_query": "",
+                "task_id": task_id,
+                "logs": {},
+            }
+            
+            # Run the workflow
+            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
+                "Submitted Answer": answer,
+                "Processing Logs": logs_text
             })
             
             print(f"Completed question {task_id}")
@@ -275,7 +291,8 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             results_log.append({
                 "Task ID": task_id,
                 "Question": question_text,
-                "Submitted Answer": f"ERROR: {e}"
+                "Submitted Answer": f"ERROR: {e}",
+                "Processing Logs": f"Error occurred: {str(e)}"
             })
 
     if not answers_payload:
@@ -358,7 +375,11 @@ with gr.Blocks() as demo:
     run_button = gr.Button("Run Evaluation & Submit All Answers")
 
     status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
-    results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
+    results_table = gr.DataFrame(
+        label="Questions and Agent Answers", 
+        wrap=True,
+        column_widths=["10%", "30%", "30%", "30%"]  # Adjust column widths for better display
+    )
 
     run_button.click(
         fn=run_and_submit_all,