Update filetypes

models/index_model.py

@@ -29,7 +29,14 @@ class Index_handler:
     
         try:
             temp_path = tempfile.TemporaryDirectory()
-            temp_file = tempfile.NamedTemporaryFile(suffix=".txt", dir=temp_path.name, delete=False)
+            if file.content_type.startswith("text/plain"):
+                suffix = ".txt"
+            elif file.content_type.startswith("application/pdf"):
+                suffix = ".pdf"
+            else:
+                await ctx.respond("Only accepts txt or pdf files")
+                return
+            temp_file = tempfile.NamedTemporaryFile(suffix=suffix, dir=temp_path.name, delete=False)
             await file.save(temp_file.name)
             index = await self.loop.run_in_executor(None, partial(self.index_file, temp_path.name))
             self.index_storage[ctx.user.id] = index

pyproject.toml

@@ -32,6 +32,7 @@ dependencies = [
   "flask",
   "beautifulsoup4",
   "gpt-index",
+  "PyPDF2",
 ]
 dynamic = ["version"]
 [project.scripts]

requirements.txt

@@ -10,4 +10,5 @@ sqlitedict==2.1.0
 backoff==2.2.1
 flask==2.2.2
 beautifulsoup4==4.11.1
-gpt-index==0.2.16
+gpt-index==0.2.16
+PyPDF2==3.0.1