Update app.py

app.py

@@ -520,55 +520,79 @@ def generate_video():
     else:
         args.device = "cpu"
         
-    generation_thread = threading.Thread(target=main, args=(args,))
-    app.config['generation_thread'] = generation_thread
-    generation_thread.start()
-    response_data = {"message": "Video generation started",
-                    "process_id": generation_thread.ident}
+    # generation_thread = threading.Thread(target=main, args=(args,))
+    # app.config['generation_thread'] = generation_thread
+    # generation_thread.start()
+    # response_data = {"message": "Video generation started",
+    #                 "process_id": generation_thread.ident}
 
-    return jsonify(response_data)
+    try:
+        base64_video, temp_file_path, duration = main(args)
+        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)
 
-@app.route("/status", methods=["GET"])
-def check_generation_status():
-    global TEMP_DIR
-    global start_time
-    response = {"base64_video": "","text_prompt":"", "status": ""}
-    process_id = request.args.get('process_id', None)
-
-    # process_id is required to check the status for that specific process
-    if process_id:
-        generation_thread = app.config.get('generation_thread')
-        if generation_thread and generation_thread.ident == int(process_id) and generation_thread.is_alive():
-            return jsonify({"status": "in_progress"}), 200
-        elif app.config.get('temp_response'):
-            # app.config['temp_response']['status'] = 'completed'
-            final_response = app.config['temp_response']
-            response["base64_video"] = final_response
-            response["text_prompt"] = app.config.get('text_prompt')
-            response["duration"] = app.config.get('final_video_duration')
-            response["status"] = "completed"
-
-            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)
-
-            # TEMP_DIR.cleanup()
-            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.")
+        print("Temporary files cleaned up, but preprocess_data is retained.")
+        
+        return jsonify({
+                'base64_video': base64_video,
+                'text_prompt': text_prompt,
+                'duration': duration,
+                'status': 'completed'
+            })
+        
+    except Exception as e:
+        return jsonify({'status': 'error', 'message': str(e)}), 500
+
+    # return jsonify(response_data)
+
+
+# @app.route("/status", methods=["GET"])
+# def check_generation_status():
+#     global TEMP_DIR
+#     global start_time
+#     response = {"base64_video": "","text_prompt":"", "status": ""}
+#     process_id = request.args.get('process_id', None)
+
+#     # process_id is required to check the status for that specific process
+#     if process_id:
+#         generation_thread = app.config.get('generation_thread')
+#         if generation_thread and generation_thread.ident == int(process_id) and generation_thread.is_alive():
+#             return jsonify({"status": "in_progress"}), 200
+#         elif app.config.get('temp_response'):
+#             # app.config['temp_response']['status'] = 'completed'
+#             final_response = app.config['temp_response']
+#             response["base64_video"] = final_response
+#             response["text_prompt"] = app.config.get('text_prompt')
+#             response["duration"] = app.config.get('final_video_duration')
+#             response["status"] = "completed"
+
+#             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)
+
+#             # TEMP_DIR.cleanup()
+#             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()
-            total_time = round(end_time - start_time, 2)
-            print("Total time taken for execution:", total_time, " seconds")
-            response["time_taken"] = total_time
-            return jsonify(response)
-    return jsonify({"error":"No process id provided"})
+#             end_time = time.time()
+#             total_time = round(end_time - start_time, 2)
+#             print("Total time taken for execution:", total_time, " seconds")
+#             response["time_taken"] = total_time
+#             return jsonify(response)
+#     return jsonify({"error":"No process id provided"})
 
 @app.route("/health", methods=["GET"])
 def health_status():