Upload 3 files

app.py

@@ -9,32 +9,43 @@ def subir_csv(file):
     df_global = pd.read_csv(file.name)
     return f"Archivo cargado con {len(df_global)} filas y {len(df_global.columns)} columnas."
 
-def analizar_pregunta(pregunta):
+def responder_pregunta(pregunta):
     global df_global
     if df_global is None:
         return "Primero debes subir un archivo CSV."
 
-    if "promedio" in pregunta.lower():
+    pregunta = pregunta.lower()
+
+    if "promedio" in pregunta:
         numeric_cols = df_global.select_dtypes(include='number')
         if numeric_cols.empty:
             return "No hay columnas numéricas para calcular el promedio."
         return numeric_cols.mean().to_string()
-    elif "columnas" in pregunta.lower():
-        return f"Las columnas del archivo son: {', '.join(df_global.columns)}"
-    else:
-        return "Pregunta no reconocida. Prueba con: ¿Cuál es el promedio de las columnas?"
 
-demo = gr.Blocks()
+    elif "columnas" in pregunta:
+        return f"Las columnas del archivo son: {', '.join(df_global.columns)}"
 
-with demo:
-    gr.Markdown("# Analizador Inteligente de CSV")
-    archivo = gr.File(label="Sube tu archivo CSV")
-    salida = gr.Textbox(label="Resultado de carga")
-    pregunta = gr.Textbox(label="Haz una pregunta sobre el CSV")
-    respuesta = gr.Textbox(label="Respuesta del análisis")
+    elif "filas" in pregunta:
+        return f"El archivo contiene {len(df_global)} filas."
 
-    archivo.upload(subir_csv, archivo, salida)
-    pregunta.submit(analizar_pregunta, pregunta, respuesta)
+    else:
+        return "No entendí la pregunta. Prueba con: ¿Cuál es el promedio?, ¿Cuáles son las columnas?, ¿Cuántas filas hay?"
+
+def app_interface(csv_file, pregunta):
+    mensaje = subir_csv(csv_file)
+    respuesta = responder_pregunta(pregunta)
+    return mensaje + "\n\n" + respuesta
+
+iface = gr.Interface(
+    fn=app_interface,
+    inputs=[
+        gr.File(label="Sube tu archivo CSV"),
+        gr.Textbox(label="Pregunta sobre el archivo")
+    ],
+    outputs=gr.Textbox(label="Respuesta"),
+    title="Analizador de CSV",
+    description="Sube un archivo CSV y haz preguntas como '¿Cuál es el promedio?' o '¿Cuáles son las columnas?'"
+)
 
 if __name__ == "__main__":
-    demo.launch()
+    iface.launch()