removed ui

rl_code_fix_env/ui/app.py

@@ -1,200 +0,0 @@
-# import streamlit as st
-import sys
-import os
-import streamlit as st
-
-sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
-
-from rl_code_fix_env.client import CodeFixerEnv
-from rl_code_fix_env.models import CodeFixerAction
-
-# Set page config
-st.set_page_config(page_title="Code Fixer Environment", layout="wide")
-
-st.title("Code Fixer Agent Environment")
-
-ENV_URL = os.getenv("ENV_URL", "http://127.0.0.1:8000")
-
-_WS_ERROR_MARKERS = ("1011", "1006", "keepalive", "timed out", "closed", "connection")
-
-
-def _is_ws_error(exc: Exception) -> bool:
-    """Return True if the exception looks like a stale WebSocket."""
-    return any(m in str(exc).lower() for m in _WS_ERROR_MARKERS)
-
-
-def _get_client() -> CodeFixerEnv:
-    """Return existing client or create a fresh one."""
-    if "client" not in st.session_state or st.session_state.client is None:
-        st.session_state.client = CodeFixerEnv(base_url=ENV_URL)
-    return st.session_state.client
-
-
-def _reset_client() -> CodeFixerEnv:
-    """Drop the stale client and create a brand-new one."""
-    try:
-        if "client" in st.session_state and st.session_state.client is not None:
-            st.session_state.client.close()
-    except Exception:
-        pass
-    st.session_state.client = CodeFixerEnv(base_url=ENV_URL)
-    return st.session_state.client
-
-
-# Initialize session state
-if "observation" not in st.session_state:
-    st.session_state.observation = None
-if "difficulty_index" not in st.session_state:
-    st.session_state.difficulty_index = 0  # cycles: easy  medium  hard
-
-_DIFFICULTY_CYCLE = ["easy", "medium", "hard"]
-
-# Ensure client exists on first load
-try:
-    _get_client()
-except Exception as e:
-    st.error(f"Failed to connect to environment: {e}")
-    st.info(f"Make sure the server is running on {ENV_URL}")
-
-
-# Sidebar for controls
-with st.sidebar:
-    st.header("Controls")
-    col1, col2 = st.columns(2)
-    
-    with col1:
-        if st.button("Load Task", type="primary"):
-            idx = st.session_state.difficulty_index
-            difficulty = _DIFFICULTY_CYCLE[idx % len(_DIFFICULTY_CYCLE)]
-            try:
-                result = _get_client().reset()
-                st.session_state.observation = result.observation
-                st.session_state.difficulty_index = idx + 1
-                st.success(f"Loaded **{difficulty}** task!")
-                st.rerun()
-            except Exception as e:
-                if _is_ws_error(e):
-                    try:
-                        result = _reset_client().reset()
-                        st.session_state.observation = result.observation
-                        st.session_state.difficulty_index = idx + 1
-                        st.success(f"Reconnected & loaded **{difficulty}** task!")
-                        st.rerun()
-                    except Exception as e2:
-                        st.error(f"Reconnect failed: {e2}")
-                else:
-                    st.error(f"Error loading task: {e}")
-    
-    with col2:
-        if st.button("Reset Env"):
-            try:
-                result = _get_client().reset()
-                st.session_state.observation = result.observation
-                st.success("Environment reset!")
-                st.rerun()
-            except Exception as e:
-                if _is_ws_error(e):
-                    try:
-                        result = _reset_client().reset()
-                        st.session_state.observation = result.observation
-                        st.success("Reconnected & environment reset!")
-                        st.rerun()
-                    except Exception as e2:
-                        st.error(f"Reconnect failed: {e2}")
-                else:
-                    st.error(f"Error resetting environment: {e}")
-
-
-    st.divider()
-
-    if st.session_state.observation:
-        st.subheader("Environment Status")
-        obs = st.session_state.observation
-        st.write(f"**Steps:** {obs.steps}")
-        st.write(f"**Test Score:** {obs.test_score:.2f} / 1.00")
-        st.write(f"**Done:** {obs.done}")
-        st.write(f"**Reward:** {obs.reward}")
-    else:
-        st.info("Click 'Load Task' to start a new problem.")
-
-# Main content area
-if st.session_state.observation:
-    obs = st.session_state.observation
-    
-    col1, col2 = st.columns([1, 1])
-    
-    with col1:
-        st.subheader("Current Code")
-        st.code(obs.code, language="python")
-        
-    with col2:
-        st.subheader("Actions")
-        
-        # Action: Apply Patch
-        st.markdown("### Apply Patch")
-        patch_text = st.text_area("Enter patch here:", height=200)
-        if st.button("Apply Patch"):
-            action = CodeFixerAction(type="apply_patch", payload=patch_text)
-            try:
-                result = _get_client().step(action)
-                st.session_state.observation = result.observation
-                st.rerun()
-            except Exception as e:
-                if _is_ws_error(e):
-                    try:
-                        result = _reset_client().step(action)
-                        st.session_state.observation = result.observation
-                        st.rerun()
-                    except Exception as e2:
-                        st.error(f"Reconnect failed: {e2}")
-                else:
-                    st.error(f"Error applying patch: {e}")
-                
-        st.divider()
-        
-        # Action: Run Tests
-        st.markdown("### Run Tests")
-        if st.button("Execute Tests"):
-            action = CodeFixerAction(type="run_tests")
-            try:
-                result = _get_client().step(action)
-                st.session_state.observation = result.observation
-                st.rerun()
-            except Exception as e:
-                if _is_ws_error(e):
-                    try:
-                        result = _reset_client().step(action)
-                        st.session_state.observation = result.observation
-                        st.rerun()
-                    except Exception as e2:
-                        st.error(f"Reconnect failed: {e2}")
-                else:
-                    st.error(f"Error running tests: {e}")
-                
-        st.divider()
-        
-        # Action: Get Logs
-        st.markdown("### Get Logs")
-        if st.button("Fetch Logs"):
-            action = CodeFixerAction(type="get_logs")
-            try:
-                result = _get_client().step(action)
-                st.session_state.observation = result.observation
-                st.rerun()
-            except Exception as e:
-                if _is_ws_error(e):
-                    try:
-                        result = _reset_client().step(action)
-                        st.session_state.observation = result.observation
-                        st.rerun()
-                    except Exception as e2:
-                        st.error(f"Reconnect failed: {e2}")
-                else:
-                    st.error(f"Error getting logs: {e}")
-                
-        # Display Logs
-        if obs.logs:
-            st.subheader("Logs")
-            st.text_area("Output Logs", obs.logs, height=200, disabled=True)
-            
-