Spaces:

Lei075fr
/

Fortune-API-Shield

Sleeping

App Files Files Community

Lei075fr commited on Apr 25

Commit

192c3a9

verified ·

1 Parent(s): f8ccb73

Update app.py

Browse files

Files changed (1) hide show

app.py +30 -64

app.py CHANGED Viewed

@@ -8,21 +8,21 @@ app = FastAPI()
 KEYS = [os.getenv(f"OR_KEY_{i}") for i in range(1, 16)]
 KEYS = [k for k in KEYS if k and k.strip()]
-# MAPEAMENTO UNIFICADO ARIA (Com sufixos únicos para o Python)
 MODEL_MAP = {
-    "aria_med": "google/gemini-2.5-pro:free",        # Vitalis
-    "aria_logic": "qwen/qwen3-coder-480b:free",      # Nexos
-    "aria_vision": "nvidia/nemotron-nano-12b-2-vl:free", # Nutrilens
-    "aria_brain": "meta-llama/llama-3.3-70b-instruct:free", # Cérebro Principal
-    "aria_default": "qwen/qwen3-coder:free"
 }
 @app.get("/", response_class=HTMLResponse)
 async def root():
     status = "🟢 SISTEMA OPERACIONAL" if KEYS else "🔴 FALHA NAS CHAVES"
     return f"""
-    <body style='background:#0a0a0a; color:#00ff99; font-family:monospace; text-align:center; padding-top:50px;'>
-        <h1 style='border: 2px solid #00ff99; display:inline-block; padding:20px;'>🛡️ ARIA SOVEREIGN SHIELD V2.5</h1>
         <p style='font-size:1.5em;'>Chaves Ativas: {len(KEYS)} | Local: Beira, MZ</p>
         <p style='color:#555;'>Status: {status}</p>
     </body>
@@ -31,64 +31,30 @@ async def root():
 @app.post("/v1/chat/completions")
 async def proxy(request: Request):
     if not KEYS:
-        return Response(content='{"error": "Adicione as chaves nos Secrets do HF!"}', status_code=500)
     body = await request.json()
-    incoming_model = body.get("model", "")
-    msg_content = str(body.get("messages", "")).lower()
-    # --- LOGICA DE ROTEAMENTO ---
-    if "fortune/" in incoming_model:
-        # Extrai o modelo real enviado pelo CEO
-        target = incoming_model.replace("fortune/", "", 1)
-    else:
-        # Seleção por contexto se o CEO não especificar fortune/
-        if any(word in msg_content for word in ["medicina", "vitalis", "paciente"]):
-            target = MODEL_MAP["aria_med"]
-        elif any(word in msg_content for word in ["nexos", "venda", "gestão", "erp"]):
-            target = MODEL_MAP["aria_logic"]
-        elif any(word in msg_content for word in ["nutrilens", "vejo", "câmara", "visão"]):
-            target = MODEL_MAP["aria_vision"]
-        else:
-            target = MODEL_MAP["aria_brain"]
-    # Correção de Segurança: Garante que modelos free tenham o sufixo :free
-    if "instruct" in target and not target.endswith(":free"):
-        target = f"{target}:free"
-    body["model"] = target
-    print(f"📡 Roteando para: {target}")
-    # --- CICLO DE RESILIÊNCIA ---
     async with httpx.AsyncClient() as client:
-        for attempt in range(5):
-            selected_key = random.choice(KEYS)
-            headers = {
-                "Authorization": f"Bearer {selected_key}",
-                "X-Title": "Aria_Sovereign",
-                "Content-Type": "application/json"
-            }
-            try:
-                resp = await client.post(
-                    "https://openrouter.ai/api/v1/chat/completions",
-                    json=body, headers=headers, timeout=120.0
-                )
-                if resp.status_code == 200:
-                    return Response(content=resp.content, status_code=200)
-                print(f"⚠️ Tentativa {attempt+1} falhou (Status {resp.status_code}). Rodando chave...")
-            except Exception as e:
-                print(f"❌ Erro de rede: {str(e)}")
-                await asyncio.sleep(0.3)
-                continue
-        # Fallback Final para o roteador livre
-        body["model"] = "openrouter/free"
-        emerg = await client.post("https://openrouter.ai/api/v1/chat/completions", json=body, headers=headers, timeout=60.0)
-        return Response(content=emerg.content, status_code=emerg.status_code)
-if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)

 KEYS = [os.getenv(f"OR_KEY_{i}") for i in range(1, 16)]
 KEYS = [k for k in KEYS if k and k.strip()]
+# MAPEAMENTO UNIFICADO (Traduz os teus apelidos para os IDs do OpenRouter)
 MODEL_MAP = {
+    "aria_med": "google/gemini-2.5-pro:free",
+    "aria_logic": "qwen/qwen3-coder-480b:free",
+    "aria_vision": "nvidia/nemotron-nano-12b-2-vl:free",
+    "aria_brain": "meta-llama/llama-3.3-70b-instruct:free",
+    "aria_default": "qwen/qwen3-coder-480b:free"
 }
 @app.get("/", response_class=HTMLResponse)
 async def root():
     status = "🟢 SISTEMA OPERACIONAL" if KEYS else "🔴 FALHA NAS CHAVES"
     return f"""
+    <body style='background:#0a0a0a; color:#00ff88; font-family:sans-serif; text-align:center; padding-top:50px;'>
+        <h1 style='border: 2px solid #00ff88; display:inline-block; padding:20px;'>ARIA SOVEREIGN SHIELD V2.5</h1>
         <p style='font-size:1.5em;'>Chaves Ativas: {len(KEYS)} | Local: Beira, MZ</p>
         <p style='color:#555;'>Status: {status}</p>
     </body>
 @app.post("/v1/chat/completions")
 async def proxy(request: Request):
     if not KEYS:
+        return Response(content='{"error": "No API keys configured"}', status_code=500)
     body = await request.json()
+    model_requested = body.get("model", "")
+    # LÓGICA DE LIMPEZA: Remove o prefixo "fortune/" se a Aria o enviar
+    clean_model = model_requested.replace("fortune/", "")
+    # TRADUÇÃO: Se for um apelido, usa o ID real. Se não, usa o que foi enviado.
+    target_model = MODEL_MAP.get(clean_model, clean_model)
+    body["model"] = target_model
+    # ROTAÇÃO DE CHAVES: Escolhe uma das 15 chaves aleatoriamente para evitar bloqueios
+    headers = {
+        "Authorization": f"Bearer {random.choice(KEYS)}",
+        "HTTP-Referer": "https://github.com/Fortnee/Ariaaa",
+        "Content-Type": "application/json"
+    }
     async with httpx.AsyncClient() as client:
+        response = await client.post(
+            "https://openrouter.ai/api/v1/chat/completions",
+            json=body,
+            headers=headers,
+            timeout=60.0
+        )
+        return Response(content=response.content, status_code=response.status_code)