Update app.py

app.py

@@ -69,17 +69,25 @@ def run_single_image_logic(prompt: str, image_path: Optional[str] = None, progre
         
         progress(0.8, desc="🖼️ 마무리 중...")
         
-        # Handle the output
+        # Handle the output - output is already a URL string or FileObject
         if output:
-            # If output is a FileOutput object, get the URL
+            # Check if output has a url attribute (FileObject)
             if hasattr(output, 'url'):
-                image_url = output.url()
+                # If url is a method, call it; if it's a property, just access it
+                image_url = output.url() if callable(output.url) else output.url
+            # If output is already a string URL
             elif isinstance(output, str):
                 image_url = output
+            # If output is a list of URLs
             elif isinstance(output, list) and len(output) > 0:
-                image_url = output[0]
+                # Check first item in list
+                first_item = output[0]
+                if hasattr(first_item, 'url'):
+                    image_url = first_item.url() if callable(first_item.url) else first_item.url
+                else:
+                    image_url = first_item
             else:
-                raise ValueError("Unexpected output format from Replicate")
+                raise ValueError(f"Unexpected output format from Replicate: {type(output)}")
             
             # Download the image from URL
             response = requests.get(image_url)
@@ -97,6 +105,8 @@ def run_single_image_logic(prompt: str, image_path: Optional[str] = None, progre
     except Exception as e:
         print(f"Error details: {e}")
         print(f"Error type: {type(e)}")
+        print(f"Output value: {output if 'output' in locals() else 'Not set'}")
+        print(f"Output type: {type(output) if 'output' in locals() else 'Not set'}")
         raise gr.Error(f"이미지 생성 실패: {e}")
 
 def run_multi_image_logic(prompt: str, images: List[str], progress=gr.Progress()) -> str:
@@ -133,17 +143,25 @@ def run_multi_image_logic(prompt: str, images: List[str], progress=gr.Progress()
         
         progress(0.8, desc="🖼️ 마무리 중...")
         
-        # Handle the output
+        # Handle the output - output is already a URL string or FileObject
         if output:
-            # If output is a FileOutput object, get the URL
+            # Check if output has a url attribute (FileObject)
             if hasattr(output, 'url'):
-                image_url = output.url()
+                # If url is a method, call it; if it's a property, just access it
+                image_url = output.url() if callable(output.url) else output.url
+            # If output is already a string URL
             elif isinstance(output, str):
                 image_url = output
+            # If output is a list of URLs
             elif isinstance(output, list) and len(output) > 0:
-                image_url = output[0]
+                # Check first item in list
+                first_item = output[0]
+                if hasattr(first_item, 'url'):
+                    image_url = first_item.url() if callable(first_item.url) else first_item.url
+                else:
+                    image_url = first_item
             else:
-                raise ValueError("Unexpected output format from Replicate")
+                raise ValueError(f"Unexpected output format from Replicate: {type(output)}")
             
             # Download the image from URL
             response = requests.get(image_url)
@@ -160,6 +178,8 @@ def run_multi_image_logic(prompt: str, images: List[str], progress=gr.Progress()
             
     except Exception as e:
         print(f"Multi-image error details: {e}")
+        print(f"Output value: {output if 'output' in locals() else 'Not set'}")
+        print(f"Output type: {type(output) if 'output' in locals() else 'Not set'}")
         raise gr.Error(f"이미지 생성 실패: {e}")
 
 # --- Gradio App UI ---