Update app.py

app.py

@@ -1,26 +1,24 @@
 import os
 os.system('pip install curl_cffi tqdm bitsandbytes tiktoken g4f pinecone-client pandas datasets sentence-transformers')
 
-# Retrieve the Pinecone API key from the user
-PINECONE_API_KEY = "3a3e9022-381d-436e-84cb-ba93464d283e"  # Use the key you set in the secrets
-PINECONE_ENVIRONMENT = "us-east-1"  # Use the environment you set in the secrets
-
 # Setup and load your keys
 import os
 from g4f import ChatCompletion
+#from google.colab import userdata
 from pinecone import Pinecone
 import pandas as pd
 from datasets import Dataset
 from sentence_transformers import SentenceTransformer
 import gradio as gr
 
+
 #TEMA GRADIO
 import gradio as gr
 
 theme = gr.themes.Default(
-    primary_hue="indigo",
-    secondary_hue="indigo",
-    neutral_hue=gr.themes.Color(c100="#384b8f", c200="#d7dbe5", c300="#1d1d20", c400="#1d1d20", c50="#657ed2", c500="#fcfcfc", c600="#3d3e52", c700="#383947", c800="#242429", c900="#29292e", c950="#292932"),
+    primary_hue=gr.themes.Color(c100="#83a1c9", c200="#000000", c300="#a5b4fc", c400="#818cf8", c50="#ffffff", c500="#6366f1", c600="#4f46e5", c700="#4338ca", c800="#3730a3", c900="#312e81", c950="#2b2c5e"),
+    secondary_hue=gr.themes.Color(c100="#ffffff", c200="#9ab9d5", c300="#000000", c400="#000000", c50="#ffffff", c500="#000000", c600="#000000", c700="#000000", c800="#3730a3", c900="#312e81", c950="#2b2c5e"),
+    neutral_hue=gr.themes.Color(c100="#e0e7ff", c200="#c7d2fe", c300="#658fd2", c400="#ffffff", c50="#eef2ff", c500="#ffffff", c600="#191a2f", c700="#29357f", c800="#4463a2", c900="#545acf", c950="#1a1a1f"),
 ).set(
     background_fill_primary='*background_fill_secondary',
     border_color_accent_subdued='*border_color_accent',
@@ -71,6 +69,10 @@ theme = gr.themes.Default(
 
 model_name = "BAAI/bge-m3"
 
+# Retrieve the Pinecone API key from the user
+PINECONE_API_KEY = "3a3e9022-381d-436e-84cb-ba93464d283e"  # Use the key you set in the secrets
+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)
 
@@ -90,6 +92,9 @@ def print_current_selection():
 # Establecer el nombre del índice automáticamente
 INDEX_NAME = 'neonatos'
 
+# Obtener la clave API de Pinecone
+#PINECONE_API_KEY = userdata.get('PINECONE_API_KEY')
+
 def connect_to_pinecone(index_name):
     global INDEX_NAME
     try:
@@ -243,6 +248,7 @@ def chat_function(message, history):
         if filtered_links:
             full_response += filtered_links[0] #f".\n\nRespuesta:\n" + filtered_links[0]
 
+
         # Now handle the images based on the detected tags
         tags_to_images = {
             "rId101": "images/rId101.png",
@@ -291,6 +297,8 @@ def chat_function(message, history):
             "rId97": "images/rId97.png",
         }
 
+
+
         for tag in tags_detected:
             for key, path in tags_to_images.items():
                 if key in tag and check_image_exists(path):
@@ -306,6 +314,7 @@ def update_image(image_url):
     else:
         return None
 
+
 images = [
     {"id": "rId101", "path": "images/rId101.png", "description": "📊 El código UBIGEO muestra que Lima (150101) tiene el mayor número de registros, seguido por Trujillo (130101) y San Juan de Miraflores (150133), destacando la concentración de atención médica en estas zonas. Un gráfico de barras visualiza los 10 códigos UBIGEO más frecuentes, facilitando la identificación de las áreas con mayor actividad en salud."},
     {"id": "rId105", "path": "images/rId105.png", "description": "🏥 Los establecimientos de nivel 3 son los más representados, con 652,948 registros, seguidos por los de nivel 4 (310,272) y nivel 2 (277,385)."},
@@ -353,6 +362,8 @@ images = [
     {"id": "rId97", "path": "images/rId97.png", "description": ""},
 ]
 
+
+
 # Gradio layout setup
 with gr.Blocks(theme=theme) as demo:
     with gr.Row():