Update app.py

app.py

@@ -46,42 +46,68 @@ def split_image(input_image, num_splits=8):
 
 @spaces.GPU 
 def infer(prompt, seed=1, randomize_seed=False, num_inference_steps=20, progress=gr.Progress()):
-    progress(0, desc="Starting...")
-    # 프롬프트 템플릿 개선
-    prompt_template = f"A sequence of 8 distinct consecutive frames showing {prompt}. Each frame should be clearly different and show progression of motion from left to right. High quality, detailed animation frames."
+    progress(0, desc="Initializing...")
+    # 각 프레임이 독립적이도록 프롬프트 수정
+    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:
         seed = random.randint(0, MAX_SEED)
     
-    progress(0.3, desc="Generating animation...")
-    generator = torch.Generator().manual_seed(seed)
+    frames = []
+    total_frames = 8
     
-    # 생성 매개변수 조정
-    image = pipe(
-        prompt=prompt_template,
-        num_inference_steps=num_inference_steps,
-        num_images_per_prompt=1,
-        generator=generator,
-        height=320,
-        width=2560,
-        guidance_scale=8.5,  # 품질 향상을 위해 약간 증가
-    ).images[0]
+    # 각 프레임을 개별적으로 생성
+    for i in range(total_frames):
+        progress((i / total_frames) * 0.8, 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)
+        
+        frame = pipe(
+            prompt=frame_prompt,
+            num_inference_steps=num_inference_steps,
+            num_images_per_prompt=1,
+            generator=generator,
+            height=320,
+            width=320,  # 단일 프레임 크기로 수정
+            guidance_scale=7.5,
+        ).images[0]
+        
+        frames.append(frame)
     
-    progress(0.8, desc="Creating GIF...")
+    progress(0.9, desc="Creating GIF...")
     gif_name = f"{uuid.uuid4().hex}-flux.gif"
     
-    # GIF 생성 최적화
-    frames = split_image(image, 8)
-    
-    # 프레임 간 전환을 부드럽게 하기 위한 처리
+    # GIF 생성
     export_to_gif(
         frames, 
         gif_name, 
-        fps=8,  # fps 증가로 더 부드러운 애니메이션
+        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!")
-    return gif_name, image, seed
+    return gif_name, preview_image, seed
+
+def create_preview_image(frames):
+    """프레임들을 가로로 연결하여 미리보기 이미지 생성"""
+    total_width = sum(frame.width for frame in frames)
+    max_height = max(frame.height for frame in frames)
+    
+    preview = Image.new('RGB', (total_width, max_height))
+    x_offset = 0
+    for frame in frames:
+        preview.paste(frame, (x_offset, 0))
+        x_offset += frame.width
+    
+    return preview
 
 # 프롬프트 예제도 더 명확하게 수정
 examples = [