Update app.py

app.py

@@ -44,8 +44,14 @@ class ProfessionalCartoonFilmGenerator:
         
         # Use a persistent directory for Hugging Face Spaces
         # Files here can be served by Gradio
-        self.temp_dir = "./outputs"
-        os.makedirs(self.temp_dir, exist_ok=True)
+        self.output_dir = "./outputs"
+        os.makedirs(self.output_dir, exist_ok=True)
+        print(f"📁 Created output directory: {self.output_dir}")
+        
+        # Also create subdirectories for organization
+        os.makedirs(os.path.join(self.output_dir, "characters"), exist_ok=True)
+        os.makedirs(os.path.join(self.output_dir, "backgrounds"), exist_ok=True)
+        os.makedirs(os.path.join(self.output_dir, "videos"), exist_ok=True)
         
         # Model configurations for ZeroGPU optimization
         self.models_loaded = False
@@ -561,7 +567,7 @@ class ProfessionalCartoonFilmGenerator:
                     else:
                         raise e
                 
-                char_path = f"{self.temp_dir}/character_{character['name'].replace(' ', '_')}.png"
+                char_path = f"{self.output_dir}/characters/character_{character['name'].replace(' ', '_')}.png"
                 image.save(char_path)
                 character_images[character['name']] = char_path
                 
@@ -659,7 +665,7 @@ class ProfessionalCartoonFilmGenerator:
                     else:
                         raise e
                 
-                bg_path = f"{self.temp_dir}/background_scene_{scene['scene_number']}.png"
+                bg_path = f"{self.output_dir}/backgrounds/background_scene_{scene['scene_number']}.png"
                 image.save(bg_path)
                 background_images[scene['scene_number']] = bg_path
                 
@@ -798,7 +804,7 @@ class ProfessionalCartoonFilmGenerator:
             # Use the optimization function to ensure CLIP compatibility
             prompt = self.optimize_prompt_for_clip(prompt)
             
-            video_path = f"{self.temp_dir}/scene_{scene['scene_number']}.mp4"
+            video_path = f"{self.output_dir}/videos/scene_{scene['scene_number']}.mp4"
             
             # Get the correct Open-Sora directory
             current_dir = os.getcwd()
@@ -813,7 +819,7 @@ class ProfessionalCartoonFilmGenerator:
                 "torchrun", "--nproc_per_node", "1", "--standalone",
                 "scripts/diffusion/inference.py",
                 "configs/diffusion/inference/t2i2v_256px.py",
-                "--save-dir", self.temp_dir,
+                "--save-dir", self.output_dir,
                 "--prompt", prompt,
                 "--num_frames", "25",  # ~1 second at 25fps
                 "--aspect_ratio", "4:3",
@@ -824,9 +830,9 @@ class ProfessionalCartoonFilmGenerator:
             
             if result.returncode == 0:
                 # Find generated video file
-                for file in os.listdir(self.temp_dir):
+                for file in os.listdir(self.output_dir):
                     if file.endswith('.mp4') and 'scene' not in file:
-                        src_path = os.path.join(self.temp_dir, file)
+                        src_path = os.path.join(self.output_dir, file)
                         os.rename(src_path, video_path)
                         return video_path
             
@@ -844,7 +850,7 @@ class ProfessionalCartoonFilmGenerator:
             print(f"❌ No background image for scene {scene_num}")
             return None
             
-        video_path = f"{self.temp_dir}/scene_{scene_num}.mp4"
+        video_path = f"{self.output_dir}/videos/scene_{scene_num}.mp4"
         
         try:
             print(f"🎬 Creating static video for scene {scene_num}...")
@@ -973,7 +979,7 @@ class ProfessionalCartoonFilmGenerator:
             print(f"❌ No background image for scene {scene_num}")
             return None
             
-        video_path = f"{self.temp_dir}/simple_scene_{scene_num}.mp4"
+        video_path = f"{self.output_dir}/videos/simple_scene_{scene_num}.mp4"
         
         try:
             print(f"🎬 Creating simple video for scene {scene_num}...")
@@ -1044,7 +1050,7 @@ class ProfessionalCartoonFilmGenerator:
             background_color = (100, 150, 200)  # Blue-ish color
             
             # Create video
-            video_path = f"{self.temp_dir}/emergency_fallback.mp4"
+            video_path = f"{self.output_dir}/videos/emergency_fallback.mp4"
             fourcc = cv2.VideoWriter_fourcc(*'mp4v')
             fps = 24
             duration = 30  # 30 seconds
@@ -1095,13 +1101,13 @@ class ProfessionalCartoonFilmGenerator:
             print("❌ No videos to merge")
             return None
             
-        final_video_path = f"{self.temp_dir}/professional_cartoon_film.mp4"
+        final_video_path = f"{self.output_dir}/videos/professional_cartoon_film.mp4"
         
         try:
             print("🎞️ Creating professional cartoon film...")
             
             # Create concat file
-            concat_file = f"{self.temp_dir}/concat_list.txt"
+            concat_file = f"{self.output_dir}/videos/concat_list.txt"
             with open(concat_file, 'w') as f:
                 for video in scene_videos:
                     if os.path.exists(video):