Update app.py

app.py

@@ -74,6 +74,9 @@ CORS(app)
 
 TEMP_DIR = None
 start_time = None
+VIDEO_DIRECTORY = None
+args = None
+unique_id = None
 
 app.config['temp_response'] = None
 app.config['generation_thread'] = None
@@ -288,7 +291,7 @@ def remove_brackets(text):
 
 @app.route("/run", methods=['POST'])
 def generate_video():
-    global start_time
+    global start_time, VIDEO_DIRECTORY
     start_time = time.time()
     global TEMP_DIR
     TEMP_DIR = create_temp_dir()
@@ -395,31 +398,48 @@ def generate_video():
         final_video_path = app.config['final_video_path']
         print('final_video_path',final_video_path)
 
-        if final_video_path and os.path.exists(final_video_path):
-            os.remove(final_video_path)
-            print("Deleted video file:", final_video_path)
-  
-        preprocess_dir = os.path.join("/tmp", "preprocess_data")
-        custom_cleanup(TEMP_DIR.name, preprocess_dir)
-
-        print("Temporary files cleaned up, but preprocess_data is retained.")
-
-        end_time = time.time()
-        time_taken = end_time - start_time
+        if temp_file_path and temp_file_path.endswith('.mp4'):
+            filename = os.path.basename(temp_file_path)
+            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(temp_file_path, destination_path)
+            video_url = f"/videos/{filename}"
+
+            if final_video_path and os.path.exists(final_video_path):
+                os.remove(final_video_path)
+                print("Deleted video file:", final_video_path)
     
-        print(f"Time taken for endpoint: {time_taken:.2f} seconds")
+            preprocess_dir = os.path.join("/tmp", "preprocess_data")
+            custom_cleanup(TEMP_DIR.name, preprocess_dir)
+
+            print("Temporary files cleaned up, but preprocess_data is retained.")
+            end_time = time.time()
+            time_taken = end_time - start_time
         
-        return jsonify({
-                'base64_video': base64_video,
-                'text_prompt': text_prompt,
-                'time_taken':time_taken,
-                'status': 'completed'
-            })
+            print(f"Time taken for endpoint: {time_taken:.2f} seconds")
+            
+            return jsonify({
+                    "message": "Video processed and saved successfully.",
+                    "video_url": video_url,
+                    "status": "success"
+                })
+        else:
+            return jsonify({
+                "message": "Failed to process the video.",
+                "status": "error"
+            }), 500
         
     except Exception as e:
         return jsonify({'status': 'error', 'message': 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)
 
-    # return jsonify(response_data)
 
 
 # @app.route("/status", methods=["GET"])