Update app.py

app.py

@@ -91,7 +91,7 @@ def generate_single_view(input_images, prompt, seed, num_inference_steps, true_g
 # --- Main Inference Function ---
 @spaces.GPU(duration=300)
 def generate_turnaround(
-    images,
+    image,
     seed=42,
     randomize_seed=False,
     true_guidance_scale=1.0,
@@ -104,23 +104,21 @@ def generate_turnaround(
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     
-    # 入力画像の読み込み
-    pil_images = []
-    if images is not None:
-        for item in images:
-            try:
-                if isinstance(item[0], Image.Image):
-                    pil_images.append(item[0].convert("RGB"))
-                elif isinstance(item[0], str):
-                    pil_images.append(Image.open(item[0]).convert("RGB"))
-                elif hasattr(item, "name"):
-                    pil_images.append(Image.open(item.name).convert("RGB"))
-            except Exception:
-                continue
-    
-    if not pil_images:
+    # 入力画像の確認
+    if image is None:
         return None, None, None, None, seed, "エラー: 入力画像をアップロードしてください"
     
+    # PIL画像として処理
+    if isinstance(image, Image.Image):
+        input_image = image.convert("RGB")
+    else:
+        try:
+            input_image = Image.open(image).convert("RGB")
+        except:
+            return None, None, None, None, seed, "エラー: 画像の読み込みに失敗しました"
+    
+    pil_images = [input_image]
+    
     # 1. 正面立ち絵を生成
     progress(0.25, desc="正面立ち絵を生成中...")
     front_image = generate_single_view(pil_images, PROMPTS["front"], seed, num_inference_steps, true_guidance_scale)
@@ -164,7 +162,7 @@ with gr.Blocks(css=css) as demo:
     
     with gr.Column(elem_id="col-container"):
         with gr.Row():
-            input_images = gr.Image(
+            input_image = gr.Image(
                 label="入力画像（キャラクター画像をアップロード）", 
                 show_label=True, 
                 type="pil", 
@@ -219,7 +217,7 @@ with gr.Blocks(css=css) as demo:
     run_button.click(
         fn=generate_turnaround,
         inputs=[
-            input_images,
+            input_image,
             seed,
             randomize_seed,
             true_guidance_scale,
@@ -229,5 +227,4 @@ with gr.Blocks(css=css) as demo:
     )
 
 if __name__ == "__main__":
-
     demo.launch()