Update tools/analyze.py

tools/analyze.py

@@ -1,21 +1,29 @@
+import os
 import base64
 import requests
-from PIL import Image
-import os
+
 
 OPENROUTER_API_KEY = "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"
 
-MODEL = "openai/gpt-4.1-mini"
+# Change this if you use another vision model
+MODEL = "google/gemini-2.5-flash"
 
 
-def analyze_image(image_path):
+def analyze_image(image_path: str):
+
+    if not os.path.exists(image_path):
+        return {
+            "description": "Image not found."
+        }
 
     with open(image_path, "rb") as f:
-        image_base64 = base64.b64encode(f.read()).decode()
+        image_base64 = base64.b64encode(f.read()).decode("utf-8")
 
     headers = {
         "Authorization": f"Bearer {OPENROUTER_API_KEY}",
         "Content-Type": "application/json",
+        "HTTP-Referer": "http://localhost",
+        "X-Title": "AI Image Studio"
     }
 
     payload = {
@@ -26,7 +34,18 @@ def analyze_image(image_path):
                 "content": [
                     {
                         "type": "text",
-                        "text": "Analyze this image in detail. Return description, objects, style and lighting."
+                        "text": """
+Analyze this image.
+
+Return:
+
+Description:
+Objects:
+Style:
+Lighting:
+Colors:
+Suggestions:
+"""
                     },
                     {
                         "type": "image_url",
@@ -36,18 +55,69 @@ def analyze_image(image_path):
                     }
                 ]
             }
-        ]
+        ],
+        "temperature": 0.2,
+        "max_tokens": 500
     }
 
-    response = requests.post(
-        "https://openrouter.ai/api/v1/chat/completions",
-        headers=headers,
-        json=payload,
-        timeout=120
-    )
+    try:
+
+        response = requests.post(
+            "https://openrouter.ai/api/v1/chat/completions",
+            headers=headers,
+            json=payload,
+            timeout=120
+        )
+
+        print("Status:", response.status_code)
+
+        result = response.json()
+
+        print("========== OPENROUTER RESPONSE ==========")
+        print(result)
+        print("=========================================")
+
+        if response.status_code != 200:
+
+            message = result.get(
+                "error",
+                {}
+            ).get(
+                "message",
+                "Unknown OpenRouter error."
+            )
+
+            return {
+                "description": f"OpenRouter Error ({response.status_code}): {message}"
+            }
+
+        if "error" in result:
+
+            return {
+                "description": result["error"].get(
+                    "message",
+                    "Unknown OpenRouter error."
+                )
+            }
+
+        if "choices" not in result:
+
+            return {
+                "description": f"Unexpected response:\n{result}"
+            }
+
+        content = (
+            result["choices"][0]
+            .get("message", {})
+            .get("content", "")
+        )
+
+        return {
+            "description": content
+        }
 
-    result = response.json()
+    except Exception as e:
 
-    return {
-        "description": result["choices"][0]["message"]["content"]
-    }
+        return {
+            "description": str(e)
+        }