Update app.py

app.py

@@ -227,6 +227,7 @@ with gr.Blocks(css=css) as demo:
                 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")
+                user_id_display = gr.Markdown(visible=True, label="User ID")
 
         # Toggle visibility of custom instruction based on selected type
         def toggle_custom_instruction(type_selection):
@@ -255,7 +256,7 @@ with gr.Blocks(css=css) as demo:
                         gr.update(visible=False),
                         gr.update(interactive=True),
                         f"You have {remaining_tests} free test(s) remaining today.",
-                        cookie,  # This cookie is not being set in the response
+                        f"User ID: {user_id}"
                     )
                 else:
                     redirect_text = f"""
@@ -281,7 +282,7 @@ with gr.Blocks(css=css) as demo:
                         gr.update(visible=True, value=redirect_text),
                         gr.update(interactive=False),
                         "You've used all your free tests for today. We invite you to book a demo for full access!",
-                        cookie,  # This cookie is not being set in the response
+                        f"User ID: {user_id}"
                     )
             else:
                 redirect_text = f"""
@@ -307,23 +308,16 @@ with gr.Blocks(css=css) as demo:
                     gr.update(visible=True, value=redirect_text),
                     gr.update(interactive=False),
                     "All free tests used. We invite you to book a demo for full access!",
-                    cookie,  # This cookie is not being set in the response
+                    f"User ID: {user_id}"
                 )
 
+
         # Update the outputs of the submit button click event
         submit_btn.click(
             handle_submit,
             inputs=[input_img, description_type, custom_instruction],
-            outputs=[output_text, redirect_message, submit_btn, test_counter_display, gr.Markdown(visible=True, label="User ID")],
+            outputs=[output_text, redirect_message, submit_btn, test_counter_display, user_id_display]
         )
-
-
-
-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")
-    user_id_display = gr.Markdown(visible=True, label="User ID")  # Add this line
     
 # Launch Gradio app
 demo.queue(api_open=False)