Update app.py

app.py

@@ -6,6 +6,7 @@ from PIL import Image  # 5. Pillow for image processing
 import openai  # 6. OpenAI library for DALL·E API calls
 import os  # 7. OS for environment variables
 from face_to_prompt import extract_face_prompt
+from desc import describe_image_with_gpt4o
 
 
 # 8. Create the FastAPI app
@@ -36,7 +37,8 @@ async def generate(
     with open("/tmp/temp_input.jpg", "wb") as f:
         f.write(await image.read())
 
-    caption = extract_face_prompt("/tmp/temp_input.jpg")
+    # caption = extract_face_prompt("/tmp/temp_input.jpg")
+    caption = describe_image_with_gpt4o("/tmp/temp_input.jpg",OPENAI_API_KEY)
 
     # 16. Construct the DALL·E prompt using the style and the caption
     prompt = (