adding files

main.py

@@ -4,6 +4,7 @@ from google.cloud import storage
 import io
 import os
 import tempfile
+from pydantic import BaseModel
 
 app = FastAPI()
 
@@ -64,12 +65,24 @@ async def upload_file(file: UploadFile = File(...)):
     append_to_gcs_csv(df, gcs_file_path)
     return {"message": "File uploaded successfully"}
 
+# @app.post("/upload-data/")
+# async def upload_data(category: str = Form(...), score: float = Form(...)):
+#     df = pd.DataFrame([[category, score]], columns=['category', 'score'])
+#     append_to_gcs_csv(df, gcs_file_path)
+#     return {"message": "Data uploaded successfully"}
+
+class DataPayload(BaseModel):
+    category: str
+    score: float
+
 @app.post("/upload-data/")
-async def upload_data(category: str = Form(...), score: float = Form(...)):
-    df = pd.DataFrame([[category, score]], columns=['category', 'score'])
+async def upload_data(payload: DataPayload):
+    df = pd.DataFrame([[payload.category, payload.score]], columns=['category', 'score'])
     append_to_gcs_csv(df, gcs_file_path)
     return {"message": "Data uploaded successfully"}
 
+
+
 @app.post("/clear-data/")
 async def clear_data():
     # Create an empty DataFrame with the same columns

requirements.txt

@@ -2,5 +2,4 @@ fastapi
 uvicorn
 pandas
 gradio
-sqlalchemy
 google-cloud-storage