Update app.py

app.py

@@ -80,6 +80,7 @@ def infer_video_depth(
     grayscale: bool = True,
     convert_from_color: bool = True,
     blur: float = 0.3,
+    loop_factor: int = 1,  # Neuer Parameter
     output_dir: str = './outputs',
     input_size: int = 518,
 ):
@@ -160,6 +161,110 @@ def infer_video_depth(
         ]
         subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         os.replace(temp_audio_path, stitched_video_path)
+    
+    # Nachdem die Videos erstellt wurden, wenden wir den Loop-Faktor an
+    if loop_factor > 1:
+        depth_looped_path = os.path.join(output_dir, os.path.splitext(os.path.basename(depth_vis_path))[0] + f'_loop{loop_factor}.mp4')
+        
+        # Erstelle eine temporäre Textdatei mit der Liste der zu wiederholenden Dateien
+        concat_file_path = os.path.join(output_dir, 'concat_list.txt')
+        with open(concat_file_path, 'w') as f:
+            for _ in range(loop_factor):
+                f.write(f"file '{depth_vis_path}'\n")
+        
+        # Verwende ffmpeg, um das Video zu wiederholen ohne Neucodierung
+        cmd = [
+            "ffmpeg",
+            "-y",
+            "-f", "concat",
+            "-safe", "0",
+            "-i", concat_file_path,
+            "-c", "copy",
+            depth_looped_path
+        ]
+        subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        
+        # Ersetze den ursprünglichen Pfad durch den neuen geloopten Pfad
+        depth_vis_path = depth_looped_path
+        
+        if stitch and stitched_video_path:
+            # Speichern wir den Originalnamen
+            original_path = stitched_video_path
+            
+            # Temporärer Pfad für das geloopte Video
+            temp_looped_path = os.path.join(output_dir, 'temp_looped_rgbd.mp4')
+            
+            # Erstelle eine temporäre Textdatei für die stitched Videos
+            concat_stitched_file_path = os.path.join(output_dir, 'concat_stitched_list.txt')
+            with open(concat_stitched_file_path, 'w') as f:
+                for _ in range(loop_factor):
+                    f.write(f"file '{original_path}'\n")
+            
+            # Verwende ffmpeg, um das stitched Video zu loopen
+            cmd = [
+                "ffmpeg",
+                "-y",
+                "-f", "concat",
+                "-safe", "0",
+                "-i", concat_stitched_file_path,
+                "-c", "copy",
+                temp_looped_path
+            ]
+            subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+            
+            # Extrahiere den Audio-Track aus dem ursprünglichen Input-Video
+            audio_path = os.path.join(output_dir, 'extracted_audio.aac')
+            extract_audio_cmd = [
+                "ffmpeg",
+                "-y",
+                "-i", input_video,
+                "-vn", "-acodec", "copy",
+                audio_path
+            ]
+            subprocess.run(extract_audio_cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+            
+            # Erstelle eine Textdatei für das Audio-Looping
+            concat_audio_file_path = os.path.join(output_dir, 'concat_audio_list.txt')
+            with open(concat_audio_file_path, 'w') as f:
+                for _ in range(loop_factor):
+                    f.write(f"file '{audio_path}'\n")
+            
+            # Erstelle den geloopten Audio-Track
+            looped_audio_path = os.path.join(output_dir, 'looped_audio.aac')
+            audio_loop_cmd = [
+                "ffmpeg",
+                "-y",
+                "-f", "concat",
+                "-safe", "0",
+                "-i", concat_audio_file_path,
+                "-c", "copy",
+                looped_audio_path
+            ]
+            subprocess.run(audio_loop_cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+            
+            # Kombiniere das geloopte Video mit dem geloopten Audio
+            # Aber schreibe zurück in den originalen Pfad
+            final_cmd = [
+                "ffmpeg",
+                "-y",
+                "-i", temp_looped_path,
+                "-i", looped_audio_path,
+                "-c:v", "copy",
+                "-c:a", "aac",
+                "-map", "0:v:0",
+                "-map", "1:a:0",
+                "-shortest",
+                original_path  # Verwenden des originalen Pfads als Ziel
+            ]
+            subprocess.run(final_cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+            
+            # Wir verwenden weiterhin den originalen Pfad
+            # (stitched_video_path bleibt unverändert)
+            
+            # Bereinige temporäre Dateien
+            for file_path in [concat_file_path, concat_stitched_file_path, concat_audio_file_path, audio_path, looped_audio_path, temp_looped_path]:
+                if os.path.exists(file_path):
+                    os.remove(file_path)
 
     gc.collect()
     torch.cuda.empty_cache()
@@ -193,6 +298,8 @@ def construct_demo():
                     grayscale_option = gr.Checkbox(label="Output Depth as Grayscale", value=True)
                     convert_from_color_option = gr.Checkbox(label="Convert Grayscale from Color", value=True)
                     blur_slider = gr.Slider(minimum=0, maximum=1, step=0.01, label="Depth Blur (can reduce edge artifacts on display)", value=0.3)
+                    # Füge den Loop-Faktor Slider hinzu
+                    loop_factor = gr.Slider(label="Loop Factor (repeats the output video)", minimum=1, maximum=20, value=1, step=1)
                 generate_btn = gr.Button("Generate")
             with gr.Column(scale=2):
                 pass
@@ -201,8 +308,8 @@ def construct_demo():
         
         generate_btn.click(
             fn=infer_video_depth,
-            inputs=[input_video, max_len, target_fps, max_res, stitch_option, grayscale_option, convert_from_color_option, blur_slider],
-            outputs=[depth_vis_video, stitched_video],  # Changed to only include the outputs we want to show
+            inputs=[input_video, max_len, target_fps, max_res, stitch_option, grayscale_option, convert_from_color_option, blur_slider, loop_factor],  # loop_factor hinzugefügt
+            outputs=[depth_vis_video, stitched_video],
         )
     
     return demo