Spaces:
Sleeping
Sleeping
durgeshshisode1988
commited on
Commit
•
359fb9e
1
Parent(s):
b9e07ca
Delete app.py
Browse files
app.py
DELETED
@@ -1,63 +0,0 @@
|
|
1 |
-
import streamlit as st
|
2 |
-
import os
|
3 |
-
from langchain_groq import ChatGroq
|
4 |
-
from langchain_community.document_loaders import WebBaseLoader
|
5 |
-
from langchain_community.embeddings import OllamaEmbeddings
|
6 |
-
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
7 |
-
from langchain.chains.combine_documents import create_stuff_documents_chain
|
8 |
-
from langchain_core.prompts import ChatPromptTemplate
|
9 |
-
from langchain.chains import create_retrieval_chain
|
10 |
-
from langchain_community.vectorstores import FAISS
|
11 |
-
import time
|
12 |
-
|
13 |
-
from dotenv import load_dotenv
|
14 |
-
|
15 |
-
load_dotenv()
|
16 |
-
|
17 |
-
## Load Groq API Key
|
18 |
-
groq_api_key = os.environ['GROQ_API_KEY']
|
19 |
-
|
20 |
-
if "vector" not in st.session_state:
|
21 |
-
st.session_state.embeddings=OllamaEmbeddings()
|
22 |
-
st.session_state.loader=WebBaseLoader("https://docs.smith.langchain.com/")
|
23 |
-
st.session_state.docs=st.session_state.loader.load()
|
24 |
-
|
25 |
-
st.session_state.text_splitter=RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
|
26 |
-
st.session_state.final_documents=st.session_state.text_splitter.split_documents(st.session_state.docs[:50])
|
27 |
-
st.session_state.vectors=FAISS.from_documents(st.session_state.final_documents, st.session_state.embeddings)
|
28 |
-
|
29 |
-
|
30 |
-
st.title("Chatgroq Demo")
|
31 |
-
llm=ChatGroq(groq_api_key=groq_api_key,
|
32 |
-
model="gemma-7b-it")
|
33 |
-
|
34 |
-
prompt = ChatPromptTemplate.from_template(
|
35 |
-
"""
|
36 |
-
Answer the question based on the provided context only.
|
37 |
-
Please provide the most accurate response based on the question.
|
38 |
-
<context>
|
39 |
-
{context}
|
40 |
-
<context>
|
41 |
-
Question: {input}
|
42 |
-
"""
|
43 |
-
)
|
44 |
-
|
45 |
-
|
46 |
-
document_chain = create_stuff_documents_chain(llm, prompt)
|
47 |
-
retriver = st.session_state.vectors.as_retriever()
|
48 |
-
retriver_chain = create_retrieval_chain(retriver, document_chain)
|
49 |
-
|
50 |
-
prompt=st.text_input("Input your prompt here")
|
51 |
-
|
52 |
-
if prompt:
|
53 |
-
start=time.process_time()
|
54 |
-
response = retriver_chain.invoke({"input": prompt})
|
55 |
-
print("Response time :",time.process_time() - start)
|
56 |
-
st.write(response['answer'])
|
57 |
-
|
58 |
-
# With a Streamlit expander
|
59 |
-
with st.expander("Document Similarity Search"):
|
60 |
-
# Find the relevant chunks
|
61 |
-
for i, doc in enumerate(response["context"]):
|
62 |
-
st.write(doc.page_content)
|
63 |
-
st.write("------------------------------------")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|