Update app.py

app.py

@@ -52,14 +52,12 @@ def run_local_h200(messages):
     return local_tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True)
 
 def run_groq(messages, model_id):
-    # Groq (Texto puro)
     for m in messages:
         if isinstance(m['content'], list):
              return "⚠️ Modelos Groq não veem imagens. Use Pixtral ou Gemini."
     
     if not groq_client: return "❌ Erro: Configure GROQ_API_KEY."
     
-    # Limpa formato para OpenAI
     clean_msgs = []
     for m in messages:
         clean_msgs.append({"role": m['role'], "content": m['content']})
@@ -87,7 +85,6 @@ def run_mistral(messages, model_id):
                 if item['type'] == 'text':
                     new_content.append({"type": "text", "text": item['text']})
                 elif item['type'] == 'image_url':
-                    # O router já mandou como data url ou path, vamos garantir
                     url = item['image_url']['url']
                     if url.startswith("data:image"):
                         new_content.append({"type": "image_url", "image_url": url})
@@ -136,25 +133,19 @@ def run_gemini(messages, model_id):
         return response.text
     except Exception as e: return f"❌ Gemini Error ({model_id}): {e}"
 
-# --- ROTEADOR INTELIGENTE (Compatível Gradio 4) ---
+# --- ROTEADOR ---
 def router(message, history, model_selector):
-    # Em Gradio 4 com multimodal=True, 'message' é um dict: {'text': '...', 'files': ['path']}
-    
-    # 1. Reconstrói histórico padronizado
+    # Formatação de histórico blindada
     formatted_history = []
     for user_msg, bot_msg in history:
-        # User (pode ter imagem antiga, mas Gradio 4 passa files separados no histórico visual)
-        # Simplificando: Assumimos texto no histórico visual do Gradio 4
-        if isinstance(user_msg, dict): # Caso raro no history
+        if isinstance(user_msg, dict):
              formatted_history.append({"role": "user", "content": user_msg.get("text")})
         else:
              formatted_history.append({"role": "user", "content": str(user_msg)})
-        
-        # Bot
         if bot_msg:
             formatted_history.append({"role": "assistant", "content": str(bot_msg)})
 
-    # 2. Formata a mensagem ATUAL (Onde a imagem está agora)
+    # Formatação da mensagem atual
     current_content = []
     text_input = message.get("text", "")
     files = message.get("files", [])
@@ -163,18 +154,16 @@ def router(message, history, model_selector):
         current_content.append({"type": "text", "text": text_input})
     
     for file_path in files:
-        # Adiciona imagem para processamento
         current_content.append({"type": "image_url", "image_url": {"url": file_path}})
 
-    # Se não tem imagem, manda string simples (melhor compatibilidade)
     if not files:
         formatted_history.append({"role": "user", "content": text_input})
     else:
         formatted_history.append({"role": "user", "content": current_content})
 
-    # 3. Roteamento
+    # Seleção de Modelo
     if "Gemini" in model_selector:
-        tid = "gemini-1.5-flash" # Default seguro
+        tid = "gemini-1.5-flash"
         if "3.0" in model_selector: tid = "gemini-3.0-pro-preview"
         elif "2.5 Pro" in model_selector: tid = "gemini-2.5-pro"
         elif "2.5 Flash" in model_selector: tid = "gemini-2.5-flash"
@@ -198,15 +187,15 @@ def router(message, history, model_selector):
     return "Modelo desconhecido."
 
 # --- INTERFACE ---
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🔀 APIDOST V7: Vision Update (Stable)")
+# AQUI ESTAVA O ERRO: Removi 'theme=gr.themes.Soft()'
+with gr.Blocks() as demo:
+    gr.Markdown("# 🔀 APIDOST V7 (Estável)")
     
     with gr.Row():
         model_dropdown = gr.Dropdown(
             choices=[
                 "✨ Google: Gemini 3.0 Pro (Experimental)",
                 "✨ Google: Gemini 2.5 Flash",
-                "✨ Google: Gemini 2.0 Flash (Exp)",
                 "☁️ Groq: Llama 3.3 70B",
                 "🇫🇷 Mistral: Pixtral Large (Vision) 🖼️",
                 "🇫🇷 Mistral: Large 2512 (Dez/25)",
@@ -217,13 +206,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             interactive=True
         )
 
-    # REMOVIDO type="messages" (Isso corrige o erro!)
     chat = gr.ChatInterface(
         fn=router,
         additional_inputs=[model_dropdown],
-        multimodal=True, # Isso ativa o upload de imagens no Gradio 4
+        multimodal=True,
     )
 
 if __name__ == "__main__":
-    # Configuração de rede padrão
     demo.launch(server_name="0.0.0.0", server_port=7860)