Update

app.py

@@ -18,6 +18,7 @@ from langchain.schema import SystemMessage
 import logging
 import sys
 from langchain_community.utilities import DuckDuckGoSearchAPIWrapper
+from openai import OpenAI
 
 # LangChain
 from langchain_core.messages import AIMessage, HumanMessage, SystemMessage, ToolMessage
@@ -74,10 +75,23 @@ def calculate(expression: str) -> str:
 
 @tool
 def generate_image_prompt(description: str) -> str:
-    """Generate a detailed prompt for image generation based on a description."""
-    logger.info(f"Generating image prompt for: {description}")
-    enhanced_prompt = f"An image of {description}, high resolution, detailed lighting, professional quality"
-    return f"Generated image prompt: {enhanced_prompt}"
+    """Generate an image based on a description using DALL-E."""
+    logger.info(f"Generating image for: {description}")
+    try:
+        client = OpenAI()
+        response = client.images.generate(
+            model="dall-e-3",
+            prompt=description,
+            size="1024x1024",
+            quality="standard",
+            n=1,
+        )
+        image_url = response.data[0].url
+        logger.info(f"Image generated successfully: {image_url}")
+        return f"Here's the generated image: {image_url}"
+    except Exception as e:
+        logger.error(f"Error generating image: {str(e)}")
+        return f"Error generating image: {str(e)}"
 
 
 # Create tools list