Add files via upload

ai.py

@@ -0,0 +1,99 @@
+#!/usr/bin/env python3
+import os
+import sys
+import base64
+import requests
+import time
+
+# --- DEFAULT FALLBACK MODELS ---
+DEFAULT_MODELS = [
+    "qwen/qwen-2.5-vl-7b-instruct:free",
+    "meta-llama/llama-3.2-11b-vision-instruct",
+    "google/gemini-2.5-flash-image",
+]
+
+# --- ARGUMENTS ---
+if len(sys.argv) < 3:
+    print("Kullanım: ai.py <image_path> <prompt> [model1,model2,...]")
+    sys.exit(1)
+
+image_path = sys.argv[1]
+prompt = sys.argv[2]
+
+# --- MODELS FROM UI (OPTIONAL) ---
+if len(sys.argv) >= 4 and sys.argv[3].strip():
+    MODELS = [m.strip() for m in sys.argv[3].split(",")]
+else:
+    MODELS = DEFAULT_MODELS
+
+# --- API KEY ---
+API_KEY = os.environ.get("OPENROUTER_API_KEY")
+if not API_KEY:
+    print("HATA: OPENROUTER_API_KEY not set")
+    sys.exit(1)
+
+# --- IMAGE LOAD ---
+if not os.path.exists(image_path):
+    print(f"HATA: Resim bulunamadı → {image_path}")
+    sys.exit(1)
+
+with open(image_path, "rb") as f:
+    image_b64 = base64.b64encode(f.read()).decode("utf-8")
+
+# --- REQUEST CONFIG ---
+URL = "https://openrouter.ai/api/v1/chat/completions"
+
+HEADERS = {
+    "Authorization": f"Bearer {API_KEY}",
+    "Content-Type": "application/json",
+    "HTTP-Referer": "http://localhost",
+    "X-Title": "ai-capture",
+}
+
+# --- MODEL TRY FUNCTION ---
+def try_model(model_name):
+    payload = {
+        "model": model_name,
+        "messages": [
+            {
+                "role": "user",
+                "content": [
+                    {"type": "text", "text": prompt},
+                    {
+                        "type": "image_url",
+                        "image_url": {
+                            "url": f"data:image/png;base64,{image_b64}"
+                        }
+                    }
+                ],
+            }
+        ],
+        "max_tokens": 512,
+    }
+
+    response = requests.post(
+        URL,
+        headers=HEADERS,
+        json=payload,
+        timeout=60,
+    )
+
+    if response.status_code == 200:
+        data = response.json()
+        return data["choices"][0]["message"]["content"]
+
+    raise Exception(f"{model_name} → {response.status_code}: {response.text}")
+
+
+# --- MAIN LOOP ---
+for model in MODELS:
+    try:
+        answer = try_model(model)
+        print(answer)
+        sys.exit(0)
+    except Exception as e:
+        print(f"[FAIL] {e}", file=sys.stderr)
+        time.sleep(1)
+
+print("HATA: Tüm modeller başarısız")
+sys.exit(1)