Update tab_comfy.py

tab_comfy.py

@@ -5,11 +5,6 @@ from io import BytesIO
 import urllib.parse
 from utils import save_config
 
-# Constants for Neko/Browser instance
-NEKO_IP = "192.168.1.51"
-NEKO_PORT = "8080"
-NEKO_BASE = f"http://{NEKO_IP}:{NEKO_PORT}"
-
 def render_single_instance(instance_config, index, all_instances):
     url = instance_config.get("url", "http://127.0.0.1:8188")
     name = instance_config.get("name", f"Server {index+1}")
@@ -35,7 +30,7 @@ def render_single_instance(instance_config, index, all_instances):
             save_config(
                 st.session_state.current_dir, 
                 st.session_state.config['favorites'], 
-                {"comfy_instances": all_instances}
+                st.session_state.config # Save full config including new keys
             )
             st.toast("Server config saved!", icon="💾")
             st.rerun()
@@ -47,11 +42,11 @@ def render_single_instance(instance_config, index, all_instances):
             save_config(
                 st.session_state.current_dir, 
                 st.session_state.config['favorites'], 
-                {"comfy_instances": all_instances}
+                st.session_state.config
             )
             st.rerun()
 
-    # --- 1. STATUS DASHBOARD (Still uses direct API fetch) ---
+    # --- 1. STATUS DASHBOARD ---
     with st.expander("📊 Server Status", expanded=True):
         col1, col2, col3, col4 = st.columns([1, 1, 1, 1])
         try:
@@ -71,14 +66,12 @@ def render_single_instance(instance_config, index, all_instances):
             col2.metric("Pending", "-")
             col3.metric("Running", "-")
             st.error(f"Could not connect to API at {COMFY_URL}")
-            # If API fails, we still show the browser because the user might be able to debug via Neko
     
-    # --- 2. LIVE VIEW (VIA NEKO) ---
+    # --- 2. LIVE VIEW (VIA REMOTE BROWSER) ---
     st.write("") 
     c_label, c_ctrl = st.columns([1, 2])
-    c_label.subheader("📺 Live View (Neko)")
+    c_label.subheader("📺 Live View")
     
-    # LIVE PREVIEW TOGGLE
     enable_preview = c_ctrl.checkbox("Enable Live Preview", value=True, key=f"live_toggle_{index}")
     
     if enable_preview:
@@ -89,18 +82,15 @@ def render_single_instance(instance_config, index, all_instances):
             key=f"h_slider_{index}"
         )
 
-        # Construct Neko URL
+        # Get Configured Viewer URL
-        # Neko usually takes a url parameter. 
+        viewer_base = st.session_state.config.get("viewer_url", "http://192.168.1.51:5800")
-        # Adjust query logic if your specific Neko docker uses a different format (e.g. ?url=...)
-        encoded_url = urllib.parse.quote(COMFY_URL)
-        neko_target = f"{NEKO_BASE}/?url={encoded_url}"
         
-        # If your neko setup just opens a browser session where you type the URL manually,
+        # Most Firefox containers (like jlesage/firefox) act as VNC clients and don't take ?url=
-        # just use NEKO_BASE. 
+        # So we just frame the VNC/Web interface.
-        # Assuming typical "browser-in-browser" behavior here:
+        final_src = viewer_base
-        final_src = NEKO_BASE
 
-        st.info(f"Viewing via Neko at: `{final_src}` targeting `{COMFY_URL}`")
+        st.info(f"Viewing via Remote Browser: `{final_src}`")
+        st.caption(f"🎯 Target ComfyUI: `{COMFY_URL}` (You may need to type this into the remote browser manually)")
 
         st.markdown(
             f"""
@@ -110,8 +100,6 @@ def render_single_instance(instance_config, index, all_instances):
             """,
             unsafe_allow_html=True
         )
-        
-        st.caption("💡 Note: You may need to type the ComfyUI IP into the Neko browser address bar manually if it doesn't auto-load.")
     else:
         st.info("Live Preview is disabled.")
 
@@ -152,6 +140,22 @@ def render_single_instance(instance_config, index, all_instances):
             st.error(f"Error fetching image: {e}")
 
 def render_comfy_monitor():
+    # --- GLOBAL SETTINGS FOR MONITOR ---
+    with st.expander("🔧 Monitor Settings (Remote Browser)", expanded=False):
+        current_viewer = st.session_state.config.get("viewer_url", "http://192.168.1.51:5800")
+        new_viewer = st.text_input("Remote Browser URL (Firefox/Neko)", value=current_viewer, help="e.g., http://192.168.1.51:5800 or http://localhost:8080")
+        
+        if st.button("💾 Save Monitor Settings"):
+            st.session_state.config["viewer_url"] = new_viewer
+            save_config(
+                st.session_state.current_dir, 
+                st.session_state.config['favorites'], 
+                st.session_state.config
+            )
+            st.success("Monitor settings saved!")
+            st.rerun()
+
+    # --- INSTANCE MANAGEMENT ---
     if "comfy_instances" not in st.session_state.config:
         st.session_state.config["comfy_instances"] = [
             {"name": "Main Server", "url": "http://192.168.1.100:8188"}
@@ -178,7 +182,7 @@ def render_comfy_monitor():
                     save_config(
                         st.session_state.current_dir, 
                         st.session_state.config['favorites'], 
-                        {"comfy_instances": instances}
+                        st.session_state.config
                     )
                     st.success("Server Added!")
                     st.rerun()