Update app.py

app.py

@@ -197,7 +197,14 @@ def generate_scene(
     if session_id is None:
         session_id = uuid.uuid4().hex
 
-    print(f"generating scene with {sample_steps} steps")
+    print(f"{session_id} is generating scene with {sample_steps} steps")
+
+    tags = ["<ENDAUDCAP>", "<AUDCAP>", "<E>", "<S>"]
+
+    found = [tag for tag in tags if tag in text_prompt]
+
+    if bool(found):
+        print(f"{session_id} used tages")
     
     return generate_video(text_prompt, 
                         sample_steps, 
@@ -266,8 +273,6 @@ def generate_video(
     if image is not None:
         image_path = image
 
-        
-
     output_dir = os.path.join(os.environ["PROCESSED_RESULTS"], session_id)
     os.makedirs(output_dir, exist_ok=True)
     output_path = os.path.join(output_dir, f"generated_video.mp4")
@@ -295,6 +300,8 @@ def generate_video(
 
     save_video(output_path, generated_video, generated_audio, fps=24, sample_rate=16000)
 
+    print(f"{session_id} video generation succeeded")
+
     return output_path
 
 
@@ -450,4 +457,3 @@ if __name__ == "__main__":
     demo.unload(cleanup)
     demo.queue()
     demo.launch(ssr_mode=False, share=True)
-