Update app.py

app.py

@@ -41,7 +41,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))
         if wav_data.ndim > 1:
@@ -62,7 +62,7 @@ def process_audio_file(file_data, url):
         if inferred_class == "Silence" and len(top_two_indices) > 1:
             inferred_class = class_names[top_two_indices[1]]
 
-        answer_dict = {'url': url, 'answer': [inferred_class]}
+        answer_dict = {'url': url, 'answer': [inferred_class], qcUser: None, "normalfileID": file_id}
         return answer_dict
     except Exception as e:
         logging.error(f"Error processing {url}: {e}")
@@ -96,19 +96,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)