Update app.py

app.py

@@ -211,14 +211,28 @@ def logout() -> Tuple:
     )
 
 # Create Gradio interface
-with gr.Blocks(theme=gr.themes.Soft(), title="Dataset Annotation Tool") as app:
+white_background_css = """
+    #login-group {
+        background-color: white !important;
+        padding: 20px;
+        border-radius: 8px;
+    },
+    #annotation-group {
+        background-color: white !important;
+        padding: 20px;
+        border-radius: 8px;
+    }
+"""
+
+# Create Gradio interface
+with gr.Blocks(theme=gr.themes.Soft(), title="Dataset Annotation Tool", css=custom_css) as app:
     gr.Markdown("# Dataset Annotation Tool")
     gr.Markdown("Annotate which response is more fluent for each prompt.")
     
     user_state = gr.State("")
     
     # Login Interface
-    with gr.Group(visible=True) as login_interface:
+    with gr.Group(visible=True, elem_id="login-group") as login_interface:
         gr.Markdown("## Login")
         gr.Markdown("Enter your annotator ID to begin. Your ID determines the order of samples you'll see.")
         
@@ -233,7 +247,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Dataset Annotation Tool") as app:
         login_status = gr.Markdown("")
     
     # Annotation Interface
-    with gr.Group(visible=False) as annotation_interface:
+    with gr.Group(visible=False, elem_id="annotation-group") as annotation_interface:
         progress_label = gr.Markdown("")
         
         # Row 1: Prompt