Update app.py

app.py

@@ -9,11 +9,23 @@ def df_from_input(df):
     return pd.DataFrame(df) if df else pd.DataFrame()
 
 
-def run_agent(query, slot_df, pick_df):
-    slotting = df_from_input(slot_df)
-    picking = df_from_input(pick_df)
+def run_agent(message, slotting_df, picking_df):
+    print("📥 INPUT MESSAGE:", message)
+    print("📄 SLOT DATA:", slotting_df)
+    print("🚚 PICKING DATA:", picking_df)
 
-    return agent.run(query, slotting, picking)
+    try:
+        result = agent.run(message, slotting_df, picking_df)
+
+        print("✅ RAW RESULT FROM AGENT:", result)
+
+        return result
+
+    except Exception as e:
+        import traceback
+        print("❌ ERROR OCCURRED:")
+        traceback.print_exc()
+        return f"Error: {str(e)}", None, None, None
 
 
 def build_ui():