Update app.py

app.py

@@ -1,39 +1,17 @@
 import gradio as gr
 import requests
 
-def pregunta_law_guatemala(pregunta):
-    # Definir los datos necesarios para la llamada a la API
+def ask_question(question):
     url = "https://api.afforai.com/api/api_completion"
-    headers = {"Content-Type": "application/json"}
-    payload = {
+    data = {
         "apiKey": "fcbfdfe8-e9ed-41f3-a7d8-b6587538e84e",
         "sessionID": "65489d7c9ad727940f2ab26f",
-        "history": [{"role": "user", "content": pregunta}],
+        "history": [{"role": "user", "content": question}],
         "powerful": False,
         "google": True
     }
+    response = requests.post(url, json=data)
+    return response.json()
 
-    # Realizar la solicitud POST a la API
-    response = requests.post(url, json=payload, headers=headers)
-
-    # Verificar si la solicitud fue exitosa
-    if response.status_code == 200:
-        data = response.json()
-        respuesta = data["completion"]
-    else:
-        respuesta = "Ocurrió un error al procesar la pregunta. Inténtalo de nuevo más tarde."
-
-    return respuesta
-
-# Crear la interfaz Gradio
-iface = gr.Interface(
-    fn=pregunta_law_guatemala,
-    inputs="text",
-    outputs="text",
-    title="Preguntas sobre las leyes de Guatemala",
-    description="Haz una pregunta sobre las leyes de Guatemala y obtén una respuesta.",
-    theme="compact"
-)
-
-# Ejecutar la aplicación
+iface = gr.Interface(fn=ask_question, inputs="text", outputs="text")
 iface.launch()