handled brainstrom endpoint for images

main.py

@@ -9,7 +9,6 @@ from my_agent.utils.utils import encode_image_to_base64 , generate_final_story,
 
 
 import json
-from my_agent.utils.nodes import generate_final_story
 
 # Store brainstorming results per thread_id
 
@@ -72,7 +71,7 @@ def generate_image_endpoint():
     image = generate_image(str(stored_data['final_story']))
     stored_data['generated_image']=image
     return {
-        'image':image
+        'response':image
     }