Update app.py

app.py

@@ -17,6 +17,7 @@ from elevenlabs import set_api_key, generate, play, clone, Voice, VoiceSettings
 # Initialize the Flask app
 app = Flask(__name__)
 TEMP_DIR = None
+VIDEO_DIRECTORY = None
 
 def run_inference(video_path, audio_path, video_out_path,
                   inference_ckpt_path, unet_config_path="configs/unet/second_stage.yaml",
@@ -110,6 +111,7 @@ def generate_audio(voice_cloning, text_prompt):
 @app.route('/run', methods=['POST'])
 def generate_video():
     global TEMP_DIR
+    global VIDEO_DIRECTORY
     TEMP_DIR = create_temp_dir()
 
     if 'video' not in request.files:
@@ -147,9 +149,25 @@ def generate_video():
             seed=int(request.form.get('seed', 1247))
         )
         # Return the output video path or further process the file for download
-        return jsonify({'output_video': output_video}), 200
+        if output_video and output_video.endswith('.mp4'):
+            filename = os.path.basename(output_video)
+            os.makedirs('videos', exist_ok=True)
+            VIDEO_DIRECTORY = os.path.abspath('videos')
+            print("VIDEO_DIRECTORY: ",VIDEO_DIRECTORY)
+            destination_path = os.path.join(VIDEO_DIRECTORY, filename)
+            shutil.copy(output_video, destination_path)
+            video_url = f"/videos/{filename}"
+            
+        return jsonify({"message": "Video processed and saved successfully.",
+                        "output_video": video_url,
+                       "status": "success"}), 200
     except Exception as e:
         return jsonify({'error': str(e)}), 500
+
+@app.route("/videos/<string:filename>", methods=['GET'])
+def serve_video(filename):
+    global VIDEO_DIRECTORY
+    return send_from_directory(VIDEO_DIRECTORY, filename, as_attachment=False)
  
 @app.route("/health", methods=["GET"])
 def health_status():