durgeshshisode1988 commited on
Commit
359fb9e
1 Parent(s): b9e07ca

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -63
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("------------------------------------")