adding files

main.py

@@ -40,15 +40,12 @@ def append_to_gcs_csv(new_data, gcs_file_path):
 
     # Check if the file exists in GCS bucket
     blob = gcs_bucket.blob(gcs_file_path)
-    exists = blob.exists()
-
-    # If file exists, read it into a DataFrame
-    if exists:
+    if exists := blob.exists():
         existing_data = pd.read_csv(io.BytesIO(blob.download_as_bytes()))
         # Ensure existing data has the right columns
         existing_data = existing_data[['category', 'score']]
         # Append new data to existing data
-        combined_data = pd.concat([existing_data, new_data], ignore_index=True)
+        combined_data = pd.concat([existing_data, new_data], ignore_index=True).dropna(how='all')
     else:
         combined_data = new_data
 
@@ -68,10 +65,13 @@ async def upload_file(file: UploadFile = File(...)):
 
 @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"}
-
+    try:
+        df = pd.DataFrame([[category, score]], columns=['category', 'score'])
+        append_to_gcs_csv(df, gcs_file_path)
+        return {"message": "Data uploaded successfully"}
+    except Exception as e:
+        raise HTTPException(status_code=422, detail=str(e))
+    
 @app.post("/upload-data-raw/")
 async def upload_data_raw(payload: dict):
     # sourcery skip: raise-from-previous-error
@@ -94,4 +94,4 @@ async def clear_data():
     csv_data = empty_df.to_csv(index=False).encode('utf-8')
     # Overwrite the existing file in GCS with the empty CSV data
     gcs_bucket.blob(gcs_file_path).upload_from_string(csv_data, content_type='text/csv')
-    return {"message": "Data cleared successfully"}
+    return {"message": "Data cleared successfully"}