Upload 3 files

.gitattributes

@@ -32,3 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+vectorIndex.json filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,81 @@
+import streamlit as st
+from streamlit_chat import message
+import requests
+from gpt_index import SimpleDirectoryReader,GPTListIndex,GPTSimpleVectorIndex,LLMPredictor,PromptHelper,ServiceContext
+from langchain import OpenAI
+import sys
+import os
+
+st.set_page_config(
+    page_title="Chatbot",
+    page_icon=":robot_face:"
+)
+
+def api_key_callback():
+    # api_key = st.session_state.api_key_input
+    os.environ["OPENAI_API_KEY"] = st.session_state.api_key_input
+    # print(st.session_state.api_key_input)
+
+def api_form():
+    with st.form(key="search_form"):
+        st.subheader("API Key")
+        st.text_input("Enter your OpenAI API key", key="api_key_input")
+        st.form_submit_button(label="Submit", on_click=api_key_callback)
+        st.caption('''
+            This api key wil not be stored and deleted at the end of your web session.
+            For more details, you can check out the 
+            [Source Code](https://huggingface.co/spaces/darthPanda/romeo_and_juliet_chatbot_with_gptIndex/tree/main). 
+            If you want to be extra careful, you can delete your 
+            [OpenAI API key](https://platform.openai.com/account/api-keys) 
+            after using this space.
+            ''')
+
+with st.sidebar:
+    st.title("Romeo and Juliet Chatbot")
+    st.markdown('''
+                <div>
+                This gpt3 based chatbot has been 
+                trained on famous play <b>Romeo and Juliet</b> by William Shakespeare
+                </div>
+                ''', unsafe_allow_html=True)
+    api_form()
+    # st.markdown("[Source Code](https://github.com/ai-yash/st-chat)")
+    # st.markdown("[Book Link](https://www.gutenberg.org/ebooks/1513)")
+    st.markdown(
+    "<div style='position: fixed; bottom: 0;'>Created by Taaha Bajwa</div>",
+    unsafe_allow_html=True,
+    )
+
+st.markdown('Ask me any question from Romeo and Juliet')
+
+if 'generated' not in st.session_state:
+    st.session_state['generated'] = []
+
+if 'past' not in st.session_state:
+    st.session_state['past'] = []
+
+def query(payload):
+    vIndex = GPTSimpleVectorIndex.load_from_disk('vectorIndex.json')
+    response = vIndex.query(payload,response_mode='compact')
+    # print(str(response))
+    return (str(response)).lstrip()
+
+def get_text():
+    input_text = st.text_input("You: ","", key="input")
+    return input_text 
+
+
+user_input = get_text()
+
+if user_input:
+    try:
+        output = query(user_input)
+        st.session_state.past.append(user_input)
+        st.session_state.generated.append(output)
+    except:
+        st.error('Invalid API key', icon="🚨")
+
+if st.session_state['generated']:
+    for i in range(len(st.session_state['generated'])-1, -1, -1):
+        message(st.session_state["generated"][i], key=str(i))
+        message(st.session_state['past'][i], is_user=True, key=str(i) + '_user')

requirements.txt

@@ -0,0 +1,4 @@
+streamlit==1.17.0
+gpt_index
+langchain
+streamlit-chat

vectorIndex.json

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:17795ddb191fef61de7b096c9cf08246fc0d9c1ffa2d4da310d754942b68e26e
+size 13648161