fixing ui

app.py

@@ -88,31 +88,30 @@ def neuron_export(model_id: str, task: str, oauth_token: gr.OAuthToken) -> str:
     try:
         api = HfApi(token=oauth_token.token)
 
-        error, commit_info = convert(api=api, model_id=model_id, task=task, token=oauth_token.token)
+        error, result_info = convert(api=api, model_id=model_id, task=task, token=oauth_token.token)
         if error != "0":
             return error
 
-        print("[commit_info]", commit_info)
+        print("[result_info]", result_info)
 
         # Save in a private dataset if repo initialized
         if repo is not None:
             repo.git_pull(rebase=True)
             with open(os.path.join(DATADIR, DATA_FILE), "a") as csvfile:
                 writer = csv.DictWriter(
-                    csvfile, fieldnames=["model_id", "pr_url", "time"]
+                    csvfile, fieldnames=["model_id", "cache_info", "time"]
                 )
                 writer.writerow(
                     {
                         "model_id": model_id,
-                        "pr_url": commit_info.pr_url,
+                        "cache_info": str(result_info),
                         "time": str(datetime.now()),
                     }
                 )
             commit_url = repo.push_to_hub()
             print("[dataset]", commit_url)
 
-        pr_revision = commit_info.pr_revision.replace("/", "%2F")
-        return f"#### Success 🔥 Yay! This model was successfully exported and a PR was opened using your token: [{commit_info.pr_url}]({commit_info.pr_url}). If you would like to use the exported model without waiting for the PR to be approved, head to https://huggingface.co/{model_id}/tree/{pr_revision}"
+        return f"#### Success 🔥 Yay! This model was successfully compiled and cached! The compilation artifacts have been synchronized to the optimum-neuron cache repository. You can now use this model with optimum-neuron and benefit from faster loading times. Model: {model_id}, Task: {task}"
 
     except Exception as e:
         return f"#### Error: {e}"
@@ -209,12 +208,88 @@ CUSTOM_CSS = """
     display: inline-block;
     margin: 2px;
 }
+
+/* Primary button styling with warm colors */
+button[variant="primary"] {
+    background: linear-gradient(135deg, #3B82F6, #10B981) !important;
+    color: white !important;
+    padding: 16px 32px !important;
+    font-size: 1.1rem !important;
+    font-weight: 700 !important;
+    border: none !important;
+    border-radius: 12px !important;
+    box-shadow: 0 0 15px rgba(59, 130, 246, 0.5) !important;
+    transition: all 0.3s cubic-bezier(0.25, 0.8, 0.25, 1) !important;
+    position: relative;
+    overflow: hidden;
+    animation: glow 1.5s ease-in-out infinite alternate;
+}
+button[variant="primary"]::before {
+    content: "✨ ";
+}
+button[variant="primary"]:hover {
+    transform: translateY(-5px) scale(1.05) !important;
+    box-shadow: 0 10px 25px rgba(59, 130, 246, 0.7) !important;
+}
+@keyframes glow {
+    from {
+        box-shadow: 0 0 10px rgba(59, 130, 246, 0.5);
+    }
+    to {
+        box-shadow: 0 0 20px rgba(59, 130, 246, 0.8), 0 0 30px rgba(16, 185, 129, 0.5);
+    }
+}
+
+
+/* Login button styling with glow effect using warm colors */
+#login-button {
+    background: linear-gradient(135deg, #FF7A00, #FFD700) !important; /* Orange to Gold */
+    color: white !important;
+    font-weight: 700 !important;
+    border: none !important;
+    border-radius: 12px !important;
+    box-shadow: 0 0 15px rgba(255, 165, 0, 0.5) !important; /* Warm glow */
+    transition: all 0.3s cubic-bezier(0.25, 0.8, 0.25, 1) !important;
+    position: relative;
+    overflow: hidden;
+    animation: glow 1.5s ease-in-out infinite alternate;
+    max-width: 300px !important;
+    margin: 0 auto !important;
+}
+
+#login-button::before {
+    content: "🔑 ";
+    display: inline-block !important;
+    vertical-align: middle !important;
+    margin-right: 5px !important;
+    line-height: normal !important;
+}
+
+#login-button:hover {
+    transform: translateY(-3px) scale(1.03) !important;
+    box-shadow: 0 10px 25px rgba(255, 140, 0, 0.7) !important; /* Deeper warm glow */
+}
+
+#login-button::after {
+    content: "";
+    position: absolute;
+    top: 0;
+    left: -100%;
+    width: 100%;
+    height: 100%;
+    background: linear-gradient(90deg, transparent, rgba(255, 255, 255, 0.2), transparent);
+    transition: 0.5s;
+}
+
+#login-button:hover::after {
+    left: 100%;
+}
 """
 
 with gr.Blocks(css=CUSTOM_CSS) as demo:
     # Login requirement notice and button
     gr.Markdown("**You must be logged in to use this space**")
-    gr.LoginButton(min_width=250)
+    gr.LoginButton(elem_id="login-button", elem_classes="center-button", min_width=250)
     
     # Centered title and image
     gr.HTML(TITLE_IMAGE)
@@ -239,7 +314,7 @@ with gr.Blocks(css=CUSTOM_CSS) as demo:
                 )
             
             # Export button below the inputs
-            btn = gr.Button("Export to Neuron", size="lg")
+            btn = gr.Button("Export to Neuron", size="lg", variant="primary")
             
             # Output section
             output = gr.Markdown(label="Output")