Update app.py

app.py

@@ -45,9 +45,8 @@ def split_image(input_image, num_splits=8):
     return output_images
 
 @spaces.GPU 
-def infer(prompt, seed=1, randomize_seed=False, num_inference_steps=20, progress=gr.Progress()):
-    progress(0, desc="Initializing...")
-    # 각 프레임이 독립적이도록 프롬프트 수정
+def infer(prompt, seed=1, randomize_seed=False, num_inference_steps=20, progress=gr.Progress(track_tqdm=True)):
+    progress(0, desc="Starting...")
     prompt_template = f"A single clear frame of {prompt}. The scene should show only one moment of the action, high quality, detailed, centered composition."
     
     if randomize_seed:
@@ -56,44 +55,44 @@ def infer(prompt, seed=1, randomize_seed=False, num_inference_steps=20, progress
     frames = []
     total_frames = 8
     
-    # 각 프레임을 개별적으로 생성
+    # 진행 상황을 더 세밀하게 표시
     for i in range(total_frames):
-        progress((i / total_frames) * 0.8, desc=f"Generating frame {i+1}/{total_frames}")
+        current_progress = (i / total_frames) * 0.8
+        progress(current_progress, desc=f"🎨 Generating frame {i+1}/{total_frames}")
         frame_prompt = f"{prompt_template} Frame {i+1} of sequence."
         
-        # 각 프레임에 대해 개별 시드 생성
         frame_seed = seed + i
         generator = torch.Generator().manual_seed(frame_seed)
         
+        # 각 프레임의 생성 단계도 표시
+        for step in range(num_inference_steps):
+            step_progress = current_progress + (step / num_inference_steps) * (0.8 / total_frames)
+            progress(step_progress, desc=f"Frame {i+1}/{total_frames} - Step {step+1}/{num_inference_steps}")
+        
         frame = pipe(
             prompt=frame_prompt,
             num_inference_steps=num_inference_steps,
             num_images_per_prompt=1,
             generator=generator,
             height=320,
-            width=320,  # 단일 프레임 크기로 수정
+            width=320,
             guidance_scale=7.5,
         ).images[0]
         
         frames.append(frame)
+        progress((i + 1) / total_frames * 0.8, desc=f"✅ Completed frame {i+1}/{total_frames}")
     
-    progress(0.9, desc="Creating GIF...")
+    progress(0.9, desc="🎬 Creating GIF...")
     gif_name = f"{uuid.uuid4().hex}-flux.gif"
     
-    # GIF 생성
-    export_to_gif(
-        frames, 
-        gif_name, 
-        fps=8
-    )
+    export_to_gif(frames, gif_name, fps=8)
     
-    # 프레임들을 가로로 연결하여 미리보기 이미지 생성
     total_width = 320 * total_frames
     preview_image = Image.new('RGB', (total_width, 320))
     for i, frame in enumerate(frames):
         preview_image.paste(frame, (i * 320, 0))
     
-    progress(1.0, desc="Done!")
+    progress(1.0, desc="✨ Done!")
     return gif_name, preview_image, seed
 
 def create_preview_image(frames):
@@ -156,23 +155,37 @@ div[class*="examples"] {
     background: transparent !important;
 }
 
-/* 프로그레스 바 스타일 개선 */
+/* 프로그레스 바 스타일 강화 */
 .progress-bar {
     background-color: #f0f0f0;
     border-radius: 10px;
-    padding: 3px;
-    margin: 10px 0;
+    padding: 5px;
+    margin: 15px 0;
+    box-shadow: 0 2px 5px rgba(0,0,0,0.1);
 }
+
 .progress-bar-fill {
     background: linear-gradient(45deg, #FF6B6B, #4ECDC4);
-    height: 20px;
+    height: 25px;
     border-radius: 7px;
     transition: width 0.3s ease-out;
+    box-shadow: 0 2px 5px rgba(0,0,0,0.1);
 }
+
 .progress-text {
     color: black;
-    font-weight: 500;
-    margin-bottom: 5px;
+    font-weight: 600;
+    margin-bottom: 8px;
+    font-size: 1.1em;
+}
+
+/* 진행 상태 텍스트 스타일 */
+.progress-label {
+    display: block;
+    text-align: center;
+    margin-top: 5px;
+    color: #666;
+    font-size: 0.9em;
 }
 """