Update app.py

app.py

@@ -167,10 +167,16 @@ def generate_object_history(object_name):
         print(f"Generating {time_period} frame: {prompt}")
         #result = agent.run(prompt)
 
-        result = agent.run(
-            general_instruction,
-            additional_args={"user_prompt": prompt}
-        )
+        try:
+            result = agent.run(
+                general_instruction,
+                additional_args={"user_prompt": prompt}
+            )
+            image = result.to_raw()
+        except Exception as e:
+            print(f"Agent failed on {time_period}: {e}")
+            continue
+            
         images.append(result.to_raw())
         image_filename = f"{object_name}_{time_period}.png"
         plot_and_save_agent_image(result, labels[time_period], save_path=image_filename)