Update app.py

app.py

@@ -1,42 +1,68 @@
 import gradio as gr
 from groq import Groq
+import openai
 
 # Función para inicializar el cliente de Groq con la API key proporcionada
-def create_client(api_key):
+def create_groq_client(api_key):
     return Groq(api_key=api_key)
 
-# Función para verificar si la API está disponible enviando un mensaje "Hola"
-def verificar_api(api_key):
-    client = create_client(api_key)
+# Función para verificar la API de Groq
+def verificar_api_groq(api_key):
+    client = create_groq_client(api_key)
     messages = [
-        {
-            "role": "user",
-            "content": "Hola"
-        }
+        {"role": "user", "content": "Hola"}
     ]
 
     try:
-        # Enviar la petición a la API
         completion = client.chat.completions.create(
             messages=messages,
-            model="llama3-70b-8192",  # Cambia el modelo si es necesario
+            model="llama3-70b-8192",
             temperature=0.7,
             max_tokens=200,
             top_p=0.9,
             stream=False
         )
-        # Retornar la respuesta de la API
-        return f"La API está disponible. Respuesta: {completion.choices[0].message.content.strip()}"
+        return f"Groq API está disponible. Respuesta: {completion.choices[0].message.content.strip()}"
     except Exception as e:
         return f"Error: {e}"
 
+# Función para verificar la API de OpenAI
+def verificar_api_openai(api_key):
+    openai.api_key = api_key
+    messages = [
+        {"role": "user", "content": "Hola"}
+    ]
+
+    try:
+        response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",  # Puedes cambiar el modelo si tienes acceso a otro
+            messages=messages,
+            max_tokens=50,
+            temperature=0.7
+        )
+        return f"OpenAI API está disponible. Respuesta: {response.choices[0].message['content']}"
+    except Exception as e:
+        return f"Error: {e}"
+
+# Función general para verificar cualquiera de las APIs
+def verificar_api(api_key, api_type):
+    if api_type == "Groq":
+        return verificar_api_groq(api_key)
+    elif api_type == "OpenAI":
+        return verificar_api_openai(api_key)
+    else:
+        return "Tipo de API no soportado."
+
 # Interfaz de Gradio
 iface = gr.Interface(
     fn=verificar_api,
-    inputs="text",
+    inputs=[
+        gr.Textbox(label="API Key"),
+        gr.Radio(["Groq", "OpenAI"], label="Tipo de API")
+    ],
     outputs="text",
-    title="Verificación de API de Groq",
-    description="Ingresa tu API key para verificar si la API está disponible enviando un mensaje 'Hola'."
+    title="Verificación de APIs (Groq y OpenAI)",
+    description="Ingresa tu API key y selecciona el tipo de API para verificar si está disponible."
 )
 
 # Ejecutar la interfaz en Hugging Face Space