adding queue system, rotate transformations, and scroll to the top when click on submit button

app.py

@@ -17,8 +17,9 @@ concepts = load_concepts()
 metadata = load_metadata()
 
 demo = build_ui(concepts, metadata, HF_API_TOKEN, HF_DATASET_NAME)
+demo.queue(max_size=50, default_concurrency_limit=5)  # Enable queue system with max 30 requests
 # demo.launch()
-demo.launch(debug=False)
+demo.launch(debug=False, server_port=7861)
 
 demo.close()
 # gr.close_all()

logic/image_transforms.py

@@ -0,0 +1,80 @@
+import cv2
+import numpy as np
+import gradio as gr
+
+
+def rotate_image_90_left(image):
+    """
+    Rotate image 90 degrees counter-clockwise (left)
+    
+    Args:
+        image: Input image as numpy array
+        
+    Returns:
+        Rotated image or None if input is None
+    """
+    if image is None:
+        return None
+    
+    try:
+        return cv2.rotate(image, cv2.ROTATE_90_COUNTERCLOCKWISE)
+    except Exception as e:
+        print(f"Error rotating image left: {e}")
+        return image
+
+
+def rotate_image_90_right(image):
+    """
+    Rotate image 90 degrees clockwise (right)
+    
+    Args:
+        image: Input image as numpy array
+        
+    Returns:
+        Rotated image or None if input is None
+    """
+    if image is None:
+        return None
+    
+    try:
+        return cv2.rotate(image, cv2.ROTATE_90_CLOCKWISE)
+    except Exception as e:
+        print(f"Error rotating image right: {e}")
+        return image
+
+
+def rotate_image_180(image):
+    """
+    Rotate image 180 degrees
+    
+    Args:
+        image: Input image as numpy array
+        
+    Returns:
+        Rotated image or None if input is None
+    """
+    if image is None:
+        return None
+    
+    try:
+        return cv2.rotate(image, cv2.ROTATE_180)
+    except Exception as e:
+        print(f"Error rotating image 180 degrees: {e}")
+        return image
+
+
+def reset_image_to_original(current_image, original_image):
+    """
+    Reset image to its original state
+    
+    Args:
+        current_image: Current modified image (fallback if no original)
+        original_image: Original image to restore (Gradio State object)
+        
+    Returns:
+        Original image if available, otherwise current image (may be None)
+    """
+    if original_image is None:
+        return current_image
+    
+    return original_image.value

ui/layout.py

@@ -6,6 +6,7 @@ from logic.data_utils import CustomHFDatasetSaver
 from data.lang2eng_map import lang2eng_mapping
 from gradio_modal import Modal
 from logic.handlers import *
+from logic.image_transforms import rotate_image_90_left, rotate_image_90_right, reset_image_to_original
 from config.settings import *
 from functools import partial
 
