abaliyan commited on
Commit
e3bbd74
1 Parent(s): 6b33c23

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +48 -0
app.py ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import os
3
+ from dotenv import load_dotenv
4
+ from utils import *
5
+ import uuid
6
+
7
+ load_dotenv()
8
+
9
+ st.title("Custom Chatbot")
10
+
11
+ if 'OPENAI_API_KEY' not in st.session_state:
12
+ st.session_state["OPENAI_API_KEY"]=''
13
+
14
+ os.environ['HUGGINGFACEHUB_API_TOKEN']=os.environ.get("HUGGINGFACE_TOKEN")
15
+
16
+ st.sidebar.title("😎🗝️")
17
+ st.session_state["OPENAI_API_KEY"]=st.sidebar.text_input("Enter your API key?", type="password")
18
+ # st.session_state["OPENAI_API_KEY"] = "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"
19
+ os.environ["OPENAI_API_KEY"] = st.session_state["OPENAI_API_KEY"]
20
+ file = st.sidebar.file_uploader("Upload your file", type="pdf")
21
+
22
+
23
+ if "unique_id" not in st.session_state:
24
+ st.session_state["unique_id"]=''
25
+
26
+ st.session_state["unique_id"] = uuid.uuid4().hex
27
+
28
+
29
+ st.subheader("Search Document..")
30
+
31
+ query = st.text_area("Enter query here..", key='question')
32
+
33
+ submit = st.button("Get Answer")
34
+
35
+ if submit:
36
+ if file and query:
37
+ st.write("lookign for:",query)
38
+ directory = f"pdf_files/{st.session_state["unique_id"]}"
39
+ content = save_pdf_to_directory(file,directory)
40
+ st.write(content)
41
+
42
+ elif file:
43
+ st.error("Enter the query first")
44
+ elif query:
45
+ st.error("Enter the document first")
46
+ else:
47
+ st.error("You gotta be kidding me.. I really wish I could read your mind")
48
+