Update app.py

app.py

@@ -16,6 +16,9 @@ from diffusers import DiffusionPipeline
 
 
 
+openai_api_key = os.getenv("OPENAI_API_KEY") 
+from langchain_openai import OpenAI 
+llm =OpenAI(temperature =0.8)
 
 
 
@@ -83,7 +86,9 @@ def generate_caption_and_image(image):
     out = model.generate(**inputs)
     caption1 = processor.decode(out[0], skip_special_tokens=True)
 
-    prompt = f'''Create a highly realistic clothing item based on the following descriptions: The design should reflect {caption1} and {caption2}, blending both themes into a single, stylish, and modern piece of clothing. Incorporate highly realistic and high-quality textures that exude sophistication, with realistic fabric lighting and fine details. Subtly hint at {selected_fabric}, featuring a {selected_pattern} motif and a {selected_textile_design} style that harmoniously balances the essence of both captions.'''
+    prompt = f"Write a prompt to generate clothes based on following info: 1. {caption1} 2. {caption2} 3. {selected_fabric} 4. {selected_pattern} 5. {selected_textile_design} return only prompt enclosed in quotes"
+    prompt =llm(prompt)
+