Update app.py

app.py

@@ -8,7 +8,7 @@ headers = {
     "x-api-key": os.getenv("JIGSAWSTACK_API_KEY")
 }
 
-def detect_objects(image_url=None, file_store_key=None, prompts=None, features=None, annotated_image=False):
+def detect_objects(image_url=None, file_store_key=None, prompts=None, features=None):
     if not image_url and not file_store_key:
         return "❌ Please provide either an image URL or file store key.", "", "", None
 
@@ -89,7 +89,6 @@ with gr.Blocks() as demo:
             # Advanced options
             prompts = gr.Textbox(label="Prompts (comma-separated)", placeholder="wine glass, bottle, cup", info="Targeted object detection prompts")
             features = gr.CheckboxGroup(choices=["object_detection", "gui"], value=["object_detection"], label="Features")
-            annotated_image = gr.Checkbox(label="Return Annotated Image", value=True)
             
             detect_btn = gr.Button("🔍 Detect Objects")
             clear_btn = gr.Button("Clear")
@@ -98,8 +97,8 @@ with gr.Blocks() as demo:
             status_box = gr.Textbox(label="Status", interactive=False)
             desc_display = gr.Textbox(label="Object Details", lines=10, interactive=False)
             
-            # Annotated image display
-            annotated_image_display = gr.Image(label="Annotated Image", visible=False)
+            # Annotated image display - always visible
+            annotated_image_display = gr.Image(label="Annotated Image")
             
             json_box = gr.Accordion("Raw JSON Response", open=False)
             with json_box:
@@ -113,7 +112,7 @@ with gr.Blocks() as demo:
 
     input_type.change(fn=toggle_inputs, inputs=input_type, outputs=[image_url, file_store_key])
 
-    def on_detect(input_mode, url, key, prompts_text, features_list, annotated):
+    def on_detect(input_mode, url, key, prompts_text, features_list):
         # Parse prompts
         prompts_list = None
         if prompts_text.strip():
@@ -123,31 +122,24 @@ with gr.Blocks() as demo:
             return detect_objects(
                 image_url=url.strip(),
                 prompts=prompts_list,
-                features=features_list,
-                annotated_image=annotated
+                features=features_list
             )
         else:
             return detect_objects(
                 file_store_key=key.strip(),
                 prompts=prompts_list,
-                features=features_list,
-                annotated_image=annotated
+                features=features_list
             )
 
-    def update_annotated_image_visibility(annotated):
-        return gr.update(visible=annotated)
-
     detect_btn.click(fn=on_detect, inputs=[
-        input_type, image_url, file_store_key, prompts, features, annotated_image
+        input_type, image_url, file_store_key, prompts, features
     ], outputs=[status_box, desc_display, json_output, annotated_image_display])
 
-    annotated_image.change(fn=update_annotated_image_visibility, inputs=annotated_image, outputs=annotated_image_display)
-
     def clear_all():
-        return "Image URL", "", "", "", "", ["object_detection"], False, "", "", "", None
+        return "Image URL", "", "", "", "", ["object_detection"], "", "", "", None
 
     clear_btn.click(fn=clear_all, inputs=[], outputs=[
-        input_type, image_url, file_store_key, prompts, features, annotated_image,
+        input_type, image_url, file_store_key, prompts, features,
         status_box, desc_display, json_output, annotated_image_display
     ])