Upload infer_serverless.py

infer_serverless.py

@@ -14,7 +14,7 @@ from scipy.io import wavfile
 from datetime import datetime
 from urllib.parse import urlparse
 from mega import Mega
-from flask import Flask, request, jsonify, send_file,render_template
+
 import base64
 import tempfile
 import threading
@@ -28,7 +28,7 @@ from threading import Lock
 from multiprocessing import Process, SimpleQueue, set_start_method,get_context
 from queue import Empty
 from pydub import AudioSegment
-from flask_dance.contrib.google import make_google_blueprint, google
+
 import io
 import runpod
 import boto3
@@ -78,13 +78,6 @@ f0method_mode = ["pm", "harvest", "crepe"]
 f0method_info = "PM is fast, Harvest is good but extremely slow, and Crepe effect is good but requires GPU (Default: PM)"
 
 
-def index():
-    # Check if user is logged in
-    return render_template("ui.html")
-    #if google.authorized:
-    #    return render_template("index.html", logged_in=True)
-    #else:
-    #   return render_template("index.html", logged_in=False)
 
 
 
@@ -181,14 +174,14 @@ def upload_file(local_file_path,bucket_name):
         response=client.upload_file(local_file_path, bucket_name, filename)
         
     except client.exceptions.NoSuchKey:
-        return jsonify({'error': 'File not found in the bucket'}), 404
+        return "error: File not found in the bucket"
     except Exception as e:
-        return jsonify({'error': str(e)}), 500
+        return "error: File not found in the bucket"
 
     # Optional: Send the file directly to the client
     # return send_file(local_file_path, as_attachment=True)
 
-    return jsonify({'success': True, 'message': 'File downloaded successfully', 'file_path': local_file_path})
+    return "success"
 
     
 
@@ -216,22 +209,17 @@ def download_file(filename,bucket_name):
     try:
         client.download_file(bucket_name, object_key, local_file_path)
     except client.exceptions.NoSuchKey:
-        return jsonify({'error': 'File not found in the bucket'}), 404
+        return "file not in buecket"
     except Exception as e:
-        return jsonify({'error': str(e)}), 500
+        return "exception"
 
     # Optional: Send the file directly to the client
     # return send_file(local_file_path, as_attachment=True)
 
-    return jsonify({'success': True, 'message': 'File downloaded successfully', 'file_path': local_file_path})
+    return "success"
     
 
-def list_weights():
-    directory = 'weights'
-    files = os.listdir(directory)
-    # Extract filenames without their extensions
-    filenames = [os.path.splitext(file)[0] for file in files if os.path.isfile(os.path.join(directory, file))]
-    return jsonify(filenames)
+
 
 
 
@@ -242,7 +230,7 @@ def get_status(audio_id):
     # Retrieve the task status using the unique ID
     print(audio_id)
     status_info = task_status_tracker.get(audio_id, {"status": "Unknown ID", "percentage": 0})
-    return jsonify({"audio_id": audio_id, "status": status_info["status"], "percentage": status_info["percentage"]})
+    return "status"
 
 processed_audio_storage = {}
 
@@ -250,7 +238,7 @@ def api_convert_voice(filename,spk_id1,unique_id):
     acquired = request_semaphore.acquire(blocking=False)
     
     if not acquired:
-        return jsonify({"error": "Too many requests, please try again later"}), 429
+        return "error in lock"
     #task_status_tracker[unique_id] = {"status": "Starting", "percentage": 0}
     try:
     
@@ -364,12 +352,6 @@ def get_vc_safe(sid, to_return_protect0):
 
 
 
-def get_processed_audio(audio_id):
-    # Retrieve the path from temporary storage or session
-    if audio_id in processed_audio_storage:
-        file_path = processed_audio_storage[audio_id]
-        return send_file(file_path, as_attachment=True)
-    return jsonify({"error": "File not found."}), 404
 
 def worker(spk_id, input_audio_path, voice_transform, unique_id, output_queue):
     try: