Rotate image

gradio_demo.py

@@ -67,6 +67,24 @@ if torch.cuda.device_count() > 0:
     else:
         llava_agent = None
 
+def rotate_anti_90(image_array):
+    if image_array is None:
+        raise gr.Error("Please provide an image to rotate.")
+
+    image_pil = Image.fromarray(image_array)
+    image_pil = Image.rotate(image_pil, -90)
+    image_array = np.array(image_pil)
+    return image_array
+
+def rotate_90(image_array):
+    if image_array is None:
+        raise gr.Error("Please provide an image to rotate.")
+
+    image_pil = Image.fromarray(image_array)
+    image_pil = Image.rotate(image_pil, 90)
+    image_array = np.array(image_pil)
+    return image_array
+
 def update_seed(is_randomize_seed, seed):
     if is_randomize_seed:
         return random.randint(0, 2147483647)
@@ -390,7 +408,7 @@ def restore(
     hours = math.floor(minutes / 60)
     minutes = minutes - (hours * 60)
     information = ("Start the process again if you want a different result. " if randomize_seed else "") + \
-    "Wait " + str(allocation) + " min before a new run to avoid quota penalty. " + \
+    "Wait " + str(allocation) + " min before a new run to avoid quota penalty or use another computer. " + \
     "The image(s) has(ve) been generated in " + \
     ((str(hours) + " h, ") if hours != 0 else "") + \
     ((str(minutes) + " min, ") if hours != 0 or minutes != 0 else "") + \
@@ -503,8 +521,10 @@ with gr.Blocks(title="SUPIR") as interface:
     gr.HTML(title_html)
 
     input_image = gr.Image(label="Input", show_label=True, type="numpy", height=600, elem_id="image-input")
+    rotate_anti_90_button = gr.Button(value="⤴ Rotate -90°", elem_id="rotate_anti_90_button")
+    rotate_90_button = gr.Button(value="⤵ Rotate +90°", elem_id="rotate_90_button")
     with gr.Group():
-        prompt = gr.Textbox(label="Image description", info="Help the AI understand what the image represents; describe as much as possible; I advise you to write in English as other languages may not be handled", value="", placeholder="A person, walking, in a town, Summer, photorealistic", lines=3)
+        prompt = gr.Textbox(label="Image description", info="Help the AI understand what the image represents; describe as much as possible, especially the details we can't see on the original image; I advise you to write in English as other languages may not be handled", value="", placeholder="A 27 years old woman, walking, in Santiago, morning, Summer, photorealistic", lines=3)
         prompt_hint = gr.HTML("You can use a <a href='"'https://huggingface.co/spaces/MaziyarPanahi/llava-llama-3-8b'"'>LlaVa space</a> to auto-generate the description of your image.")
         upscale = gr.Radio([["x1", 1], ["x2", 2], ["x3", 3], ["x4", 4], ["x5", 5], ["x6", 6], ["x7", 7], ["x8", 8]], label="Upscale factor", info="Resolution x1 to x8", value=2, interactive=True)
         allocation = gr.Radio([["1 min", 1], ["2 min", 2], ["3 min", 3], ["4 min", 4], ["5 min", 5], ["6 min", 6], ["7 min", 7], ["8 min", 8], ["9 min", 9], ["10 min", 10]], label="GPU allocation time", info="lower=May abort run, higher=Quota penalty for next runs", value=6, interactive=True)
@@ -591,6 +611,7 @@ with gr.Blocks(title="SUPIR") as interface:
         event_id = gr.Textbox(label="Event ID", value="", visible=False)
 
     gr.Examples(
+        run_on_click = True,
         fn = stage2_process,
 	    inputs = [
             input_image,
@@ -696,6 +717,22 @@ with gr.Blocks(title="SUPIR") as interface:
     with gr.Row():
         gr.Markdown(claim_md)
     
+    input_image.upload(fn = check, inputs = [
+        input_image
+    ], outputs = [], queue = False, show_progress = False)
+        
+    rotate_anti_90_button.click(fn = rotate_anti_90, inputs = [
+        input_image
+    ], outputs = [
+        input_image
+    ], queue = False, show_progress = False)
+    
+    rotate_90_button.click(fn = rotate_90, inputs = [
+        input_image
+    ], outputs = [
+        input_image
+    ], queue = False, show_progress = False)
+
     denoise_button.click(fn = check, inputs = [
         input_image
     ], outputs = [], queue = False, show_progress = False).success(fn = stage1_process, inputs = [
@@ -724,10 +761,7 @@ with gr.Blocks(title="SUPIR") as interface:
         seed
     ], queue = False, show_progress = False).then(fn = check, inputs = [
         input_image
-    ], outputs = [], queue = False, show_progress = False).success(fn = reset_feedback, inputs = [], outputs = [
-        fb_score,
-        fb_text
-    ], queue = False, show_progress = False).success(fn=stage2_process, inputs = [
+    ], outputs = [], queue = False, show_progress = False).success(fn=stage2_process, inputs = [
         input_image,
         denoise_image,
         prompt,