Update app.py

app.py

@@ -2,7 +2,9 @@ import gradio as gr
 import spaces
 import torch
 import os
+import time
 import base64
+from collections import defaultdict
 from PIL import Image
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from groq import Groq
@@ -10,6 +12,32 @@ from mistralai import Mistral
 import google.generativeai as genai
 from huggingface_hub import snapshot_download 
 
+# --- CONFIGURAÇÃO DE SEGURANÇA (RATE LIMIT) ---
+# Berta: Configuração para impedir que engraçadinhos queimem sua API.
+# Permite 5 mensagens a cada 60 segundos por IP.
+RATE_LIMIT_MAX_REQUESTS = 5
+RATE_LIMIT_WINDOW_SECONDS = 600
+ip_tracker = defaultdict(list)
+
+def check_rate_limit(request: gr.Request):
+    """Verifica se o IP está abusando."""
+    if not request:
+        return True # Se rodar local sem request, libera.
+    
+    client_ip = request.client.host
+    current_time = time.time()
+    
+    # Limpa requisições antigas desse IP
+    ip_tracker[client_ip] = [t for t in ip_tracker[client_ip] if current_time - t < RATE_LIMIT_WINDOW_SECONDS]
+    
+    # Verifica se excedeu
+    if len(ip_tracker[client_ip]) >= RATE_LIMIT_MAX_REQUESTS:
+        return False
+    
+    # Registra nova requisição
+    ip_tracker[client_ip].append(current_time)
+    return True
+
 # --- CONFIGURAÇÕES ---
 LOCAL_MODEL_ID = "Qwen/Qwen2.5-Coder-32B-Instruct"
 local_model = None
@@ -151,33 +179,28 @@ def run_gemini(messages, model_id):
         return response.text
     except Exception as e: return f"❌ Gemini Error ({model_id}): {e}"
 
-# --- ROTEADOR (AGORA BLINDADO!) ---
-def router(message, history, model_selector):
+# --- ROTEADOR COM PROTEÇÃO DE IP ---
+def router(message, history, model_selector, request: gr.Request): # <--- Request injetado pelo Gradio
+    
+    # 1. VERIFICAÇÃO DE SPAM
+    if not check_rate_limit(request):
+        return f"⛔ **BLOQUEIO DE SEGURANÇA:** Você excedeu o limite de requisições ({RATE_LIMIT_MAX_REQUESTS}/min). Aguarde um pouco."
+
     formatted_history = []
     
-    # --- BERTA FIX: Tratamento Universal de Histórico ---
-    # Isso resolve o erro "too many values to unpack"
+    # Tratamento de histórico blindado
     for turn in history:
-        # CASO 1: Formato Antigo [[user, bot]]
         if isinstance(turn, (list, tuple)) and len(turn) >= 2:
              user_content = turn[0]
              bot_content = turn[1]
-             
-             # Extrai texto se for complexo
              if isinstance(user_content, dict) and 'text' in user_content:
                  user_content = user_content['text']
-                 
              formatted_history.append({"role": "user", "content": str(user_content)})
              if bot_content:
                  formatted_history.append({"role": "assistant", "content": str(bot_content)})
-        
-        # CASO 2: Formato Novo/Messages (Dicionário)
         elif isinstance(turn, dict):
-             # Já está no formato certo, só copiamos
              formatted_history.append(turn)
     
-    # --- FIM DO FIX ---
-
     # Processa mensagem ATUAL
     current_content = []
     text_input = message.get("text", "")
@@ -221,7 +244,7 @@ def router(message, history, model_selector):
 
 # --- INTERFACE ---
 with gr.Blocks() as demo:
-    gr.Markdown("# 🔀 APIDOST (Robust Mode)")
+    gr.Markdown("# 🔀 APIDOST (Protected Mode)")
     
     with gr.Row():
         model_dropdown = gr.Dropdown(