fix dataframe

app/main.py

@@ -222,19 +222,25 @@ def extract_features(file_path):
     
     # Combine all features into a single Series
     features = pd.concat([mfccs_mean, spectral_centroids, spectral_rolloff, spectral_flux, spectral_contrast, pitch_mean, zero_crossings])
-    return features
+
+    return pd.DataFrame([features])
 
 
 @app.post("/mlp")
 async def handle_audio(file: UploadFile = File(...)):
     try:
         # Ensure that we are handling an MP3 file
-        if file.content_type not in ["audio/mpeg", "audio/wav", "audio/mp3"]:
-            raise HTTPException(status_code=400, detail="Invalid file type. Please upload an MP3 file.")
+        if file.content_type == "audio/mpeg" or file.content_type == "audio/mp3":
+            file_extension = ".mp3"
+        elif file.content_type == "audio/wav":
+            file_extension = ".wav"
+        else:
+            raise HTTPException(status_code=400, detail="Invalid file type. Supported types: MP3, WAV.")
 
         # Read the file's content
         contents = await file.read()
-        temp_filename = f"app/{uuid4().hex}.mp3"
+        temp_filename = f"app/{uuid4().hex}{file_extension}"
+
 
         # Save file to a temporary file if needed or process directly from memory
         with open(temp_filename, "wb") as f: