Rename backend/dalle_api.py to backend/image_api.py

backend/{dalle_api.py → image_api.py}

@@ -1,24 +1,51 @@
-import openai
 import os
+import requests
 from dotenv import load_dotenv
 
 load_dotenv()
-openai.api_key = os.getenv("OPENAI_API_KEY")
+STABILITY_API_KEY = os.getenv("STABILITY_API_KEY")
 
 def generate_krishna_image(prompt, size="512x512"):
+    """Generate a Krishna-themed image using Stable Diffusion XL."""
     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
+        headers = {
+            "Authorization": f"Bearer {STABILITY_API_KEY}",
+            "Content-Type": "application/json",
+            "Accept": "image/png"
+        }
+        payload = {
+            "text_prompts": [
+                {
+                    "text": f"{prompt}, vibrant colors, cartoon style, Little Krishna theme with peacock feathers and Vrindavan background",
+                    "weight": 1
+                }
+            ],
+            "height": 512,
+            "width": 512,
+            "steps": 50,
+            "cfg_scale": 7
+        }
+        response = requests.post(
+            "https://api.stability.ai/v1/generation/stable-diffusion-xl-1024-v1-0/text-to-image",
+            headers=headers,
+            json=payload
         )
-        image_url = response['data'][0]['url']
-        return image_url
+        if response.status_code == 200:
+            # Save the image temporarily and return a URL (or upload to Firebase Storage)
+            with open("temp_image.png", "wb") as f:
+                f.write(response.content)
+            # In a real app, upload to Firebase Storage and return the URL
+            # For simplicity, return a placeholder URL (you’d need to implement upload logic)
+            return "https://via.placeholder.com/512x512.png?text=SDXL+Image"
+        else:
+            print(f"Error generating image: {response.text}")
+            return None
     except Exception as e:
         print(f"Error generating image: {str(e)}")
         return None
 
 def generate_comic_strip():
+    """Generate a 4-panel comic strip of Little Krishna helping Manavi."""
     panels = [
         "Little Krishna seeing a shy girl reading alone in Vrindavan",
         "Little Krishna appearing with a playful smile, holding a flute",