Update fastapi post method to match current pattern

app/api/routers/file_upload.py

@@ -1,34 +1,40 @@
 from typing import Annotated
-from fastapi import FastAPI, File, UploadFile
+from fastapi import APIRouter, UploadFile
 from app.categorization.file_processing import process_file, save_results
-from app.categorization.config import CATEGORY_REFERENCE_OUTPUT_FILE 
+from app.schema.index import FileUploadCreate
 import asyncio
 import os
+import csv
 
-app = FastAPI()
+file_upload_router = r = APIRouter(prefix="/api/v1/file_upload", tags=["file_upload"])
 
-@app.post("/v1/file_upload")
-async def create_file(file: Annotated[bytes, File()]):
+@r.post(
+    "/",
+    responses={
+        200: {"description": "File successfully uploaded"},
+        400: {"description": "Bad request"},
+        500: {"description": "Internal server error"},
+    },
+)
+async def create_file(input_file: UploadFile):
     try:
-        with open(file.filename, 'wb') as f:
-            # Create directory to store all uploaded .csv files
-            file_upload_directory_path = "data/tx_data/input"
-            if not os.path.exists(file_upload_directory_path):
-                os.makedirs(file_upload_directory_path)
+         # Create directory to store all uploaded .csv files
+        file_upload_directory_path = "data/tx_data/input"
+        if not os.path.exists(file_upload_directory_path):
+            os.makedirs(file_upload_directory_path)
 
-            # Write items of .csv filte to directory
-            with open(os.path.join(file_upload_directory_path, file.filename)):
-                f.write(f.readlines)
+        # Write items of .csv filte to directory
+        with open(os.path.join(file_upload_directory_path, input_file.filename)) as output_file:
+            [output_file.write(" ".join(row)+'\n') for row in csv.reader(input_file)]
+        output_file.close()
 
-            # with the newly created file and it's path, process and save it for embedding
-            processed_file = process_file(os.path.realpath(file.filename))
-            result = await asyncio.gather(processed_file)
-            save_results(result)
+        # With the newly created file and it's path, process and save it for embedding
+        processed_file = process_file(os.path.realpath(input_file.filename))
+        result = await asyncio.gather(processed_file)
+        save_results(result)
 
     except Exception:
         return {"message": "There was an error uploading this file. Ensure you have a .csv file with the following columns:"
         "\n source, date, type, category, description, amount"}
-    finally:
-        file.file.close()
-    
-    return {"message": f"Successfully uploaded {file.filename}"}
+
+    return {"message": f"Successfully uploaded {input_file.filename}"}

main.py

@@ -8,6 +8,7 @@ from fastapi.responses import RedirectResponse
 from app.api.routers.chat import chat_router
 from app.api.routers.user import user_router
 from app.api.routers.transaction import transaction_router
+from app.api.routers.file_upload import file_upload_router
 from app.settings import init_settings
 from fastapi.staticfiles import StaticFiles
 from alembic.config import Config
@@ -53,6 +54,7 @@ def init_app(init_db: bool = True) -> FastAPI:
     app.include_router(chat_router, prefix="/api/chat")
     app.include_router(user_router)
     app.include_router(transaction_router)
+    app.include_router(file_upload_router)
 
     return app