Update app.py

app.py

@@ -157,9 +157,20 @@ def generate_object_history(object_name):
         "future": f"{object_name} - Future"
     }
 
+
+    general_instruction = (
+        "Search the necessary information and features for the following prompt, "
+        "then generate an image of it."
+    )
+
     for time_period, prompt in prompts.items():
         print(f"Generating {time_period} frame: {prompt}")
-        result = agent.run(prompt)
+        #result = agent.run(prompt)
+
+        result = agent.run(
+            general_instruction,
+            additional_args={"user_prompt": prompt}
+        )
         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)