Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 import requests
-import os
 import json
+import os
 
-# Función para generar respuestas usando una API externa
+# Función para generar respuestas usando DeepSeek a través de DeepInfra
 def generate_response(user_message):
    try:
        if not user_message.strip():
@@ -13,29 +13,33 @@ def generate_response(user_message):
        with open("prompt.txt", "r", encoding="utf-8") as f:
            system_prompt = f.read().strip()
        
-       # Configurar la solicitud a la API (ejemplo con OpenAI)
-       api_key = os.environ.get("OPENAI_API_KEY", "")
+       # Configurar la solicitud a DeepInfra (DeepSeek)
+       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 OPENAI_API_KEY."
+           return "Error: No se ha configurado la clave API. Por favor, configura la variable de entorno DEEPINFRA_API_KEY."
        
        headers = {
            "Authorization": f"Bearer {api_key}",
            "Content-Type": "application/json"
        }
        
+       # Formato de prompt para DeepSeek
+       messages = [
+           {"role": "system", "content": system_prompt},
+           {"role": "user", "content": user_message}
+       ]
+       
        data = {
-           "model": "gpt-3.5-turbo",
-           "messages": [
-               {"role": "system", "content": system_prompt},
-               {"role": "user", "content": user_message}
-           ],
+           "model": "deepseek-ai/deepseek-chat-1.3b",  # Modelo pequeño de DeepSeek
+           "messages": messages,
+           "max_tokens": 500,
            "temperature": 0.7,
-           "max_tokens": 500
+           "top_p": 0.9
        }
        
        # Enviar solicitud a la API
        response = requests.post(
-           "https://api.openai.com/v1/chat/completions",
+           "https://api.deepinfra.com/v1/openai/chat/completions",
            headers=headers,
            json=data
        )
@@ -65,7 +69,8 @@ demo = gr.Interface(
        ["¿Cuál es la mejor estrategia para conseguir mis primeros clientes?"],
        ["¿Debería invertir en publicidad en redes sociales o en SEO?"]
    ],
-   allow_flagging="never"
+   allow_flagging="never",
+   theme=gr.themes.Soft(primary_hue="blue")
 )
 
 # Lanzar la aplicación