Update app.py

app.py

@@ -11,29 +11,32 @@ from langchain_pinecone import PineconeVectorStore
 st.set_page_config('Opositor')
 st.header("Pregunta al trebep")
 
-# CARGAMOS LLM
-os.environ["GROQ_API_KEY"] = "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"
-model = 'llama3-8b-8192'
-llm = ChatGroq(model = model)
 
-# CARGAMOS MODELO DE EMBEDDING
-model_name = 'intfloat/multilingual-e5-base'
-embedding = HuggingFaceEmbeddings(model_name=model_name)
-
-# CARGAMOS EL VECTORSTORE DE PINECONE
-os.environ["PINECONE_API_KEY"] ='4bf0b4cf-4ced-4f70-8977-d60bb8ae405a'
-index_name = "boe-intfloat-multilingual-e5-base"
-namespace = "trebep"
-vectorstore = PineconeVectorStore(index_name=index_name, namespace=namespace, embedding=embedding)
-
-# CREAMOS EL RETRIEVAL
-qa = RetrievalQA.from_chain_type(
-    llm=llm,
-    chain_type="stuff",
-    retriever=vectorstore.as_retriever(),
-    return_source_documents=True,
-    #verbose=True
-)
+@st.cache_resource 
+def incializar():
+    # CARGAMOS LLM
+    os.environ["GROQ_API_KEY"] = "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"
+    model = 'llama3-8b-8192'
+    llm = ChatGroq(model = model)
+    
+    # CARGAMOS MODELO DE EMBEDDING
+    model_name = 'intfloat/multilingual-e5-base'
+    embedding = HuggingFaceEmbeddings(model_name=model_name)
+    
+    # CARGAMOS EL VECTORSTORE DE PINECONE
+    os.environ["PINECONE_API_KEY"] ='4bf0b4cf-4ced-4f70-8977-d60bb8ae405a'
+    index_name = "boe-intfloat-multilingual-e5-base"
+    namespace = "trebep"
+    vectorstore = PineconeVectorStore(index_name=index_name, namespace=namespace, embedding=embedding)
+    
+    # CREAMOS EL RETRIEVAL
+    qa = RetrievalQA.from_chain_type(
+        llm=llm,
+        chain_type="stuff",
+        retriever=vectorstore.as_retriever(),
+        return_source_documents=True,
+        #verbose=True
+    )
 
 # Función para mostrar logs
 def mostrar_logs(logs,hints):
@@ -55,6 +58,9 @@ logs = []
 user_question = st.text_input("¡A jugar! Haz una pregunta al trebep:")
 if user_question:
 
+    # Inicializar entorno
+    incializar()
+
     # Obtenemos la respuesta
     respuesta = qa.invoke(user_question)
     docs = respuesta['source_documents']