Update app.py

app.py

@@ -90,50 +90,41 @@ agent = CodeAgent(tools=[image_generation_tool, search_tool], model=llm_engine)
 def generate_object_history(object_name):
     images = []
     prompts = generate_prompts_for_object(object_name)
-    labels = {
-        "past": f"{object_name} - Past",
-        "present": f"{object_name} - Present",
-        "future": f"{object_name} - Future"
-    }
-
 
     general_instruction = (
         "Search the necessary information and features for the following prompt, "
         "then generate an image of it."
     )
 
+    image_paths = []
     for time_period, prompt in prompts.items():
         print(f"Generating {time_period} frame: {prompt}")
-        #result = agent.run(prompt)
 
         try:
             result = agent.run(
                 general_instruction,
                 additional_args={"user_prompt": prompt}
             )
-            
-            #if isinstance(result, (list, tuple)):
-            result = result[0]
-            
+
+            if isinstance(result, (list, tuple)):
+                result = result[0]
+
             image = result.to_raw()
+            image_filename = f"{object_name}_{time_period}.png"
+            image.save(image_filename)
+            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
-            
-        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)
 
     gif_path = f"{object_name}_evolution.gif"
-    images[0].save(gif_path, save_all=True, append_images=images[1:], duration=1000, loop=0)
-    #return [
-    #    f"{object_name}_past.png",
-    #    f"{object_name}_present.png",
-    #    f"{object_name}_future.png"], gif_path
-    return [(f"{object_name}_past.png", labels["past"]),
-           (f"{object_name}_present.png", labels["present"]),
-           (f"{object_name}_future.png", labels["future"])], gif_path
-    #return images, gif_path
+    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
@@ -148,32 +139,24 @@ def create_gradio_interface():
         """)
 
         default_images = [
-            ("car_past.png", "Car - Past"),
-            ("car_present.png", "Car - Present"),
-            ("car_future.png", "Car - Future")
+            "car_past.png",
+            "car_present.png",
+            "car_future.png"
         ]
-        #default_images = [
-        #        "car_past.png",
-        #        "car_present.png",
-        #        "car_future.png"
-        #    ]
         default_gif_path = "car_evolution.gif"
 
         with gr.Row():
             with gr.Column():
                 object_name_input = gr.Textbox(label="Enter an object name", placeholder="e.g. bicycle, car, phone")
                 generate_button = gr.Button("Generate Evolution")
-                image_gallery = gr.Gallery(label="Generated Images", columns=3, rows=1, value=default_images, type="filepath" )
-                #image_gallery = gr.Gallery(label="Generated Images", columns=3, rows=1, type="filepath")
-                gif_output = gr.Image(label="Generated GIF", value=default_gif_path)
+                image_gallery = gr.Gallery(label="Generated Images", columns=3, rows=1, value=default_images, type="filepath")
+                gif_output = gr.Image(label="Generated GIF", value=default_gif_path, type="filepath")
 
         generate_button.click(fn=generate_object_history, inputs=[object_name_input], outputs=[image_gallery, gif_output])
 
     return demo
 
-# =========================================================
-# Run the app
-# =========================================================
 
+# Launch the interface
 demo = create_gradio_interface()
-demo.launch(share=True)
+demo.launch()