Update app.py

app.py

@@ -40,7 +40,7 @@ def convert_mp3_to_wav(mp3_data):
     wav_buffer.seek(0)
     return wav_buffer.getvalue()
 
-def process_audio_file(file_data, url):
+def process_audio_file(file_data, url,file_id):
     try:
         sample_rate, wav_data = wavfile.read(BytesIO(file_data))      
         
@@ -81,7 +81,7 @@ def process_audio_file(file_data, url):
         confident_classes = sorted(confident_classes)
         confident_classes_list = list(confident_classes)
         
-        answer_dict = {'url': url, 'answer': confident_classes_list}
+        answer_dict = {'url': url, 'answer': confident_classes_list, "qcUser" : None, "normalfileID", file_id}
         return answer_dict
     
     except Exception as e:
@@ -116,19 +116,20 @@ def process_audio(params):
         return {"error": f"Invalid JSON input: {e.msg} at line {e.lineno} column {e.colno}"}
 
     audio_files = params.get("urls", [])
+    file_ids = params.get("normalfileID",[])
     # api = params.get("api", "")
     # job_id = params.get("job_id", "")
 
     solutions = []
-    for audio_url in audio_files:
+    for audio_url, file_id in zip(audio_files, file_ids):
         audio_data = get_audio_data(audio_url)
 
         if audio_url.endswith(".mp3"):            
             wav_data = convert_mp3_to_wav(audio_data)
-            result = process_audio_file(wav_data, audio_url)
+            result = process_audio_file(wav_data, audio_url, file_id)
 
         elif audio_url.endswith(".wav"):           
-            result = process_audio_file(audio_data, audio_url)
+            result = process_audio_file(audio_data, audio_url, file_id)
         
         if result:
             solutions.append(result)