Update app.py

app.py

@@ -3,14 +3,25 @@ import fitz  # PyMuPDF
 import openai
 from fpdf import FPDF
 import os
+import tempfile
 
 # Function to extract text from a PDF file
-def extract_text_from_pdf(pdf_path):
-    doc = fitz.open(pdf_path)
+def extract_text_from_pdf(pdf_file):
+    # Save the uploaded file to a temporary location
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
+    temp_file.write(pdf_file.read())
+    temp_file.close()  # Close the file to ensure it's saved
+
+    # Open the saved PDF file
+    doc = fitz.open(temp_file.name)
     text = ""
     for page_num in range(len(doc)):
         page = doc.load_page(page_num)
         text += page.get_text()
+
+    # Delete the temporary file after reading (clean up)
+    os.remove(temp_file.name)
+
     return text
 
 # Function to ensure the summary ends with a full stop
@@ -25,10 +36,8 @@ def summarize_text(api_key, text):
     openai.api_key = api_key
     response = openai.ChatCompletion.create(
         model="gpt-3.5-turbo",  # Use "gpt-4" if you have access
-        messages=[
-            {"role": "system", "content": "You are a helpful assistant."},
-            {"role": "user", "content": f"Summarize the following text:\n\n{text}"}
-        ],
+        messages=[{"role": "system", "content": "You are a helpful assistant."},
+                  {"role": "user", "content": f"Summarize the following text:\n\n{text}"}],
         max_tokens=500,
         temperature=0.5
     )
@@ -40,10 +49,8 @@ def predict_topic(api_key, text):
     openai.api_key = api_key
     response = openai.ChatCompletion.create(
         model="gpt-3.5-turbo",  # Use "gpt-4" if you have access
-        messages=[
-            {"role": "system", "content": "You are a helpful assistant."},
-            {"role": "user", "content": f"What is the main topic of the following text?\n\n{text}"}
-        ],
+        messages=[{"role": "system", "content": "You are a helpful assistant."},
+                  {"role": "user", "content": f"What is the main topic of the following text?\n\n{text}"}],
         max_tokens=500,
         temperature=0.5
     )