Update app.py

app.py

@@ -22,4 +22,24 @@ PINECONE_API_KEY = userdata.get("PINECONE_API_KEY")  # Use the key you set in th
 PINECONE_ENVIRONMENT = "us-east-1"  # Use the environment you set in the secrets
 
 # Initialize Pinecone with the API key
-pc = Pinecone(api_key=PINECONE_API_KEY)
+pc = Pinecone(api_key=PINECONE_API_KEY)
+
+# Ruta del archivo CSV
+file_path = '/content/dataset.csv'
+
+# Cargar el dataset y establecerlo como variable global
+try:
+    df = pd.read_csv(file_path)
+    DATASET = Dataset.from_pandas(df)
+    print(f"Dataset '{file_path}' loaded successfully.\n")
+    print(DATASET)
+except Exception as e:
+    DATASET = None
+    print(f"Error loading dataset: {e}")
+
+# Función para imprimir la información del dataset actual
+def print_dataset_info():
+    print(f"Current dataset: {DATASET}" if DATASET else "No dataset loaded.")
+
+# Guardar la variable global para uso en otras celdas
+get_ipython().run_cell_magic('capture', '', '%store DATASET')