Update app.py

app.py

@@ -3,7 +3,7 @@ import requests
 import json
 import os
 
-# Función para generar respuestas usando DeepSeek a través de DeepInfra
+# Función para generar respuestas usando DeepInfra
 def generate_response(user_message):
    try:
        if not user_message.strip():
@@ -13,7 +13,7 @@ def generate_response(user_message):
        with open("prompt.txt", "r", encoding="utf-8") as f:
            system_prompt = f.read().strip()
        
-       # Configurar la solicitud a DeepInfra (DeepSeek)
+       # Configurar la solicitud a DeepInfra
        api_key = os.environ.get("DEEPINFRA_API_KEY", "")
        if not api_key:
            return "Error: No se ha configurado la clave API. Por favor, configura la variable de entorno DEEPINFRA_API_KEY."
@@ -23,14 +23,15 @@ def generate_response(user_message):
            "Content-Type": "application/json"
        }
        
-       # Formato de prompt para DeepSeek
+       # Formato de prompt para modelos de chat
        messages = [
            {"role": "system", "content": system_prompt},
            {"role": "user", "content": user_message}
        ]
        
+       # Usar un modelo que sí está disponible en DeepInfra
        data = {
-           "model": "deepseek-ai/deepseek-chat-1.3b",  # Modelo pequeño de DeepSeek
+           "model": "meta-llama/Llama-2-7b-chat-hf",  # Modelo disponible en DeepInfra
            "messages": messages,
            "max_tokens": 500,
            "temperature": 0.7,
@@ -49,7 +50,19 @@ def generate_response(user_message):
            result = response.json()
            return result["choices"][0]["message"]["content"]
        else:
-           return f"Error en la API: {response.status_code} - {response.text}"
+           # Intentar con otro modelo si el primero falla
+           data["model"] = "mistralai/Mistral-7B-Instruct-v0.2"
+           response = requests.post(
+               "https://api.deepinfra.com/v1/openai/chat/completions",
+               headers=headers,
+               json=data
+           )
+           
+           if response.status_code == 200:
+               result = response.json()
+               return result["choices"][0]["message"]["content"]
+           else:
+               return f"Error en la API: {response.status_code} - {response.text}"
    
    except Exception as e:
        return f"Lo siento, ha ocurrido un error: {str(e)}"
@@ -75,4 +88,4 @@ demo = gr.Interface(
 
 # Lanzar la aplicación
 if __name__ == "__main__":
-   demo.queue(max_size=1).launch(share=False, debug=False)
+   demo.queue(max_size=1).launch(share=False, debug=False)