fix: Correct role assignment and add robustness to websocket connection

src.py

@@ -0,0 +1 @@
+ 

src/app.py

@@ -228,7 +228,7 @@ async def start_debate(websocket: WebSocket, data: dict):
                 await websocket.send_text(json.dumps({ "type": "round_info", "message": f"--- 第{round_num}轮 ---" }))
 
             speaker_name, speaker_model = speakers[i % 2]
-            role = "正方" if speaker_name == pro_model_name else "反方"
+            role = "正方" if (i % 2 == 0) else "反方"
             
             await websocket.send_text(json.dumps({ "type": "model_speaking", "model": speaker_name, "role": role }))
 

static/js/script.js

@@ -94,7 +94,7 @@ function connect() {
     websocket = new WebSocket(wsUrl);
     websocket.onopen = () => { isConnected = true; startBtn.disabled = false; };
     websocket.onmessage = (event) => handleWebSocketMessage(JSON.parse(event.data));
-    websocket.onclose = () => { isConnected = false; startBtn.disabled = true; stopBtn.disabled = true; };
+    websocket.onclose = () => { isConnected = false; startBtn.disabled = true; stopBtn.disabled = true; judgeBtn.disabled = true; };
     websocket.onerror = (error) => { console.error('WebSocket Error:', error); };
 }