Spaces:

Lei075fr
/

Fortune-API-Shield

Sleeping

App Files Files Community

Lei075fr commited on about 1 month ago

Commit

f8ccb73

verified ·

1 Parent(s): 20e3a21

Update app.py

Browse files

Files changed (1) hide show

app.py +35 -46

app.py CHANGED Viewed

@@ -4,78 +4,72 @@ from fastapi.responses import HTMLResponse
 app = FastAPI()
-# Pool de 15 Chaves - O Arsenal Soberano da Aria
 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 - ECOSSISTEMA ARIA (2026)
-# Usamos sufixos para que o Python aceite as chaves únicas
 MODEL_MAP = {
-    "aria_med": "google/gemini-2.5-pro:free",        # Medicina/Protocolos (Vitalis)
-    "aria_logic": "qwen/qwen3-coder-480b:free",      # Gestão/ERP/Lógica (Nexos)
-    "aria_vision": "nvidia/nemotron-nano-12b-2-vl:free", # Visão/Saúde (Nutrilens)
-    "aria_brain": "meta-llama/llama-3.3-70b-instruct:free", # Cérebro padrão da Aria
-    "aria_default": "qwen/qwen3-coder:free"          # Coding e automação
 }
 @app.get("/", response_class=HTMLResponse)
 async def root():
-    status = "🟢 ATIVO" if KEYS else "🔴 SEM CHAVES"
     return f"""
-    <body style='background:#0f0f0f; color:#00ff99; font-family:monospace; text-align:center; padding-top:50px;'>
-        <h1 style='border-bottom: 2px solid #00ff99; display:inline-block; padding-bottom:10px;'>🛡️ Escudo Fortune MAX: ARIA EDITION</h1>
-        <p style='font-size:1.5em;'>Status: {status} | Chaves em Rotação: {len(KEYS)}</p>
-        <p style='color:#666;'>Ponte de Guerrilha: Beira -> OpenRouter -> Aria Ecosystem</p>
     </body>
     """
 @app.post("/v1/chat/completions")
 async def proxy(request: Request):
     if not KEYS:
-        return Response(content='{"error": "Configure OR_KEY_1 até 15 nos Secrets!"}', status_code=500)
     body = await request.json()
     incoming_model = body.get("model", "")
     msg_content = str(body.get("messages", "")).lower()
-    # --- ROTEADOR INTELIGENTE ARIA ---
-    # 1. Ordem Direta (Prefixo Fortune)
     if "fortune/" in incoming_model:
         target = incoming_model.replace("fortune/", "", 1)
-        print(f"🎯 Comando Manual: {target}")
-    # 2. Seleção Automática (Aria detecta o contexto)
     else:
-        # Detecta palavras-chave para trocar de "especialização"
-        if any(word in msg_content for word in ["vitalis", "medicina", "paciente", "diagnóstico"]):
             target = MODEL_MAP["aria_med"]
-        elif any(word in msg_content for word in ["nexos", "erp", "gestão", "stock", "venda"]):
             target = MODEL_MAP["aria_logic"]
-        elif any(word in msg_content for word in ["nutrilens", "visão", "vejo", "imagem"]):
             target = MODEL_MAP["aria_vision"]
-        elif "aria" in msg_content:
-            target = MODEL_MAP["aria_brain"]
         else:
-            target = MODEL_MAP["aria_default"]
-    body["model"] = target
-    # --- CICLO DE ROTAÇÃO E SOBREVIVÊNCIA ---
-    max_retries = 5
     async with httpx.AsyncClient() as client:
-        for attempt in range(max_retries):
             selected_key = random.choice(KEYS)
             headers = {
                 "Authorization": f"Bearer {selected_key}",
-                "X-Title": "Aria_Sovereign_System",
-                "HTTP-Referer": "https://fortune-dev-moz.io",
                 "Content-Type": "application/json"
             }
             try:
-                # Timeout de 120s para modelos massivos
                 resp = await client.post(
                     "https://openrouter.ai/api/v1/chat/completions",
                     json=body, headers=headers, timeout=120.0
@@ -84,22 +78,17 @@ async def proxy(request: Request):
                 if resp.status_code == 200:
                     return Response(content=resp.content, status_code=200)
-                if resp.status_code in [429, 402, 502, 503, 504]:
-                    print(f"⚠️ Chave bloqueada. Tentando nova chave ({attempt+1}/{max_retries})...")
-                    continue
             except Exception as e:
-                print(f"⚠️ Erro na tentativa {attempt+1}: {str(e)}")
                 await asyncio.sleep(0.3)
                 continue
-        # --- MODO DE EMERGÊNCIA ---
         body["model"] = "openrouter/free"
-        try:
-            emerg_resp = await client.post("https://openrouter.ai/api/v1/chat/completions", json=body, headers=headers, timeout=60.0)
-            return Response(content=emerg_resp.content, status_code=emerg_resp.status_code)
-        except:
-            return Response(content='{"error": "Falha crítica no Escudo. Reinicie as chaves."}', status_code=503)
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)

 app = FastAPI()
+# Pool de 15 Chaves - Arsenal Aria
 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>
     """
 @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)