Update app.py

app.py

@@ -87,6 +87,8 @@ agent = CodeAgent(tools=[image_generation_tool, search_tool], model=llm_engine)
 # Main logic for image generation
 # =========================================================
 
+from PIL import Image
+
 def generate_object_history(object_name):
     images = []
     prompts = generate_prompts_for_object(object_name)
@@ -106,26 +108,37 @@ def generate_object_history(object_name):
                 additional_args={"user_prompt": prompt}
             )
 
+            # result is tuple: (filepath, seed)
             if isinstance(result, (list, tuple)):
-                result = result[0]
+                image_filepath = result[0]
+            else:
+                image_filepath = result  # fallback in case result is just a string
+
+            # Open the image from filepath
+            image = Image.open(image_filepath)
 
-            image = result.to_raw()
+            # Save the image to your naming convention
             image_filename = f"{object_name}_{time_period}.png"
             image.save(image_filename)
+
+            # Optional: call your plotting function (if needed)
             plot_and_save_agent_image(image, f"{object_name} - {time_period.title()}", save_path=image_filename)
 
             image_paths.append(image_filename)
             images.append(image)
+
         except Exception as e:
             print(f"Agent failed on {time_period}: {e}")
             continue
 
+    # Create GIF from generated images if any
     gif_path = f"{object_name}_evolution.gif"
     if images:
         images[0].save(gif_path, save_all=True, append_images=images[1:], duration=1000, loop=0)
 
     return image_paths, gif_path
 
+
 # =========================================================
 # Gradio Interface
 # =========================================================