Update process_text_file() function

app.py

@@ -59,9 +59,20 @@ def process_text_file(file: AskFileResponse):
     with open(temp_file_path, "wb") as f:
         f.write(file.content)
 
-    text_loader = TextFileLoader(temp_file_path)
-    documents = text_loader.load_documents()
-    texts = text_splitter.split_texts(documents)
+    if file.type == 'text/plain':
+        text_loader = TextFileLoader(temp_file_path)
+        documents = text_loader.load_documents()
+        texts = text_splitter.split_texts(documents)
+    elif file.type == 'application/pdf':
+        import pymupdf
+        doc = pymupdf.open(temp_file_path)
+        texts = ""
+        for page_num in range(len(doc)):
+            page = doc.load_page(page_num)
+            texts += page.get_text()
+    else:
+        raise ValueError("Provide a .txt or .pdf file")
+    
     return texts