Update app.py

app.py

@@ -34,20 +34,28 @@ os.makedirs(LOGS_DIR, exist_ok=True)
 
 def log_to_file(task_id: str, question: str, log_data: Dict[str, Any]):
     """Store logs for a question in a JSON file."""
-    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    filename = f"{LOGS_DIR}/question_{task_id}_{timestamp}.json"
-    
-    log_entry = {
-        "task_id": task_id,
-        "question": question,
-        "timestamp": timestamp,
-        "logs": log_data
-    }
-    
-    with open(filename, 'w', encoding='utf-8') as f:
-        json.dump(log_entry, f, indent=2, ensure_ascii=False)
-    
-    print(f"Logs saved to {filename}")
+    try:
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        filename = f"{LOGS_DIR}/question_{task_id}_{timestamp}.json"
+        
+        log_entry = {
+            "task_id": task_id,
+            "question": question,
+            "timestamp": timestamp,
+            "logs": log_data
+        }
+        
+        print(f"\n=== Saving Logs ===")
+        print(f"Task ID: {task_id}")
+        print(f"Question: {question}")
+        print(f"Log Data: {json.dumps(log_data, indent=2)}")
+        
+        with open(filename, 'w', encoding='utf-8') as f:
+            json.dump(log_entry, f, indent=2, ensure_ascii=False)
+        
+        print(f"Logs saved to {filename}")
+    except Exception as e:
+        print(f"Error saving logs: {e}")
 
 class AgentState(TypedDict):
     question: Annotated[str, override]
@@ -154,10 +162,12 @@ Return ONLY a Python dictionary in this exact format, with no other text or expl
             print(f"LLM Response: {llm_response}")
             
             # Log the analysis step
-            state["logs"]["analyze"] = {
-                "prompt": prompt,
-                "response": llm_response,
-                "timestamp": datetime.now().isoformat()
+            state["logs"] = {
+                "analyze": {
+                    "prompt": prompt,
+                    "response": llm_response,
+                    "timestamp": datetime.now().isoformat()
+                }
             }
             
             analysis = ast.literal_eval(llm_response)
@@ -176,9 +186,11 @@ Return ONLY a Python dictionary in this exact format, with no other text or expl
             state["current_step"] = 'search'
             
             # Log the error
-            state["logs"]["analyze_error"] = {
-                "error": str(e),
-                "timestamp": datetime.now().isoformat()
+            state["logs"] = {
+                "analyze_error": {
+                    "error": str(e),
+                    "timestamp": datetime.now().isoformat()
+                }
             }
         
         return state
@@ -300,12 +312,18 @@ Return ONLY the direct answer to the question. Do not include any explanations,
             # Run the workflow
             final_state = self.workflow.invoke(initial_state)
             
+            # Ensure logs directory exists
+            os.makedirs(LOGS_DIR, exist_ok=True)
+            
             # Save logs to file
-            log_to_file(
-                task_id=final_state["task_id"],
-                question=final_state["question"],
-                log_data=final_state["logs"]
-            )
+            if final_state["logs"]:  # Only save if we have logs
+                log_to_file(
+                    task_id=final_state["task_id"],
+                    question=final_state["question"],
+                    log_data=final_state["logs"]
+                )
+            else:
+                print("No logs to save in final state")
             
             return final_state["final_answer"]