Update app.py

app.py

@@ -44,7 +44,7 @@ def duplicate_horizontally(img):
 mask = Image.open("mask_square.png")
 
 @spaces.GPU
-def generate(image, prompt_user):
+def generate(image, prompt_user, progress=gr.Progress(track_tqdm=True)):
     prompt_structure = "The two-panel image showcases the logo of a brand, [LEFT] the left panel is showing the logo [RIGHT] the right panel has this logo applied to "
     prompt = prompt_structure + prompt_user
     
@@ -68,19 +68,6 @@ def generate(image, prompt_user):
     image_2 = out.crop((half_width, 0, width, height))
     return image_2
 
-def process_image(input_image, prompt):
-    try:
-        if input_image is None:
-            return None, "Please upload an image first."
-        
-        if not prompt:
-            return None, "Please provide a prompt."
-            
-        result = generate(input_image, prompt)
-        return result, "Generation completed successfully!"
-    except Exception as e:
-        return None, f"Error during generation: {str(e)}"
-
 with gr.Blocks() as demo:
     gr.Markdown("# Logo in Context")
     gr.Markdown("### In-Context LoRA + Image-to-Image, apply your logo to anything")
@@ -101,10 +88,6 @@ with gr.Blocks() as demo:
         
         with gr.Column():
             output_image = gr.Image(label="Generated Application")
-            status_text = gr.Textbox(
-                label="Status",
-                interactive=False
-            )
     
     with gr.Row():
         gr.Markdown("""
@@ -118,7 +101,7 @@ with gr.Blocks() as demo:
     
     # Set up the click event
     generate_btn.click(
-        fn=process_image,
+        fn=generate,
         inputs=[input_image, prompt_input],
         outputs=[output_image]
     )