Update app.py

app.py

@@ -11,37 +11,30 @@ import google.generativeai as genai
 from huggingface_hub import snapshot_download
 
 # --- 1. SEGURANÇA (RATE LIMIT) ---
-# Simples e eficiente: bloqueia spammer sem quebrar o app.
 MAX_REQUESTS_PER_MINUTE = 15
 BLOCK_TIME_SECONDS = 60
 ip_tracker = {}
 
 def check_spam(request: gr.Request):
-    if not request: return True # Local run
+    if not request: return True 
     client_ip = request.client.host
     now = time.time()
     
-    # Limpa histórico antigo do IP
     if client_ip in ip_tracker:
         ip_tracker[client_ip] = [t for t in ip_tracker[client_ip] if now - t < BLOCK_TIME_SECONDS]
     
-    # Verifica bloqueio
     if client_ip in ip_tracker and len(ip_tracker[client_ip]) >= MAX_REQUESTS_PER_MINUTE:
         return False
         
-    # Registra
     if client_ip not in ip_tracker: ip_tracker[client_ip] = []
     ip_tracker[client_ip].append(now)
     return True
 
 # --- 2. CONFIGURAÇÕES GLOBAIS ---
-
-# LOCAL (H200 - ZeroGPU)
 LOCAL_MODEL_ID = "Qwen/Qwen2.5-Coder-32B-Instruct"
 local_model = None
 local_tokenizer = None
 
-# CLIENTES API
 groq_client = Groq(api_key=os.environ.get("GROQ_API_KEY")) if os.environ.get("GROQ_API_KEY") else None
 mistral_client = Mistral(api_key=os.environ.get("MISTRAL_API_KEY")) if os.environ.get("MISTRAL_API_KEY") else None
 if os.environ.get("GEMINI_API_KEY"):
@@ -60,7 +53,6 @@ def encode_image(image_path):
 def run_local_h200(messages):
     global local_model, local_tokenizer
     
-    # Validação rápida de imagem
     for m in messages:
         if isinstance(m['content'], list): return "⚠️ Modelo Local não suporta imagens. Use Gemini/Pixtral."
 
@@ -81,7 +73,6 @@ def run_groq(messages, model_id):
         if isinstance(m['content'], list): return "⚠️ Groq não suporta imagens. Use Gemini/Pixtral."
     if not groq_client: return "❌ Erro: GROQ_API_KEY ausente."
     
