Update pages/AI_Chatbot.py

pages/AI_Chatbot.py

@@ -4,7 +4,7 @@ import streamlit as st
 openai.api_key = os.getenv("OPENAI_API_KEY")
 from langchain.document_loaders import PyPDFLoader
 
-st.title("AI Chatbot")
+st.title("Chat with data")
 
 uploaded_file = st.file_uploader("Choose a file")
 def extract(uploaded_file):
@@ -15,8 +15,8 @@ def extract(uploaded_file):
         res.append(i.page_content.replace('\n',''))
     a = " ".join(res)
     return a
-context = extract(uploaded_file)    
-def lang(ques):
+def lang(uploaded_file,ques):
+    context = extract(uploaded_file)    
     docs =  Document(page_content=context)
     index2 = VectorstoreIndexCreator().from_documents([docs])
     answer = index2.query(llm = model, question = ques)
@@ -25,11 +25,9 @@ def lang(ques):
 def qna(uploaded_file,ques):
     session_state['answer']= lang(jury_url)
     
-st.title("Jury Records")
-
 ques= st.text_area(label= "Please enter the Question that you wanna ask.", 
               placeholder="Question")
 
 st.text_area("result", value=session_state['answer'])
 
-st.button("Get answer dictionary", on_click=qna, args=[ques])
+st.button("Get answer dictionary", on_click=qna, args=[uploaded_file,ques])