Update backend/image_api.py

backend/image_api.py

@@ -2,41 +2,45 @@ import os
 import requests
 from dotenv import load_dotenv
 
+# Load environment variables (Hugging Face Space secrets)
 load_dotenv()
-STABILITY_API_KEY = os.getenv("STABILITY_API_KEY")
+HUGGINGFACE_API_TOKEN = os.getenv("HUGGINGFACE_API_TOKEN")
 
 def generate_krishna_image(prompt, size="512x512"):
-    """Generate a Krishna-themed image using Stable Diffusion XL."""
+    """
+    Generate a Krishna-themed image using Stable Diffusion.
+    
+    Args:
+        prompt (str): The prompt for image generation.
+        size (str): Image size (default: "512x512").
+    
+    Returns:
+        str: URL of the generated image (or placeholder if failed).
+    """
     try:
         headers = {
-            "Authorization": f"Bearer {STABILITY_API_KEY}",
-            "Content-Type": "application/json",
-            "Accept": "image/png"
+            "Authorization": f"Bearer {HUGGINGFACE_API_TOKEN}",
+            "Content-Type": "application/json"
         }
         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
+            "inputs": f"{prompt}, vibrant colors, cartoon style, Little Krishna theme with peacock feathers and Vrindavan background",
+            "parameters": {
+                "num_inference_steps": 50,
+                "guidance_scale": 7.5
+            }
         }
         response = requests.post(
-            "https://api.stability.ai/v1/generation/stable-diffusion-xl-1024-v1-0/text-to-image",
+            "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5",
             headers=headers,
             json=payload
         )
         if response.status_code == 200:
-            # Save the image temporarily and return a URL (or upload to Firebase Storage)
+            # The response is a binary image; save it temporarily
             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"
+            # For simplicity, return a placeholder URL
+            return "https://via.placeholder.com/512x512.png?text=Stable+Diffusion+Image"
         else:
             print(f"Error generating image: {response.text}")
             return None
@@ -45,7 +49,12 @@ def generate_krishna_image(prompt, size="512x512"):
         return None
 
 def generate_comic_strip():
-    """Generate a 4-panel comic strip of Little Krishna helping Manavi."""
+    """
+    Generate a 4-panel comic strip of Little Krishna helping Manavi.
+    
+    Returns:
+        list: List of image URLs for the 4 panels.
+    """
     panels = [
         "Little Krishna seeing a shy girl reading alone in Vrindavan",
         "Little Krishna appearing with a playful smile, holding a flute",