Update app.py

app.py

@@ -1,3 +1,4 @@
+import tempfile
 import PyPDF2
 import streamlit as st
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
@@ -6,15 +7,6 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer
 tokenizer = GPT2Tokenizer.from_pretrained("openai-community/gpt2")
 model = GPT2LMHeadModel.from_pretrained("openai-community/gpt2")
 
-pdf_file = st.file_uploader("Upload an pdf file", type=["pdf"], accept_multiple_files=False)
-
-if pdf_file is not None:
-    with tempfile.NamedTemporaryFile(delete=False) as tmp_file:
-        tmp_file.write(pdf_file.read())
-        st.success("PDF file successfully uploaded and stored temporally.")
-    file_path = tmp_file.name
-else:
-    st.markdown('File not found!')
         
 def extract_text_from_pdf(file_path):
     text = ""
@@ -33,12 +25,17 @@ def generate_response(user_input):
 def main():
     st.title("PDF Chatbot")
 
-    # File upload
-    uploaded_file = st.file_uploader("Upload PDF", type=["pdf"])
+    pdf_file = st.file_uploader("Upload an pdf file", type=["pdf"], accept_multiple_files=False)
 
-    if uploaded_file is not None:
-        pdf_text = extract_text_from_pdf(uploaded_file)
+    if pdf_file is not None:
+        with tempfile.NamedTemporaryFile(delete=False) as tmp_file:
+            tmp_file.write(pdf_file.read())
+            st.success("PDF file successfully uploaded and stored temporally.")
+        file_path = tmp_file.name
+        pdf_text = extract_text_from_pdf(file_path)
         st.text_area("PDF Content", pdf_text)
+    else:
+        st.markdown('File not found!')        
 
     user_input = st.text_input("You:", "")
     if st.button("Send"):