Update app.py

app.py

@@ -105,7 +105,7 @@ def tool_selector(state: AgentState) -> AgentState:
             "  {'action':'audio'}\n"
             "  {'action':'excel'}\n"
             "  {'action':'code', 'snippet':'<python code>'}\n"
-            "  {'action':'code', 'file':'<name>.py'}\n"
+            "  {'action':'code'}\n"
             "  {'action':'final'}\n"
 
             
@@ -163,13 +163,18 @@ def excel_tool(state: AgentState) -> AgentState:
     return state
 
 def code_tool(state: AgentState) -> AgentState:
-    out = {"analysis": analyze_code_tool({
-        "task_id": state.task_id,
-        "snippet": state.snippet,
-        "file": state.file
-    })}
+    if state.snippet:
+        out = {"analysis": analyze_code_tool({
+            "task_id": state.task_id,
+            "snippet": state.snippet,
+        })}
+    else:
+        out = {"analysis": analyze_code_tool({
+            "task_id": state.task_id,
+            "snippet": ""
+        })}
     state.tool_calls += 1
-    state.add(SystemMessage(content=f"CODE_TOOL_OUT: {brief(out)}"))
+    state.add(SystemMessage(content=f"CODE_TOOL_OUT: {out}"))
     state.next_action = None
     return state