Create app.py (#4)

- Create app.py (a4242ca47d9cde18a158d710f94577dac7875e6d)

app.py

@@ -1,6 +1,7 @@
 from dotenv import load_dotenv
 load_dotenv()
-
+import os
+from tempfile import NamedTemporaryFile
 import streamlit as st
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.document_loaders import PyPDFLoader,DirectoryLoader
@@ -70,7 +71,12 @@ def displayPDF(file):
 #streamlit code
 st.set_page_config(layout='wide')
 
-import os
+
+
+def main():
+    st.title('Content Summarizer')
+
+    uploaded_file = st.file_uploader("Upload your PDF file", type=['pdf'])
 
 def main():
     st.title('Content Summarizer')
@@ -81,22 +87,18 @@ def main():
         if st.button("Summarize"):
             col1, col2 = st.columns(2)
 
-            # Ensure the directory exists
-            data_dir = "data"
-            if not os.path.exists(data_dir):
-                os.makedirs(data_dir)
-
-            filepath = os.path.join(data_dir, uploaded_file.name)
-            with open(filepath, 'wb') as temp_file:
+            # Save the uploaded file to a temporary location
+            with NamedTemporaryFile(delete=False, suffix=".pdf") as temp_file:
                 temp_file.write(uploaded_file.read())
+                temp_filepath = temp_file.name
 
             with col1:
                 st.info("Uploaded PDF File")
-                pdf_viewer = displayPDF(filepath)
+                pdf_viewer = displayPDF(temp_filepath)
 
             with col2:
                 st.info("Summarization is below")
-                summary = llm_pipleline(filepath)
+                summary = llm_pipleline(temp_filepath)
                 st.success(summary)
     else:
         st.warning("Please upload a valid PDF file.")