Update app.py

add toggle for tags

app.py

@@ -174,7 +174,7 @@ def generate(text, file_name, model="gemini-2.0-flash-exp"):
                 print(f"Error deleting uploaded file {file.name}: {cleanup_e}")
 
 # Main processing function for Gradio
-def process_image_and_prompt(composite_pil, prompt):
+def process_image_and_prompt(composite_pil, prompt, enable_tagging=True):
     composite_path = None # Initialize input temp file path for finally block
     output_image_path = None # Initialize output temp file path for finally block
     try:
@@ -191,15 +191,20 @@ def process_image_and_prompt(composite_pil, prompt):
         model = "gemini-2.0-flash-exp" # Specify the model here
 
         # 2. Call get_image_tags to get tags from the original image
-        tagging_prompt = "Analyze this image. Provide a JSON object containing a single key, 'tags', whose value is a JSON array of strings, representing relevant keywords or tags for the image content. Example: {\"tags\": [\"apple\", \"fruit\", \"red\"]}. Provide ONLY the JSON object and nothing else."
-        tag_json_string = get_image_tags(file_name, tagging_prompt, model=model)
+        tag_json_string = ""
+        if enable_tagging:
+            tagging_prompt = "Analyze this image. Provide a JSON object containing a single key, 'tags', whose value is a JSON array of strings, representing relevant keywords or tags for the image content. Example: {\"tags\": [\"apple\", \"fruit\", \"red\"]}. Provide ONLY the JSON object and nothing else."
+            tag_json_string = get_image_tags(file_name, tagging_prompt, model=model)
 
         # 3. Call generate for the main image processing based on the user prompt
         output_image_path, main_text_response = generate(text=prompt, file_name=file_name, model=model)
 
         # 4. Combine the tag JSON string and the main text response
         # Format the output clearly
-        final_text_output = f"{tag_json_string},{main_text_response}"
+        if tag_json_string:
+            final_text_output = f"{tag_json_string},{main_text_response}"
+        else:
+            final_text_output = main_text_response
 
         # 5. Prepare the image output for the Gradio gallery
         result_img = None
@@ -298,6 +303,7 @@ with gr.Blocks( # css_paths="style.css", # Тимчасово закоменту
                 label="Prompt for Gemini",
                 elem_classes="prompt-input"
             )
+            with_tags = gr.Checkbox(label="Enable Tagging", value=True)
             submit_btn = gr.Button("Generate", elem_classes="generate-btn")
 
         with gr.Column(elem_classes="output-column"):
@@ -312,7 +318,7 @@ with gr.Blocks( # css_paths="style.css", # Тимчасово закоменту
     # Connect the button click to the updated processing function
     submit_btn.click(
         fn=process_image_and_prompt,
-        inputs=[image_input, prompt_input],
+        inputs=[image_input, prompt_input, with_tags],
         outputs=[output_gallery, output_text],
     )
 
@@ -335,4 +341,4 @@ with gr.Blocks( # css_paths="style.css", # Тимчасово закоменту
         elem_id="examples-grid"
     )
 
-demo.queue(max_size=50).launch()
+demo.queue(max_size=50).launch()