Update app.py

app.py

@@ -49,22 +49,10 @@ def virtual_tryon(human_img, garment_img, request: gr.Request):
         print(f"Error in virtual_tryon: {str(e)}")
         raise e
 
-def get_remaining_attempts(request: gr.Request):
-    user_context = get_user_context(request)
-    current_attempts = data_manager.get_attempts(user_context)
-    remaining = MAX_FREE_TRIALS - current_attempts
-    
-    print(f"Getting remaining attempts for {user_context}: {remaining}")
-    
-    if remaining <= 0:
-        return f"Daily limit reached ({MAX_FREE_TRIALS}/{MAX_FREE_TRIALS}). Sign up for unlimited access!"
-    else:
-        return f"Remaining free generations: {remaining}/{MAX_FREE_TRIALS}"
-
 def create_footer():
     return f"""
-    <div style="text-align: center; margin-top: 30px; padding: 15px; background-color: #f5f6fa; border-radius: 8px;">
-        <p style="margin: 5px 0; color: #636e72;">© 2025 {COMPANY_NAME}. All rights reserved.</p>
+    <div style="text-align: center; margin-top: 30px; padding: 15px; border-radius: 8px;">
+        <p style="margin: 5px 0;">© 2025 {COMPANY_NAME}. All rights reserved.</p>
         <p style="margin: 5px 0;">
             <a href="{COMPANY_URL}" target="_blank" style="color: #0984e3; text-decoration: none;">Website</a> | 
             <a href="mailto:{CONTACT_EMAIL}" style="color: #0984e3; text-decoration: none;">Contact Us</a> | 
@@ -75,14 +63,14 @@ def create_footer():
 
 def create_instructions():
     return """
-    <div style="background-color: #f5f6fa; padding: 15px; border-radius: 8px; margin-bottom: 20px;">
-        <h3 style="margin-top: 0; color: #2d3436;">How to use:</h3>
+    <div style="padding: 15px; border-radius: 8px; margin-bottom: 20px;">
+        <h2 style="margin-top: 0;">How to use:</h2>
         <ol style="color: #636e72;">
             <li>Upload a clear photo of a person (front view works best)</li>
             <li>Upload an image of the garment you want to try on</li>
             <li>Click "Run Virtual Try-On" to see the result</li>
         </ol>
-        <p style="color: #636e72; margin-bottom: 0;"><strong>Tip:</strong> For best results, use images with plain backgrounds.</p>
+        <p style="margin-bottom: 0;"><strong>Tip:</strong> For best results, use images with plain backgrounds.</p>
     </div>
     """
 
@@ -161,19 +149,6 @@ footer {visibility: hidden}
     font-weight: bold;
     color: #1976d2;
 }
-:root {
-    --primary-color: #6c5ce7;
-    --secondary-color: #a29bfe;
-    --accent-color: #fd79a8;
-    --text-color: #2d3436;
-    --light-text: #636e72;
-    --bg-color: #f5f6fa;
-    --card-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
-}
-
-body {
-    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
-}
 
 #col-container {
     max-width: 1400px;
@@ -181,7 +156,6 @@ body {
 }
 
 .step-column {
-    background-color: white;
     padding: 20px;
     border-radius: 8px;
     box-shadow: var(--card-shadow);
@@ -190,7 +164,6 @@ body {
 
 .step-title {
     color: var(--primary-color);
-    font-weight: 600;
     text-align: center;
     margin-bottom: 15px;
 }
@@ -241,9 +214,6 @@ with gr.Blocks(css=css, title=f"{COMPANY_NAME} Virtual Try-On", theme=gr.themes.
     # Instructions
     gr.HTML(create_instructions())
     
-    # Usage counter
-    usage_display = gr.HTML(elem_classes="attempts-counter")
-    
     with gr.Row(elem_id="col-container"):
         # Step 1: Person Image
         with gr.Column(elem_classes="step-column"):
@@ -332,21 +302,12 @@ with gr.Blocks(css=css, title=f"{COMPANY_NAME} Virtual Try-On", theme=gr.themes.
                 examples_per_page=3
             )
     
-    # Update usage display on page load
-    demo.load(
-        fn=get_remaining_attempts,
-        outputs=usage_display
-    )
-    
     # Button action
     try_button.click(
         fn=virtual_tryon,
         inputs=[human_img, garment_img],
         outputs=result_img,
         api_name="virtual_tryon"
-    ).then(
-        fn=get_remaining_attempts,
-        outputs=usage_display
     )
 
     gr.HTML('<a href="https://visitorbadge.io/status?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FMiragic-AI%2FMiragic-Virtual-Try-On"><img src="https://api.visitorbadge.io/api/combined?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FMiragic-AI%2FMiragic-Virtual-Try-On&label=VISITORS&labelColor=%2337d67a&countColor=%23f47373&style=plastic&labelStyle=upper" /></a>')