Update app.py

app.py

@@ -2,17 +2,16 @@ import gradio as gr
 import spaces
 import torch
 import os
+import gradio as gr
 
 @spaces.GPU
 def greet(n):
     print(zero.device) # <-- 'cuda:0' 🤗
     return f"Hello {zero + n} Tensor"
 
-#demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
-#demo.launch()
 
 
-import gradio as gr
+
 
 IMAGE_FOLDER = "./images"  # Change this to your folder path
 
@@ -29,6 +28,11 @@ def load_image(selected_file):
         return None
     return os.path.join(IMAGE_FOLDER, selected_file)
 
+# generate input image 
+def generate_input_image(image_path,input_choice):
+    print(f"image_path: '{image_path}'")
+    print(f"input_choice: '{input_choice}'")
+    
 # Generation pipeline function using both the source image and text prompt
 def process_and_generate(selected_file, prompt):
     if not selected_file:
@@ -92,7 +96,11 @@ with gr.Blocks() as demo:
         inputs=image_dropdown, 
         outputs=input_preview
     )
-    
+    generate_inp_btn.click(
+        fn=generate_input_image,
+        inputs=[file_input,image_dropdown],
+        outputs=[]
+    )
     generate_btn.click(
         fn=process_and_generate,
         inputs=[image_dropdown, prompt_input],