Update app.py

app.py

@@ -12,6 +12,7 @@ from pydub import AudioSegment
 import io
 from io import BytesIO
 
+
 # Load the model
 model = hub.load('Audio_Multiple_v1')
 
@@ -61,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 = {'audio_url': url, 'answer': [inferred_class]}
+        answer_dict = {'url': url, 'answer': [inferred_class]}
         return answer_dict
     except Exception as e:
         logging.error(f"Error processing {url}: {e}")
@@ -72,21 +73,21 @@ def get_audio_data(url):
     response.raise_for_status()
     return response.content
 
-def send_results_to_api(data, result_url):
-    headers = {"Content-Type": "application/json"}
-    try:
-        response = requests.post(result_url, json=data, headers=headers)
-        response.raise_for_status()  # Raise error for non-200 responses
-        return response.json()  # Return any JSON response from the API
-    except requests.exceptions.HTTPError as http_err:
-        logging.error(f"HTTP error occurred: {http_err}")
-        return {"error": f"HTTP error occurred: {http_err}"}
-    except requests.exceptions.RequestException as req_err:
-        logging.error(f"Request error occurred: {req_err}")
-        return {"error": f"Request error occurred: {req_err}"}
-    except ValueError as val_err:
-        logging.error(f"Error decoding JSON response: {val_err}")
-        return {"error": f"Error decoding JSON response: {val_err}"}
+# def send_results_to_api(data, result_url):
+#     headers = {"Content-Type": "application/json"}
+#     try:
+#         response = requests.post(result_url, json=data, headers=headers)
+#         response.raise_for_status()  # Raise error for non-200 responses
+#         return response.json()  # Return any JSON response from the API
+#     except requests.exceptions.HTTPError as http_err:
+#         logging.error(f"HTTP error occurred: {http_err}")
+#         return {"error": f"HTTP error occurred: {http_err}"}
+#     except requests.exceptions.RequestException as req_err:
+#         logging.error(f"Request error occurred: {req_err}")
+#         return {"error": f"Request error occurred: {req_err}"}
+#     except ValueError as val_err:
+#         logging.error(f"Error decoding JSON response: {val_err}")
+#         return {"error": f"Error decoding JSON response: {val_err}"}
 
 def process_audio(params):
     try:
@@ -95,8 +96,8 @@ def process_audio(params):
         return {"error": f"Invalid JSON input: {e.msg} at line {e.lineno} column {e.colno}"}
 
     audio_files = params.get("audio_files", [])
-    api = params.get("api", "")
-    job_id = params.get("job_id", "")
+    # api = params.get("api", "")
+    # job_id = params.get("job_id", "")
 
     solutions = []
     for audio_url in audio_files:
@@ -112,15 +113,15 @@ def process_audio(params):
         if result:
             solutions.append(result)
 
-    result_url = f"{api}/{job_id}"
-    send_results_to_api(solutions, result_url)
+    # result_url = f"{api}/{job_id}"
+    # send_results_to_api(solutions, result_url)
 
-    return json.dumps({"solutions": solutions}, indent=4)
+    return json.dumps({"solutions": solutions})
 
 import gradio as gr
 
-inputt = gr.Textbox(label="Parameters (JSON format) Eg. {'audio_files':['file1.mp3','file2.wav'], 'api':'https://api.example.com', 'job_id':'12345'}")
+inputt = gr.Textbox(label="Parameters (JSON format) Eg. {'audio_files':['file1.mp3','file2.wav']}")
 outputs = gr.JSON()
 
 application = gr.Interface(fn=process_audio, inputs=inputt, outputs=outputs, title="Audio Classification with API Integration")
-application.launch(share=True)
+application.launch()