Update pages/AI_Chatbot.py

pages/AI_Chatbot.py

@@ -13,9 +13,9 @@ openai.api_key = os.getenv("OPENAI_API_KEY")
 from langchain.document_loaders import PyPDFLoader
 from langchain.chat_models import ChatOpenAI
 
+
 st.title("Chat with data")
 model = ChatOpenAI(model = 'gpt-4', max_tokens = 100,temperature=0)
-
 uploaded_file = st.file_uploader("Choose a file")
 if uploaded_file is not None:
         # Make temp file path from uploaded file
@@ -31,15 +31,15 @@ def extract(uploaded_file):
         res.append(i.page_content.replace('\n',''))
     a = " ".join(res)
     return a
-def lang(uploaded_file,ques):
-    context = extract(uploaded_file)    
+def lang(ques):
+    context = extract(tmp_file.name)    
     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(uploaded_file,ques)
+def qna(ques):
+    session_state['answer']= lang(ques)
     
 if 'answer' not in session_state:
     session_state['answer']= ""
@@ -49,4 +49,4 @@ ques= st.text_area(label= "Please enter the Question that you wanna ask.",
 
 st.text_area("result", value=session_state['answer'])
 
-st.button("Get answer dictionary", on_click=qna, args=[uploaded_file,ques])
+st.button("Get answer dictionary", on_click=qna, args=[ques])