Update app.py

app.py

@@ -1,106 +1,79 @@
 import gradio as gr
 import requests
 import json
-import time
 
-def chat_deepseek(message, history):
+def chat_with_ai(message, history):
     """
-    Função para conversar com DeepSeek-V3
+    Função para conversar com um modelo menor e mais estável
     """
     try:
-        # URL da API do Hugging Face para DeepSeek-V3
-        API_URL = "https://api-inference.huggingface.co/models/deepseek-ai/DeepSeek-V3"
+        # Usando um modelo menor que funciona melhor
+        API_URL = "https://api-inference.huggingface.co/models/microsoft/DialoGPT-medium"
         
-        # Preparar o prompt
-        conversation = ""
-        for human, assistant in history:
-            conversation += f"Human: {human}\nAssistant: {assistant}\n"
-        conversation += f"Human: {message}\nAssistant:"
-        
-        # Configurar a requisição
+        # Preparar payload simples
         payload = {
-            "inputs": conversation,
+            "inputs": message,
             "parameters": {
-                "max_new_tokens": 500,
-                "temperature": 0.7,
-                "do_sample": True,
-                "return_full_text": False
+                "max_length": 200,
+                "temperature": 0.7
             }
         }
         
-        headers = {
-            "Content-Type": "application/json"
-        }
-        
-        # Fazer a requisição
-        response = requests.post(API_URL, headers=headers, json=payload, timeout=60)
+        response = requests.post(API_URL, json=payload, timeout=30)
         
         if response.status_code == 200:
             result = response.json()
             if isinstance(result, list) and len(result) > 0:
-                generated_text = result[0].get('generated_text', '')
-                # Limpar a resposta
-                generated_text = generated_text.strip()
-                if generated_text:
-                    return generated_text
-                else:
-                    return "Desculpe, não consegui gerar uma resposta válida."
-            else:
-                return "Formato de resposta inesperado da API."
+                return result[0].get('generated_text', 'Não consegui gerar resposta.')
+            return "Formato inesperado de resposta."
         
         elif response.status_code == 503:
-            return "🔄 Modelo está carregando... Tente novamente em alguns segundos."
+            return "🔄 Modelo carregando... aguarde alguns segundos e tente novamente."
         
         else:
-            return f"❌ Erro na API: {response.status_code} - {response.text}"
+            return f"Erro {response.status_code}: {response.text}"
             
-    except requests.exceptions.Timeout:
-        return "⏱️ Timeout - O modelo demorou muito para responder. Tente uma pergunta mais simples."
-    
     except Exception as e:
-        return f"❌ Erro: {str(e)}"
+        return f"Erro: {str(e)}"
+
+# Interface mais simples
+def respond(message, chat_history):
+    if not message.strip():
+        return chat_history, ""
+    
+    # Resposta temporária
+    chat_history.append([message, "🤔 Pensando..."])
+    
+    # Obter resposta da IA
+    try:
+        bot_response = chat_with_ai(message, chat_history)
+        chat_history[-1][1] = bot_response
+    except:
+        chat_history[-1][1] = "Desculpe, ocorreu um erro. Tente novamente."
+    
+    return chat_history, ""
 
-# Criar a interface do chat
-with gr.Blocks(title="DeepSeek-V3 Chat", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🤖 Chat com DeepSeek-V3")
-    gr.Markdown("*Modelo pode demorar para carregar na primeira vez*")
+# Criar interface
+with gr.Blocks(title="Chat IA", theme=gr.themes.Default()) as demo:
+    gr.Markdown("# 🤖 Chat com IA")
+    gr.Markdown("*Testando modelos de linguagem*")
     
-    chatbot = gr.Chatbot(
-        height=500,
-        placeholder="As mensagens aparecerão aqui..."
-    )
+    chatbot = gr.Chatbot(height=400)
     
     with gr.Row():
         msg = gr.Textbox(
-            placeholder="Digite sua mensagem aqui...",
+            placeholder="Digite sua mensagem...",
             container=False,
-            scale=7
+            scale=4
         )
-        send_btn = gr.Button("Enviar", scale=1, variant="primary")
-        clear_btn = gr.Button("Limpar", scale=1)
+        send = gr.Button("Enviar", scale=1)
     
-    # Função para processar mensagem
-    def respond(message, chat_history):
-        if not message.strip():
-            return chat_history, ""
-        
-        # Adicionar mensagem do usuário ao histórico
-        chat_history.append([message, "🤔 Pensando..."])
-        
-        # Obter resposta do DeepSeek
-        bot_message = chat_deepseek(message, chat_history[:-1])
-        
-        # Atualizar com a resposta real
-        chat_history[-1][1] = bot_message
-        
-        return chat_history, ""
+    clear = gr.Button("Limpar Chat")
     
-    # Conectar eventos
-    send_btn.click(respond, [msg, chatbot], [chatbot, msg])
+    # Eventos
+    send.click(respond, [msg, chatbot], [chatbot, msg])
     msg.submit(respond, [msg, chatbot], [chatbot, msg])
-    clear_btn.click(lambda: ([], ""), outputs=[chatbot, msg])
+    clear.click(lambda: [], outputs=chatbot)
 
-# Lançar a aplicação
 if __name__ == "__main__":
     demo.launch()
-