Update tools/analyze.py

tools/analyze.py

@@ -1,23 +1,31 @@
 import os
+import json
 import base64
 import requests
 
-
 OPENROUTER_API_KEY = "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"
 
-# Change this if you use another vision model
 MODEL = "google/gemini-2.5-flash"
 
+API_URL = "https://openrouter.ai/api/v1/chat/completions"
+
 
 def analyze_image(image_path: str):
 
+    if not image_path:
+        return {
+            "success": False,
+            "description": "No image provided."
+        }
+
     if not os.path.exists(image_path):
         return {
-            "description": "Image not found."
+            "success": False,
+            "description": f"Image not found: {image_path}"
         }
 
     with open(image_path, "rb") as f:
-        image_base64 = base64.b64encode(f.read()).decode("utf-8")
+        image = base64.b64encode(f.read()).decode("utf-8")
 
     headers = {
         "Authorization": f"Bearer {OPENROUTER_API_KEY}",
@@ -26,6 +34,40 @@ def analyze_image(image_path: str):
         "X-Title": "AI Image Studio"
     }
 
+    prompt = """
+You are an expert computer vision assistant.
+
+Analyze this image.
+
+Return ONLY valid JSON.
+
+{
+    "description":"",
+    "objects":[],
+    "people":"",
+    "style":"",
+    "lighting":"",
+    "colors":[],
+    "background":"",
+    "camera_angle":"",
+    "quality":"",
+    "editing_prompt":"",
+    "suggestions":[]
+}
+
+Rules:
+
+Return JSON only.
+
+Do not write markdown.
+
+Do not use ```.
+
+editing_prompt should preserve every important detail so future editing keeps the same image.
+
+suggestions should contain 6 editing ideas.
+"""
+
     payload = {
         "model": MODEL,
         "messages": [
@@ -34,67 +76,44 @@ def analyze_image(image_path: str):
                 "content": [
                     {
                         "type": "text",
-                        "text": """
-Analyze this image.
-
-Return:
-
-Description:
-Objects:
-Style:
-Lighting:
-Colors:
-Suggestions:
-"""
+                        "text": prompt
                     },
                     {
                         "type": "image_url",
                         "image_url": {
-                            "url": f"data:image/jpeg;base64,{image_base64}"
+                            "url": f"data:image/jpeg;base64,{image}"
                         }
                     }
                 ]
             }
         ],
         "temperature": 0.2,
-        "max_tokens": 500
+        "max_tokens": 1500
     }
 
     try:
 
         response = requests.post(
-            "https://openrouter.ai/api/v1/chat/completions",
+            API_URL,
             headers=headers,
             json=payload,
-            timeout=120
+            timeout=180
         )
 
         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(
+                "success": False,
+                "description": result.get(
+                    "error",
+                    {}
+                ).get(
                     "message",
                     "Unknown OpenRouter error."
                 )
@@ -103,21 +122,52 @@ Suggestions:
         if "choices" not in result:
 
             return {
-                "description": f"Unexpected response:\n{result}"
+                "success": False,
+                "description": "No choices returned.",
+                "raw": result
             }
 
-        content = (
-            result["choices"][0]
-            .get("message", {})
-            .get("content", "")
-        )
+        content = result["choices"][0]["message"]["content"]
 
-        return {
-            "description": content
-        }
+        content = content.replace("```json", "")
+        content = content.replace("```", "")
+        content = content.strip()
+
+        try:
+
+            analysis = json.loads(content)
+
+            analysis["success"] = True
+
+            return analysis
+
+        except Exception:
+
+            return {
+                "success": True,
+                "description": content,
+                "objects": [],
+                "people": "",
+                "style": "",
+                "lighting": "",
+                "colors": [],
+                "background": "",
+                "camera_angle": "",
+                "quality": "",
+                "editing_prompt": content,
+                "suggestions": [
+                    "Make it Anime",
+                    "Pixar Style",
+                    "Remove Background",
+                    "Replace Background",
+                    "Enhance Quality",
+                    "Change Colors"
+                ]
+            }
 
     except Exception as e:
 
         return {
+            "success": False,
             "description": str(e)
         }