p

app.py

@@ -0,0 +1,17 @@
+import streamlit as st
+from model import rag_chain
+
+st.set_page_config(page_title="Chatbot Legal 🇪🇨", page_icon="⚖️")
+
+st.title("🤖 Chatbot sobre el Código Penal de Ecuador ⚖️")
+st.write("Este agente conversacional responde preguntas sobre las leyes del Código Penal ecuatoriano.")
+
+user_input = st.text_input("✍️ Escribe tu pregunta:")
+
+if st.button("🔎 Consultar"):
+    if user_input:
+        response = rag_chain.invoke({"query": user_input})
+        st.write("### 📜 Respuesta:")
+        st.write(response)
+    else:
+        st.warning("⚠️ Escribe una pregunta antes de consultar.")

model.py

@@ -0,0 +1,35 @@
+from langchain.chains import RetrievalQA
+from langchain.vectorstores import Chroma
+from langchain.embeddings import HuggingFaceEmbeddings
+from langchain.llms import ChatMistralAI
+import chromadb
+from huggingface_hub import hf_hub_download
+import os
+# Descargar los archivos de embeddings desde Hugging Face
+embedding_path = hf_hub_download(repo_id="VictorCarr02/Conversational-Agent-LawsEC", filename="mis_embeddings")
+chroma_path = hf_hub_download(repo_id="VictorCarr02/Conversational-Agent-LawsEC", filename="chroma/chroma.sqlite3")
+
+# Cargar ChromaDB y los embeddings
+chromadb_client = chromadb.PersistentClient(path=chroma_path)
+collection = chromadb_client.get_or_create_collection(name="mis_embeddings")
+embeddings = HuggingFaceEmbeddings(model_name="mistralai/MistralAIEmbeddings", path=embedding_path)
+vector_store = Chroma(collection=collection, embedding_function=embeddings)
+
+
+
+# Acceder a la clave API desde la variable de entorno
+api_key = os.getenv("MISTRAL_API_KEY")
+
+# Verifica si la clave fue obtenida correctamente
+if api_key is None:
+    raise ValueError("La clave API MISTRAL_API_KEY no está configurada como variable de entorno.")
+
+# Crear el modelo LLM con la clave API
+llm = ChatMistralAI(api_key=api_key)
+
+# Crear el agente RAG
+rag_chain = RetrievalQA.from_chain_type(
+    llm=llm,
+    retriever=vector_store.as_retriever(),
+    chain_type="stuff"
+)

requirements.txt

@@ -0,0 +1,5 @@
+streamlit
+langchain
+chromadb
+huggingface_hub
+MistralAIEmbeddings