Update main.py

main.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, HTTPException, File, UploadFile
+from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from contextlib import asynccontextmanager
 from langchain_community.document_loaders import PyPDFLoader
@@ -25,7 +25,6 @@ import docx2txt
 import pptx
 import os
 import utils
-import shutil
 
 ## APPLICATION LIFESPAN
 # Load the environment variables using FastAPI lifespan event so that they are available throughout the application
@@ -96,36 +95,6 @@ async def set_api_key(api_key: APIKey):
     os.environ["OPENAI_API_KEY"] = api_key.api_key
     return "API key set successfully!"
 
-## POST - /upload_file_on_server
-@app.post("/upload_file_on_server")
-async def upload_file_on_server(file: UploadFile = File(...)):
-    # file_content = file.read()  # Load the document
-    
-    # # Create a directory if it doesn't exist
-    data_dir = "data"
-    # # os.makedirs(data_dir, exist_ok=True)
-    
-    # # Create a temporary file in the data directory
-    # with tempfile.NamedTemporaryFile(delete=False, dir=data_dir) as temp_file:
-    #     temp_file.write(file_content)  # Write the uploaded file content to the temporary file
-    #     temp_file_path = temp_file.name  # Get the path of the temporary file
-    #     return temp_file_path
-    # Save the uploaded file to the temporary directory
-    #file_path = os.path.join(data_dir, file.filename)
-    #print(file_path)
-    
-    # Save the uploaded file to a temporary location
-    # with open(file_path, "wb") as buffer:
-    #     shutil.copyfileobj(file.file, buffer)
-
-    # with open(file_path, "wb") as f:
-    #     f.write(file.file.read())
-    
-    # Return a response
-    # return {"filename": file.filename, "message": "File uploaded successfully"}
-    # return file.filename
-    return "hiiiiiiiiiiiiiiiiiiiiiiiiiii"
-
 ## POST - /load_file
 # Load the file, split it into document chunks, and upload the document embeddings into a vectorstore   
 @app.post("/load_file/{llm}")