@@ -39,6 +40,17 @@ function() {
 }
 """
 
+scroll_to_top_js = """
+function() {
+    // Scroll to the top of the page smoothly
+    window.scrollTo({
+        top: 0,
+        behavior: 'smooth'
+    });
+    return "";
+}
+"""
+
 def login_user(email, password):
     result = auth_handler.login(email, password)
     if result['success']:
@@ -324,6 +336,11 @@ def build_ui(concepts_dict, metadata_dict, HF_API_TOKEN, HF_DATASET_NAME):
         modal_submit = cmp_main_ui["modal_submit"]
         vlm_cancel_btn = cmp_main_ui["vlm_cancel_btn"]
         vlm_model_dropdown = cmp_main_ui["vlm_model_dropdown"]
+        
+        # Image rotation buttons
+        rotate_left_btn = cmp_main_ui["rotate_left_btn"]
+        rotate_right_btn = cmp_main_ui["rotate_right_btn"]
+        reset_image_btn = cmp_main_ui["reset_image_btn"]
 
         # dictionary to store ALL vlm_outputs and vlm_models by exampleid
         vlm_captions = gr.State(None)
@@ -442,6 +459,25 @@ def build_ui(concepts_dict, metadata_dict, HF_API_TOKEN, HF_DATASET_NAME):
             outputs=[image_inp] # is_blurred
         )
 
+        # ============= Image Rotation ============= #
+        rotate_left_btn.click(
+            fn=rotate_image_90_left,
+            inputs=[image_inp],
+            outputs=[image_inp]
+        )
+
+        rotate_right_btn.click(
+            fn=rotate_image_90_right,
+            inputs=[image_inp],
+            outputs=[image_inp]
+        )
+
+        reset_image_btn.click(
+            fn=reset_image_to_original,
+            inputs=[image_inp, ori_img],
+            outputs=[image_inp]
+        )
+
         # ===============================
         with gr.Column(visible=False, elem_id="browse_data") as browse_data_placeholder:
             # Browse Data
@@ -600,7 +636,8 @@ def build_ui(concepts_dict, metadata_dict, HF_API_TOKEN, HF_DATASET_NAME):
         # submit_btn.click(lambda: Modal(visible=True), None, modal_vlm)
         submit_btn.click(submit_button_clicked,
                         inputs=[vlm_output],
-                        outputs=[modal_vlm, modal_submit])
+                        outputs=[modal_vlm, modal_submit],
+                        js=scroll_to_top_js)
 
         # submit_btn.click(partial(submit_button_clicked, save_fn=hf_writer.save, 
         #                             data_outputs=data_outputs),
@@ -714,7 +751,9 @@ def build_ui(concepts_dict, metadata_dict, HF_API_TOKEN, HF_DATASET_NAME):
         gen_button.click(
             fn=generate_vlm_caption, # processor=processor, model=model
             inputs=[image_inp, vlm_model_dropdown],
-            outputs=[vlm_output, vlm_feedback, vlm_done_btn, vlm_no_btn, gen_button, vlm_model_dropdown]
+            outputs=[vlm_output, vlm_feedback, vlm_done_btn, vlm_no_btn, gen_button, vlm_model_dropdown],
+            concurrency_limit=1,
+            concurrency_id="vlm_queue"
         )
         # vlm_output.change(
         #     fn=lambda : gr.update(interactive=False) if vlm_output.value else gr.update(interactive=True),

ui/main_page.py

@@ -87,6 +87,10 @@ def build_main_page(concepts_dict, metadata_dict, local_storage):
                         hide_all_faces_btn = gr.Button("👤 Hide All Faces", elem_id="hide_all_faces_btn")
                         hide_faces_btn = gr.Button("👤 Hide Specific Faces", elem_id="hide_faces_btn")
                         unhide_faces_btn = gr.Button("👀 Unhide Faces", elem_id="unhide_faces_btn")
+                    with gr.Row():
+                        rotate_left_btn = gr.Button("↺ Rotate Left", elem_id="rotate_left_btn")
+                        rotate_right_btn = gr.Button("↻ Rotate Right", elem_id="rotate_right_btn")
+                        reset_image_btn = gr.Button("🔄 Reset Image", elem_id="reset_image_btn")
 
             with gr.Column():
                 # short_caption_inp = gr.Textbox(lines=2, label="Short Description", elem_id="short_caption_inp")
@@ -252,6 +256,9 @@ def build_main_page(concepts_dict, metadata_dict, local_storage):
         "hide_faces_btn": hide_faces_btn,
         "hide_all_faces_btn": hide_all_faces_btn,
         "unhide_faces_btn": unhide_faces_btn,
+        "rotate_left_btn": rotate_left_btn,
+        "rotate_right_btn": rotate_right_btn,
+        "reset_image_btn": reset_image_btn,
         "exclude_btn": exclude_btn,
         "modal_exclude_confirm": modal_exclude_confirm,
         "cancel_exclude_btn": cancel_exclude_btn,