Final

Dockerfile

@@ -24,7 +24,7 @@ RUN apt-get update && apt-get install -y \
 COPY requirements.txt ./
 COPY src/ ./src/
 COPY entrypoint.sh /usr/local/bin/entrypoint.sh
-COPY frontend/app/dist/ /app/src/static/
+COPY frontend/app/dist/ ./frontend/app/dist/
 
 RUN python3 -m venv /opt/venv
 ENV PATH="/opt/venv/bin:$PATH"

src/app.py

@@ -14,10 +14,7 @@ from flask_socketio import SocketIO, emit
 
 event_queue = queue.Queue()
 
-if "SPACE_ID" in os.environ:
-    app = Flask(__name__, static_folder='static', static_url_path='/')
-else:
-    app = Flask(__name__, static_folder='../frontend/app/dist', static_url_path='/')
+app = Flask(__name__, static_folder='../frontend/app/dist', static_url_path='/')
 
 socketio = SocketIO(app, cors_allowed_origins="*")
 

src/codeagent/src/codeagent/main.py

@@ -15,20 +15,19 @@ from app import socketio
 # from codeagent.src.codeagent.crews.git_crew.git_crew import GitCrew
 from codeagent.src.codeagent.crews.SqlCrew.sqlcrew import SqlCrew
 
-from io import TextIOWrapper
+import io
 
-class SocketWriter(TextIOWrapper):
+class SocketWriter(io.TextIOBase):
     def __init__(self, emit_func):
         self.emit = emit_func
-        super().__init__(None, write_through=True)
 
-    def write(self, text):
-        if text.strip():
-            self.emit('log', {'msg': text.strip()})
-        super().write(text)
+    def write(self, s):
+        if s.strip():
+            self.emit("log", {'msg': s.strip()})
+        return len(s)
 
-old_stdout = sys.stdout
-sys.stdout = SocketWriter(lambda event, data: socketio.emit(event, data))
+    def flush(self):
+        pass
 
 class FlowState(BaseModel):
     user_input: str = ""
@@ -77,9 +76,13 @@ class CodeAgentFlow(Flow[FlowState]):
             # self.state.result = result.raw
             pass
         elif self.state.decision == "SqlCrew":
-            result = SqlCrew().crew().kickoff(inputs={"task": self.state.user_input})
-            self.state.result = result.raw
-            sys.stdout = old_stdout
+            old_stdout = sys.stdout
+            sys.stdout = SocketWriter(lambda event, data: socketio.emit(event, data))
+            try:
+                result = SqlCrew().crew().kickoff(inputs={"task": self.state.user_input})
+                self.state.result = result.raw
+            finally:
+                sys.stdout = old_stdout
         else:
             self.state.result = f"Invalid decision from router: {self.state.decision}"