fixing 3

app.py

@@ -26,18 +26,10 @@ except Exception as e:
         return f"// ERROR: Failed to load core agent code. Details: {e}", "", ""
 
 # --- Gradio UI setup below ---
-with gr.Blocks(
-    title="DeepV for RTL (Model-Agnostic)",
-    # Use a single color value for the background fill
-    theme=gr.themes.Soft().set(
-        background_fill_secondary="#eee",  # A light gray that works for both themes
-        button_primary_background_fill_dark="#4CAF50",
-        button_primary_background_fill_hover_dark="#3e8e41",
-        button_primary_text_color_dark="#fff"
-    )
-) as demo:
+with gr.Blocks(title="DeepV for RTL (Model-Agnostic)", theme=gr.themes.Soft()) as demo:
     gr.Markdown("## DeepV for RTL Code Generation — Model-Agnostic (Bring Your Own API Key)")
 
+    # Status Bar for success/error messages
     status_bar = gr.Textbox(
         label="Status",
         lines=1,
@@ -78,8 +70,16 @@ with gr.Blocks(
                 top_p = gr.Slider(0.1, 1.0, value=0.9, step=0.05, label="Top-p")
                 max_new_tokens = gr.Slider(128, 4096, value=768, step=64, label="Max tokens")
 
+            # --- Loading State Components ---
             with gr.Row():
                 run_btn = gr.Button("Generate Verilog", variant="primary")
+                # New loading state components
+                loading_state = gr.Textbox(
+                    value="Generating...",
+                    show_copy_button=False,
+                    visible=False,
+                    container=False,
+                )
                 
         with gr.Column(scale=3):
             gr.Markdown("**Output**")
@@ -111,16 +111,31 @@ with gr.Blocks(
     def copy_to_clipboard_fn(text):
         return text
 
+    # Helper function to show loading state
+    def show_loading():
+        return [gr.update(visible=False), gr.update(visible=True)]
+
+    # Helper function to hide loading state
+    def hide_loading():
+        return [gr.update(visible=True), gr.update(visible=False)]
+
+    # --- Event Handlers ---
     run_btn.click(
+        fn=show_loading,
+        inputs=[],
+        outputs=[run_btn, loading_state],
+        show_progress=False # Disable default progress spinner
+    ).then(
         fn=run_generation,
         inputs=[spec, use_rag, top_k, model_choice, api_key, temperature, top_p, max_new_tokens],
         outputs=[out_code, retrieved_list, retrieved_raw],
-        show_progress=True 
     ).then(
-        fn=lambda: "Success!",
-        outputs=[status_bar]
+        fn=hide_loading,
+        inputs=[],
+        outputs=[run_btn, loading_state]
     )
-    
+
+    # Status bar updates
     clear_btn.click(fn=lambda: "Ready", outputs=[status_bar])
     spec.submit(fn=lambda: "Ready", outputs=[status_bar])
 
@@ -150,8 +165,8 @@ with gr.Blocks(
             top: 20px;
             right: 10px;
             z-index: 1000;
-            background-color: #333;
-            color: #eee;
+            background-color: #F0F0F0;
+            color: #333;
             border-radius: 5px;
             border: none;
             box-shadow: 0 2px 4px rgba(0,0,0,0.1);