Update app.py

app.py

@@ -1,3 +1,5 @@
+# app.py
+
 import os
 import gradio as gr
 import requests
@@ -123,15 +125,36 @@ Ejemplo de prompt (usa como guía para estructura):
 Mantén respuestas concisas y enfocadas.
 """
 
+# ==========================
+# FUNCIÓN PARA OBTENER API KEY
+# ==========================
+
+def get_api_key():
+    """Obtiene la API key desde variables de entorno/secretos"""
+    # Para Hugging Face Spaces
+    api_key = os.environ.get("SAMBANOVA_API_KEY")
+    
+    # Si no está en variables de entorno, intenta desde secrets (para Gradio)
+    if not api_key:
+        try:
+            # Esto funciona en Hugging Face Spaces con secretos
+            api_key = os.environ.get("SAMBANOVA_API_KEY")
+        except:
+            pass
+            
+    return api_key
+
 # ==========================
 # FUNCIÓN DE CHAT COMPATIBLE CON HF SPACES
 # ==========================
 
-def chat_voyeur(message, chat_history, api_key):
+def chat_voyeur(message, chat_history):
+    api_key = get_api_key()
+    
     if not api_key:
         # Formato de diccionario para HF Spaces
         chat_history.append({"role": "user", "content": message})
-        chat_history.append({"role": "assistant", "content": "❌ Ingresa tu API Key"})
+        chat_history.append({"role": "assistant", "content": "❌ API Key no configurada. Por favor, configura la variable de entorno SAMBANOVA_API_KEY."})
         return "", chat_history
 
     client = SambaNovaClient(api_key)
@@ -159,9 +182,11 @@ def chat_voyeur(message, chat_history, api_key):
 # PRUEBA DE API
 # ==========================
 
-def test_sambanova_api(api_key):
+def test_sambanova_api():
+    api_key = get_api_key()
+    
     if not api_key:
-        return "❌ Ingresa tu API Key primero"
+        return "❌ API Key no configurada. Configura la variable de entorno SAMBANOVA_API_KEY."
 
     client = SambaNovaClient(api_key)
     test_messages = [
@@ -181,12 +206,7 @@ def create_interface():
 
         with gr.Row():
             with gr.Column(scale=1):
-                api_key_input = gr.Textbox(
-                    label="🔑 API Key de SambaNova",
-                    placeholder="Ingresa tu API key...",
-                    type="password",
-                )
-                test_btn = gr.Button("🧪 Probar API")
+                test_btn = gr.Button("🧪 Probar Conexión API")
                 test_output = gr.Textbox(label="Resultado de prueba", interactive=False)
                 gr.Markdown("### 📋 Especialidades")
                 for specialty in VOYEUR_SPECIALIST_CONFIG["specialties"]:
@@ -194,6 +214,11 @@ def create_interface():
                 gr.Markdown("### 🎯 Experiencia Técnica")
                 for expertise in VOYEUR_SPECIALIST_CONFIG["technical_expertise"]:
                     gr.Markdown(f"• {expertise}")
+                
+                # Información sobre configuración
+                gr.Markdown("### ⚙️ Configuración")
+                gr.Markdown("API Key configurada desde variables de entorno")
+                
             with gr.Column(scale=2):
                 # Chatbot sin parámetros problemáticos
                 chatbot = gr.Chatbot(
@@ -222,7 +247,7 @@ def create_interface():
 
         test_btn.click(
             test_sambanova_api,
-            inputs=[api_key_input],
+            inputs=[],
             outputs=[test_output],
         )
 
@@ -239,7 +264,7 @@ def create_interface():
             outputs=[msg, chatbot],
         ).then(
             chat_voyeur,
-            inputs=[msg, chatbot, api_key_input],
+            inputs=[msg, chatbot],
             outputs=[msg, chatbot],
         )
 
@@ -250,7 +275,7 @@ def create_interface():
             outputs=[msg, chatbot],
         ).then(
             chat_voyeur,
-            inputs=[msg, chatbot, api_key_input],
+            inputs=[msg, chatbot],
             outputs=[msg, chatbot],
         )
 
@@ -261,5 +286,5 @@ def create_interface():
 
 if __name__ == "__main__":
     demo = create_interface()
-    # CORRECCIÓN: Eliminar share=True para Hugging Face Spaces
+    # Para Hugging Face Spaces, no usar share=True
     demo.launch(debug=True)