feat: add dele folder

constants.py

@@ -41,6 +41,7 @@ N_BATCH = 1024
 # N_BATCH = 512
 
 
+
 # https://python.langchain.com/en/latest/_modules/langchain/document_loaders/excel.html#UnstructuredExcelLoader
 DOCUMENT_MAP = {
     ".txt": TextLoader,

main.py

@@ -3,15 +3,12 @@ from fastapi.staticfiles import StaticFiles
 
 
 from pydantic import BaseModel
-import pickle
-import uvicorn
-
-import logging
 import os
 import shutil
 import subprocess
+import shutil
 
-import torch
+# import torch
 from langchain.chains import RetrievalQA
 from langchain.embeddings import HuggingFaceInstructEmbeddings
 from langchain.prompts import PromptTemplate
@@ -188,7 +185,19 @@ async def create_upload_file(file: UploadFile):
 
     # check the content type (MIME type)
     content_type = file.content_type
-    if content_type not in ["image/jpeg", "image/png", "image/gif"]:
+
+    if content_type not in [
+        "text/plain",
+        "text/markdown",
+        "text/x-markdown",
+        "text/csv",
+        "application/msword",
+        "application/pdf",
+        "application/vnd.ms-excel",
+        "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
+        "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
+        "text/x-python",
+        "application/x-python-code"]:
         raise HTTPException(status_code=400, detail="Invalid file type")
 
     # do something with the valid file