Upload myinfer_latest.py

myinfer_latest.py

@@ -31,6 +31,8 @@ from pydub import AudioSegment
 from flask_dance.contrib.google import make_google_blueprint, google
 import io
 
+import boto3
+
 
 
 
@@ -58,6 +60,8 @@ app.config["GOOGLE_OAUTH_CLIENT_ID"] = "144930881143-n3e3ubers3vkq7jc9doe4iirasg
 app.config["GOOGLE_OAUTH_CLIENT_SECRET"] = "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"
 google_blueprint = make_google_blueprint(scope=["profile", "email"])
 app.register_blueprint(google_blueprint, url_prefix="/login")
+ACCESS_ID = os.getenv('ACCESS_ID', '')
+SECRET_KEY = os.getenv('SECRET_KEY', '')
 
 #set_start_method('spawn', force=True)
 from lib.infer_pack.models import (
@@ -82,10 +86,11 @@ f0method_info = "PM is fast, Harvest is good but extremely slow, and Crepe effec
 @app.route("/")
 def index():
     # Check if user is logged in
-    if google.authorized:
-        return render_template("index.html", logged_in=True)
-    else:
-        return render_template("index.html", logged_in=False)
+    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)
 
 
 
@@ -175,6 +180,39 @@ def create_song():
     # Assuming we're interested in whether any model has been created
     #model_exists = len(models) > 0
     return render_template("ui.html", email=email)    
+    
+@app.route('/download/<filename>', methods=['GET'])
+def download_file(filename):
+    # Configure the client with your credentials
+    session = boto3.session.Session()
+    client = session.client('s3',
+                            region_name='nyc3',
+                            endpoint_url='https://nyc3.digitaloceanspaces.com',
+                            aws_access_key_id=ACCESS_ID,
+                            aws_secret_access_key=SECRET_KEY)
+    
+    # Define the bucket and object key
+    bucket_name = 'sing'  # Your bucket name
+    object_key = f'{filename}'  # Construct the object key
+
+    # Define the local path to save the file
+    local_file_path = os.path.join('weights', filename)
+
+    # Download the file from the bucket
+    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
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+    # 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})
+    
+
+
 @app.route("/logout")
 def logout():
     # Clear the session