Update flask_api_full_song.py

flask_api_full_song.py

@@ -17,6 +17,8 @@ secret_key = os.getenv('SECRET_KEY')
 region = 'na-siliconvalley'
 bucket_name = 'xiaohei-cat-ai-1304646510'
 
+print("Starting download the model and config...")
+
 config = CosConfig(Region=region, SecretId=secret_id, SecretKey=secret_key)
 client = CosS3Client(config)
 
@@ -32,6 +34,9 @@ response2 = client.get_object(
 )
 response2['Body'].get_stream_to_file('/tmp/config.json')
 
+print("Download complete!")
+
+print("Starting service...")
 
 model_name = "/tmp/G_full.pth"  # 模型地址
 config_name = "/tmp/config.json"  # config地址
@@ -128,7 +133,10 @@ def get_task_status(task_id):
 
 @app.route('/download/<filename>', methods=['GET'])
 def download(filename):
-    return send_file("/tmp/" + filename, as_attachment=True)
+    if filename == 'G_full.pth' or filename == 'config.json':
+        return jsonify({"error": "File not found"}), 404
+    else:
+        return send_file("/tmp/" + filename, as_attachment=True)
 
 
 if __name__ == '__main__':