Update app.py

app.py

@@ -4,8 +4,8 @@ import json
 import os
 import requests
 from PyPDF2 import PdfReader
-from io import BytesIO
 import gradio as gr
+import gdown
 
 load_dotenv(override=True)
 
@@ -76,20 +76,25 @@ class Me:
         self.openai = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
         self.name = "Jacob Isaacson"
 
-        # Load LinkedIn PDF from Google Drive
-        linkedin_url = "https://drive.google.com/uc?export=download&id=1xz2RowkImpI8odYv8zvKdlRHaKfILn40"
-        pdf_response = requests.get(linkedin_url)
-        reader = PdfReader(BytesIO(pdf_response.content))
+        # Download LinkedIn PDF using gdown
+        linkedin_id = "1xz2RowkImpI8odYv8zvKdlRHaKfILn40"
+        linkedin_path = "Linkedin.pdf"
+        gdown.download(f"https://drive.google.com/uc?id={linkedin_id}", linkedin_path, quiet=False)
+        reader = PdfReader(linkedin_path)
+
         self.linkedin = ""
         for page in reader.pages:
             text = page.extract_text()
             if text:
                 self.linkedin += text
 
-        # Load summary.txt from Google Drive
-        summary_url = "https://drive.google.com/uc?export=download&id=1hjJz082YFSVjFtpO0pwT6Tyy3eLYYj6-"
-        summary_response = requests.get(summary_url)
-        self.summary = summary_response.text
+        # Download summary.txt using gdown
+        summary_id = "1hjJz082YFSVjFtpO0pwT6Tyy3eLYYj6-"
+        summary_path = "summary.txt"
+        gdown.download(f"https://drive.google.com/uc?id={summary_id}", summary_path, quiet=False)
+
+        with open(summary_path, "r", encoding="utf-8") as f:
+            self.summary = f.read()
 
     def handle_tool_call(self, tool_calls):
         results = []
@@ -130,7 +135,6 @@ If the user is engaging in discussion, try to steer them towards getting in touc
                 done = True
         return response.choices[0].message.content
 
-
 if __name__ == "__main__":
     me = Me()
     gr.ChatInterface(me.chat, type="messages").launch()