Update app.py

app.py

@@ -163,12 +163,16 @@ with gr.Blocks(css=css) as demo:
                     value="Short Formal 📝",
                     elem_classes="gr-box"
                 )
+
                 custom_instruction = gr.Textbox(
                     label="Custom Instruction (Only for 'Other')", visible=False, elem_classes="gr-box"
                 )
                 submit_btn = gr.Button(value="Submit", elem_classes="gr-box")
+
             with gr.Column():
                 output_text = gr.Markdown(label="Output Text", show_copy_button=True, elem_classes="output-text")
+                redirect_message = gr.Markdown(visible=False, elem_classes="redirect-message")
+                test_counter_display = gr.Markdown(visible=True, label="Test Counter")
 
         # Toggle visibility of custom instruction based on selected type
         def toggle_custom_instruction(type_selection):
@@ -218,12 +222,9 @@ with gr.Blocks(css=css) as demo:
         submit_btn.click(
             handle_submit,
             inputs=[input_img, description_type, custom_instruction],
-            outputs=[output_text, redirect_message, submit_btn, gr.Markdown(visible=True, label="Test Counter")],
+            outputs=[output_text, redirect_message, submit_btn, test_counter_display],
         )
 
-# Launch Gradio app
-demo.queue(api_open=False)
-demo.launch(debug=True)
 
 # Launch Gradio app
 demo.queue(api_open=False)