Create app.py

app.py

@@ -0,0 +1,74 @@
+import streamlit as st
+import os
+
+from groq import Groq
+from PyPDF2 import PdfReader
+from datetime import datetime
+from langchain.text_splitter import RecursiveCharacterTextSplitter
+from langchain_community.embeddings import HuggingFaceEmbeddings 
+#from langchain.vectorstores import FAISS
+from langchain_community.vectorstores import FAISS
+from langchain_groq import ChatGroq
+#from langchain.chat_models import ChatOpenAI
+from langchain.chains.question_answering import load_qa_chain
+from langchain.chains import RetrievalQA
+
+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-small'
+embedding = HuggingFaceEmbeddings(model_name=model_name)
+
+# CARGAMOS EL VECTORSTORE DE PINECONE
+index_name = "boe-intfloat-multilingual-e5-base"
+namespace = "trebep"
+vectorstore = PineconeVectorStore(index_name=index_name, namespace=names, 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):
+    # Crear un contenedor desplegable
+    with st.expander("Chunks"):
+        for hint in hints:
+            st.write(hint.page_content)
+            st.write("-" * 30)
+
+    st.sidebar.header("Registro de preguntas")
+    for entry in logs:
+        st.sidebar.write(f"**Pregunta: {entry['Pregunta']}**")
+        st.sidebar.write(f"Respuesta: {entry['Respuesta']}")
+
+
+# Lista para almacenar preguntas y respuestas
+logs = []
+
+if pdf_obj:
+    user_question = st.text_input("¡A jugar! Haz una pregunta sobre tu PDF:")
+    if user_question:
+
+        # Obtenemos la respuesta
+        respuesta = qa.invoke(user_question)
+
+        # Mostrar la variable en color verde
+        st.subheader("Respuesta")
+        st.write(f":green[{str(respuesta)}]")
+
+        # Guardar pregunta y respuesta en los logs
+        logs.append({"Pregunta": user_question, "Respuesta": respuesta})
+
+        # Mostrar logs actualizados
+        mostrar_logs(logs,docs)