Update backend/dalle_api.py

backend/dalle_api.py

@@ -1,30 +1,35 @@
 import openai
 import os
 from dotenv import load_dotenv
-from messages import krishna_blessings
-from ayush_messages import ayush_surprises
 
 load_dotenv()
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
-def get_krishna_response(user_input):
-    # Check for predefined responses
-    if "birthday" in user_input.lower():
-        return ayush_surprises["birthday"]
-    if "shy" in user_input.lower():
-        return krishna_blessings["shy"]
-    if "hello" in user_input.lower() or "hi" in user_input.lower():
-        return krishna_blessings["greeting"]
+def generate_krishna_image(prompt, size="512x512"):
+    try:
+        response = openai.Image.create(
+            prompt=f"{prompt}, vibrant colors, cartoon style, Little Krishna theme with peacock feathers and Vrindavan background",
+            n=1,
+            size=size
+        )
+        image_url = response['data'][0]['url']
+        return image_url
+    except Exception as e:
+        print(f"Error generating image: {str(e)}")
+        return None
 
-    # Use OpenAI for dynamic responses
-    prompt = (
-        "You are Little Krishna, playful and wise. Respond to this in a fun, Krishna-like way: "
-        f"'{user_input}'"
-    )
-    response = openai.Completion.create(
-        engine="text-davinci-003",
-        prompt=prompt,
-        max_tokens=60,
-        temperature=0.9
-    )
-    return response.choices[0].text.strip()
+def generate_comic_strip():
+    panels = [
+        "Little Krishna seeing a shy girl reading alone in Vrindavan",
+        "Little Krishna appearing with a playful smile, holding a flute",
+        "Little Krishna handing the girl a book, surrounded by cows",
+        "Little Krishna and the girl smiling together, with peacocks dancing"
+    ]
+    comic_images = []
+    for panel_prompt in panels:
+        image_url = generate_krishna_image(panel_prompt)
+        if image_url:
+            comic_images.append(image_url)
+        else:
+            comic_images.append("https://via.placeholder.com/512x512.png?text=Error+Generating+Image")
+    return comic_images