Initial Commit

app.py

@@ -285,6 +285,28 @@ class FluxInterface:
 # Initialize the interface
 interface = FluxInterface()
 
+# 直接将 GPU 装饰器应用在最外层的处理函数上
+@spaces.GPU(duration=300)
+def process_request(input_image, prompt="", guidance_scale=3.5, num_inference_steps=28, num_images=2, seed=None, aspect_ratio="1:1"):
+    """主处理函数，直接处理用户请求"""
+    try:
+        interface = FluxInterface()
+        if interface.models is None:
+            interface.load_models()
+            
+        return interface.generate(
+            input_image=input_image,
+            prompt=prompt,
+            guidance_scale=guidance_scale,
+            num_inference_steps=num_inference_steps,
+            num_images=num_images,
+            seed=seed,
+            aspect_ratio=aspect_ratio
+        )
+    except Exception as e:
+        logger.error(f"Error during generation: {str(e)}")
+        raise gr.Error(f"Generation failed: {str(e)}")
+
 # Create Gradio interface
 with gr.Blocks(
     theme=gr.themes.Soft(),
@@ -409,32 +431,8 @@ with gr.Blocks(
                 - 🎲 Use seeds for reproducible results
             """)
     
-    # Set up the generation function
-    def generate_with_error_handling(*args):
-        try:
-            logger.info("Starting image generation with args: %s", str(args))
-            
-            # 输入参数验证
-            input_image, prompt, guidance, steps, num_images, seed, aspect_ratio = args
-            logger.info(f"Input validation - Image: {type(input_image)}, Prompt: '{prompt}', "
-                       f"Guidance: {guidance}, Steps: {steps}, Num Images: {num_images}, "
-                       f"Seed: {seed}, Aspect Ratio: {aspect_ratio}")
-            
-            if input_image is None:
-                raise ValueError("No input image provided")
-            
-            gr.Info("Starting image generation...")
-            results = interface.generate(*args)
-            logger.info("Generation completed successfully")
-            gr.Info("Generation complete!")
-            return [results, None]
-        except Exception as e:
-            error_msg = str(e)
-            logger.error(f"Error in generate_with_error_handling: {error_msg}", exc_info=True)
-            return [None, error_msg]
-
     submit_btn.click(
-        fn=generate_with_error_handling,
+        fn=process_request,
         inputs=[
             input_image,
             prompt,
@@ -443,11 +441,8 @@ with gr.Blocks(
             num_images,
             seed,
             aspect_ratio
-        ],
-        outputs=[
-            output_gallery,
-            error_message
-        ],
+        ], 
+        outputs=[output_gallery],
         show_progress=True
     )