Update app.py

app.py

@@ -120,43 +120,10 @@ custom_theme = gr.themes.Soft(
     text_size=gr.themes.sizes.text_lg,
     font=[gr.themes.GoogleFont("Inter"), "ui-sans-serif", "system-ui", "sans-serif"],
     font_mono=[gr.themes.GoogleFont("IBM Plex Mono"), "ui-monospace", "Consolas", "monospace"],
-).set(
-    # Button styling
-    button_primary_background_fill="#0071e3",
-    button_primary_background_fill_hover="#0077ed",
-    button_primary_text_color="white",
-    button_large_padding="16px 40px",
-    button_large_radius="12px",
-    button_shadow="0 2px 8px rgba(0, 113, 227, 0.2)",
-    button_shadow_hover="0 4px 12px rgba(0, 113, 227, 0.3)",
-    
-    # Input styling
-    input_background_fill="#fbfbfd",
-    input_background_fill_focus="white",
-    input_border_color="#d2d2d7",
-    input_border_color_focus="#0071e3",
-    input_radius="12px",
-    input_padding="16px",
-    input_shadow_focus="0 0 0 4px rgba(0, 113, 227, 0.1)",
-    
-    # Container styling
-    block_background_fill="white",
-    block_border_width="1px",
-    block_border_color="#e5e5e7",
-    block_radius="12px",
-    block_padding="24px",
-    
-    # Body styling
-    body_background_fill="#f5f5f7",
-    body_text_color="#1d1d1f",
-    
-    # Link styling
-    link_text_color="#0071e3",
-    link_text_color_hover="#0077ed",
 )
 
-# Minimal additional CSS for layout refinements
-minimal_css = """
+# Custom CSS for additional styling
+custom_css = """
 .gradio-container {
     max-width: 900px !important;
     margin: 0 auto !important;
@@ -208,6 +175,16 @@ minimal_css = """
     margin: 0.25rem 0;
 }
 
+button.primary {
+    background-color: #0071e3 !important;
+    transition: all 0.2s ease !important;
+}
+
+button.primary:hover {
+    background-color: #0077ed !important;
+    box-shadow: 0 4px 12px rgba(0, 113, 227, 0.3) !important;
+}
+
 @media (max-width: 768px) {
     .main-header h1 {
         font-size: 2rem !important;
@@ -222,8 +199,6 @@ minimal_css = """
 # Create the Gradio interface
 with gr.Blocks(
     title="Z-Image Turbo",
-    theme=custom_theme,
-    css=minimal_css,
     fill_height=False
 ) as demo:
     
@@ -265,8 +240,7 @@ with gr.Blocks(
         type="pil",
         label="Generated Image",
         show_label=False,
-        show_download_button=True,
-        show_share_button=False,
+        buttons=["download"],
         container=True
     )
     
@@ -297,5 +271,7 @@ with gr.Blocks(
 if __name__ == "__main__":
     demo.launch(
         share=False,
-        show_error=True
+        show_error=True,
+        theme=custom_theme,
+        css=custom_css
     )