-    # Limpa formato para Groq
     clean_msgs = [{"role": m['role'], "content": m['content']} for m in messages]
     try:
         completion = groq_client.chat.completions.create(
@@ -93,7 +84,6 @@ def run_groq(messages, model_id):
 def run_mistral(messages, model_id):
     if not mistral_client: return "❌ Erro: MISTRAL_API_KEY ausente."
     
-    # Formata imagens para Mistral
     formatted_msgs = []
     for m in messages:
         content = m['content']
@@ -121,7 +111,6 @@ def run_gemini(messages, model_id):
         model = genai.GenerativeModel(model_id)
         chat_history = []
         
-        # Converte histórico para Gemini
         for m in messages[:-1]:
             role = "user" if m['role'] == "user" else "model"
             parts = []
@@ -135,7 +124,6 @@ def run_gemini(messages, model_id):
                         if os.path.exists(path): parts.append(Image.open(path))
             if parts: chat_history.append({"role": role, "parts": parts})
 
-        # Última mensagem
         last_parts = []
         lc = messages[-1]['content']
         if isinstance(lc, str): last_parts.append(lc)
@@ -153,29 +141,23 @@ def run_gemini(messages, model_id):
 
 # --- 5. ROTEADOR CENTRAL ---
 def router(message, history, model_selector, request: gr.Request):
-    # Check Spam
     if not check_spam(request):
         return "⛔ BLOQUEADO: Limite de mensagens excedido. Aguarde."
 
-    # Prepara Histórico (Blindado contra formatos variados do Gradio)
     messages = []
     if history:
         for turn in history:
-            # Formato antigo [user, bot]
             if isinstance(turn, (list, tuple)):
                 u_text = turn[0]
-                if isinstance(u_text, dict) and 'text' in u_text: u_text = u_text['text'] # Extrai texto se for dict
-                
+                if isinstance(u_text, dict) and 'text' in u_text: u_text = u_text['text']
                 messages.append({"role": "user", "content": str(u_text)})
                 if len(turn) > 1 and turn[1]:
                     messages.append({"role": "assistant", "content": str(turn[1])})
-            # Formato novo {role: user...}
             elif isinstance(turn, dict):
                 messages.append(turn)
 
-    # Prepara Mensagem Atual (Multimodal ou Texto)
     current_content = []
-    if isinstance(message, dict): # Multimodal
+    if isinstance(message, dict): 
         text = message.get("text", "")
         files = message.get("files", [])
         if text: current_content.append({"type": "text", "text": text})
@@ -183,13 +165,11 @@ def router(message, history, model_selector, request: gr.Request):
         
         if not files: messages.append({"role": "user", "content": text})
         else: messages.append({"role": "user", "content": current_content})
-    else: # Texto puro
+    else:
         messages.append({"role": "user", "content": str(message)})
 
-    # Roteamento
     print(f"🔀 Roteando para: {model_selector}")
 
-    # GEMINI
     if "Gemini" in model_selector:
         tid = "gemini-1.5-flash"
         if "3.0" in model_selector: tid = "gemini-3.0-pro-preview"
@@ -198,30 +178,28 @@ def router(message, history, model_selector, request: gr.Request):
         elif "2.0" in model_selector: tid = "gemini-2.0-flash-exp"
         return run_gemini(messages, tid)
     
-    # MISTRAL
     elif "Mistral" in model_selector:
         tid = "mistral-large-latest"
         if "Pixtral" in model_selector: tid = "pixtral-large-latest"
-        elif "2509" in model_selector: tid = "magistral-medium-2509" # <--- Seu Magistral VIP
+        elif "2509" in model_selector: tid = "magistral-medium-2509" 
         elif "2512" in model_selector: tid = "mistral-large-2512"
         elif "Codestral" in model_selector: tid = "codestral-2508"
         return run_mistral(messages, tid)
 
-    # GROQ
     elif "Groq" in model_selector:
         tid = "llama-3.3-70b-versatile"
-        if "120B" in model_selector: tid = "openai/gpt-oss-120b" # <--- GPT OSS 120B
-        elif "20B" in model_selector: tid = "openai/gpt-oss-20b"   # <--- GPT OSS 20B
+        if "120B" in model_selector: tid = "openai/gpt-oss-120b"
+        elif "20B" in model_selector: tid = "openai/gpt-oss-20b"
         return run_groq(messages, tid)
         
-    # LOCAL
     elif "H200" in model_selector:
         return run_local_h200(messages)
         
     return "⚠️ Modelo não reconhecido."
 
-# --- 6. INTERFACE ---
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
+# --- 6. INTERFACE (SEM TEMA - CORREÇÃO) ---
+# Berta: Removi theme=gr.themes.Soft() pois sua versão do Gradio não suporta.
+with gr.Blocks() as demo:
     gr.Markdown("# 🔀 APIDOST v8: The Arsenal")
     
     with gr.Row():
@@ -245,7 +223,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             interactive=True
         )
     
-    # Berta: multimodal=True é vital para as imagens funcionarem no seu index.html
     chat = gr.ChatInterface(
         fn=router, 
         additional_inputs=[model_dropdown],
@@ -253,8 +230,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     )
 
 if __name__ == "__main__":
-    # Pré-download do modelo local para não travar no primeiro uso
     try: snapshot_download(repo_id=LOCAL_MODEL_ID)
     except: pass
-    
     demo.queue().launch()