Update app.py

app.py

@@ -54,7 +54,7 @@ def increment_user_counter(user_id):
 # Function to get or create user ID from cookie
 def get_or_create_user_id(request: gr.Request):
     if request is None:
-        return str(uuid.uuid4())
+        return str(uuid.uuid4()), None
     
     cookies = SimpleCookie()
     cookies.load(request.headers.get('cookie', ''))
@@ -62,13 +62,9 @@ def get_or_create_user_id(request: gr.Request):
     
     if user_id is None:
         user_id = str(uuid.uuid4())
-        cookies['user_id'] = user_id
-        cookies['user_id']['httponly'] = True
-        cookies['user_id']['max-age'] = 365 * 24 * 60 * 60  # 1 year
+        return user_id, ('user_id', user_id, {'httponly': True, 'max_age': 365 * 24 * 60 * 60})
     else:
-        user_id = user_id.value
-    
-    return user_id, cookies.output(header='', sep=';')
+        return user_id.value, None
 
 # Function to encode the image
 def encode_image(image_array):
@@ -259,25 +255,25 @@ 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,
+                        cookie,  # This cookie is not being set in the response
                     )
                 else:
                     redirect_text = f"""
                     Thank you for trying our product description generation tool!
-
+        
                     You've used all 4 of your free tests for today. We hope you found them helpful and insightful.
-
+        
                     To unlock unlimited access and discover how our AI-powered solution can revolutionize your content creation process:
-
+        
                     [Book a Personalized Demo with Our Experts]({demo_link})
-
+        
                     During the demo, you'll get:
                     • A tailored walkthrough of our full suite of features
                     • Insights on how to optimize your specific content workflow
                     • Answers to any questions you may have about our solution
-
+        
                     We're excited to show you how we can help scale your content creation!
-
+        
                     (Your free tests will reset tomorrow, allowing you to try 4 more if you need additional time to decide.)
                     """
                     return (
@@ -285,25 +281,25 @@ 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,
+                        cookie,  # This cookie is not being set in the response
                     )
             else:
                 redirect_text = f"""
                 Thank you for your interest in our product description generation tool!
-
+        
                 You've already used your 4 free tests for today. We hope they've given you a glimpse of how our AI can enhance your content creation.
-
+        
                 Ready to explore the full potential of our solution?
-
+        
                 [Book a Personalized Demo with Our Experts]({demo_link})
-
+        
                 In this demo, you'll discover:
                 • How our AI can be customized for your specific needs
                 • Advanced features for scaling your content production
                 • ROI projections based on your current content workflow
-
+        
                 Let's discuss how we can help you achieve your content goals!
-
+        
                 (Remember, your free tests will reset tomorrow if you need more time to evaluate.)
                 """
                 return (
@@ -311,15 +307,21 @@ 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,
+                    cookie,  # This cookie is not being set in the response
                 )
 
         submit_btn.click(
             handle_submit,
             inputs=[input_img, description_type, custom_instruction],
-            outputs=[output_text, redirect_message, submit_btn, test_counter_display],
+            outputs=[output_text, redirect_message, submit_btn, test_counter_display, gr.Markdown(visible=True, label="User ID")],
         )
 
+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)
 demo.launch(debug=True)