Update app.py

app.py

@@ -331,6 +331,14 @@ def create_simple_interface():
             )
             load_btn = gr.Button("🔁 Load Selected Model", variant="secondary")
 
+        # Model status indicator
+        model_status = gr.Textbox(
+            label="Model Status",
+            value=f"✅ Current Model: {default_model}",
+            interactive=False,
+            show_copy_button=True
+        )
+
         # Generation controls
         with gr.Row():
             with gr.Column():
@@ -369,22 +377,18 @@ def create_simple_interface():
             interactive=False
         )
 
-        # Load model on button click
-        load_status = gr.Textbox(label="Model Status", interactive=False, visible=False)
-
         def load_model_wrapper(model_name):
             success = app.load_model_by_name(model_name)
-            status = "✅ Model loaded successfully!" if success else "❌ Failed to load model"
+            if success:
+                status = f"✅ Current Model: {model_name} (Ready to use!)"
+            else:
+                status = f"❌ Failed to load: {model_name}"
             return status
 
         load_btn.click(
             fn=load_model_wrapper,
             inputs=model_choice,
-            outputs=load_status
-        ).then(
-            fn=lambda: gr.update(visible=True),
-            inputs=None,
-            outputs=load_status
+            outputs=model_status
         )
 
         # Generate molecule