Update app.py

app.py

@@ -1,40 +1,48 @@
 import os
-import shutil
-import tempfile
-from fastapi import FastAPI, UploadFile, File, HTTPException
+from flask import Flask, request, jsonify
 from gradio_client import Client
+import tempfile
+
+
 
-app = FastAPI()
+app = Flask(__name__)
 
 def process_input(input_text):
     # 处理函数
     result = "Processed: " + input_text
     return result
 
-@app.post('/')
-async def api(audio: UploadFile = File(...)):
-    try:
-        if not audio.filename.endswith('.wav'):
-            raise HTTPException(status_code=400, detail="Only WAV files are supported")
-        
+@app.route('/', methods=["POST"])
+@require_api_key
+def api():
+     # try:
+        if 'audio' not in request.files:
+            return jsonify({"error": "No audio file provided"})
+
+        file = request.files['audio']
+        filename = request.form.get('filename')
+
         # Create a temporary directory
         temp_dir = tempfile.mkdtemp()
 
-        # Save the uploaded audio file to the temporary directory
-        destination_path = os.path.join(temp_dir, audio.filename)
-        with open(destination_path, 'wb') as dest_file:
-            shutil.copyfileobj(audio.file, dest_file)
+        # Save the audio file to the temporary directory with the specified filename and .wav extension
+        destination_path = os.path.join(temp_dir, f"{filename}.wav")
+        file.save(destination_path)
+        print(destination_path)
         
         client = Client("https://xxiani-speechbrain-asr-wav2vec2-transformer-aishells.hf.space/")
         result = client.predict(
-            destination_path,    # str (filepath or URL to file) in 'Input' Audio component
-            api_name="/predict"
+             
+                        destination_path,	# str (filepath or URL to file) in 'Input' Audio component
+                        api_name="/predict"
         )
-        
+
         os.remove(destination_path)
         os.rmdir(temp_dir)
-        
-        return {'result': result}
-    except Exception as e:
-        return {'error': str(e)}
+        return jsonify({'result': result})
+
+
+
 
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=app.config.get("PORT", 23456), debug=app.config.get("DEBUG", False))