Update pages/AI_Chatbot.py

pages/AI_Chatbot.py

@@ -2,32 +2,34 @@ import openai
 import os
 import streamlit as st
 openai.api_key = os.getenv("OPENAI_API_KEY")
+from langchain.document_loaders import PyPDFLoader
 
 st.title("AI Chatbot")
 
-if "openai_model" not in st.session_state:
-    st.session_state["openai_model"] = "gpt-3.5-turbo"
+uploaded_file = st.file_uploader("Choose a file")
+def extract(uploaded_file):
+    res = []
+    loader = PyPDFLoader(link)
+    pages = loader.load()
+    for i in pages:
+        res.append(i.page_content.replace('\n',''))
+    a = " ".join(res)
+    return a
+context = extract(uploaded_file)    
+def lang(ques):
+    docs =  Document(page_content=context)
+    index2 = VectorstoreIndexCreator().from_documents([docs])
+    answer = index2.query(llm = model, question = ques)
+    index2.vectorstore.delete_collection()
+    return answer
+def qna(uploaded_file,ques):
+    session_state['answer']= lang(jury_url)
+    
+st.title("Jury Records")
 
-if "messages" not in st.session_state:
-    st.session_state.messages = []
+ques= st.text_area(label= "Please enter the Question that you wanna ask.", 
+              placeholder="Question")
 
-for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        st.markdown(message["content"])
+st.text_area("result", value=session_state['answer'])
 
-if prompt := st.chat_input("What is up?"):
-    st.session_state.messages.append({"role": "user", "content": prompt})
-    with st.chat_message("user"):
-        st.markdown(prompt)
-
-    with st.chat_message("assistant"):
-        stream = openai.ChatCompletion.create(
-            model=st.session_state["openai_model"],
-            messages=[
-                {"role": m["role"], "content": m["content"]}
-                for m in st.session_state.messages
-            ],
-            stream=True,
-        )
-        response = st.write_stream(stream)
-    st.session_state.messages.append({"role": "assistant", "content": response})
+st.button("Get answer dictionary", on_click=qna, args=[ques])