fez5587/commander-ai-lab
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

CoachController: add streaming, precon refresh tab, mobile layout (U9 #54)

Browse Source
This commit is contained in:
KoalaTrapLord authored and GitHub committed 2026-03-18 22:39:47 -04:00
1 parent d21928da53
commit 7901fef495
1 file changed
+297 -31
Download Patch File Download Diff File Expand all files Collapse all files
unity-client/Assets/Scripts/UI/CoachController.cs
+297 -31
View File
@@ -1,11 +1,14 @@
using System;
using System.Collections;
using System.Collections.Generic;
using System.Text;
using UnityEngine;
using UnityEngine.UI;
using UnityEngine.SceneManagement;
using UnityEngine.Networking;
using TMPro;
using Newtonsoft.Json;
using Newtonsoft.Json.Linq;
using CommanderAILab.Models;
using CommanderAILab.Services;
@@ -15,7 +18,9 @@ namespace CommanderAILab.UI
/// Coach scene — AI chat + analytics charts.
/// Chat bubble UI, deck selector, "Ask Coach" button,
/// deck analytics panel (mana curve, color distribution, card types),
/// "Export to Simulator" button, precon refresh tab.
/// streaming token-by-token response display,
/// "Export to Simulator" button, precon refresh tab,
/// mobile-responsive layout (fullscreen chat / swipe analytics).
/// </summary>
public class CoachController : MonoBehaviour
{
@@ -50,14 +55,32 @@ namespace CommanderAILab.UI
[Header("Actions")]
[SerializeField] private Button exportToSimButton;
[Header("Precon Refresh")]
[SerializeField] private GameObject preconPanel;
[SerializeField] private TMP_Text preconCompareText;
[SerializeField] private Button preconRefreshButton;
[SerializeField] private TMP_Dropdown preconDropdown;
[Header("Mobile Layout")]
[SerializeField] private GameObject chatFullscreenPanel;
[SerializeField] private GameObject analyticsSwipePanel;
[SerializeField] private Button toggleViewButton;
[SerializeField] private TMP_Text toggleViewLabel;
[Header("Error")]
[SerializeField] private GameObject errorPanel;
[SerializeField] private TMP_Text errorText;
private readonly List<DeckSummary> _decks = new();
private readonly List<ChatMessage> _chatHistory = new();
private readonly List<PreconEntry> _precons = new();
private string _selectedDeckId;
private bool _isWaiting = false;
private bool _isMobile = false;
private bool _showingChat = true;
private GameObject _streamingBubble;
private TMP_Text _streamingText;
private StringBuilder _streamingBuffer = new();
private void Start()
{
@@ -67,16 +90,52 @@ namespace CommanderAILab.UI
refreshDecksButton.onClick.AddListener(LoadDecks);
exportToSimButton.onClick.AddListener(OnExportToSim);
deckDropdown.onValueChanged.AddListener(OnDeckSelected);
preconRefreshButton.onClick.AddListener(OnPreconRefresh);
preconDropdown.onValueChanged.AddListener(OnPreconSelected);
if (toggleViewButton != null)
toggleViewButton.onClick.AddListener(ToggleMobileView);
chatInput.onSubmit.AddListener(_ => OnSend());
errorPanel.SetActive(false);
typingIndicator.gameObject.SetActive(false);
if (preconPanel != null) preconPanel.SetActive(false);
DetectMobileLayout();
LoadDecks();
LoadPrecons();
AddCoachBubble("Hello! Select a deck and ask me anything about your strategy.");
}
// ── Deck Loading ───────────────────────────────────────────────
// ── Mobile Layout ────────────────────────────────────────────
private void DetectMobileLayout()
{
_isMobile = (Screen.width < 800 || Application.isMobilePlatform);
if (_isMobile)
{
if (chatFullscreenPanel != null) chatFullscreenPanel.SetActive(true);
if (analyticsSwipePanel != null) analyticsSwipePanel.SetActive(false);
if (toggleViewButton != null) toggleViewButton.gameObject.SetActive(true);
if (toggleViewLabel != null) toggleViewLabel.text = "Analytics";
_showingChat = true;
}
else
{
if (chatFullscreenPanel != null) chatFullscreenPanel.SetActive(true);
if (analyticsSwipePanel != null) analyticsSwipePanel.SetActive(true);
if (toggleViewButton != null) toggleViewButton.gameObject.SetActive(false);
}
}
private void ToggleMobileView()
{
_showingChat = !_showingChat;
if (chatFullscreenPanel != null) chatFullscreenPanel.SetActive(_showingChat);
if (analyticsSwipePanel != null) analyticsSwipePanel.SetActive(!_showingChat);
if (toggleViewLabel != null)
toggleViewLabel.text = _showingChat ? "Analytics" : "Chat";
}
// ── Deck Loading ────────────────────────────────────────────
private void LoadDecks()
{
ApiClient.Instance.GetRaw("/api/decks",
@@ -106,13 +165,14 @@ namespace CommanderAILab.UI
{
_selectedDeckId = null;
analyticsPanel.SetActive(false);
if (preconPanel != null) preconPanel.SetActive(false);
return;
}
_selectedDeckId = _decks[index - 1].id;
LoadDeckAnalytics(_selectedDeckId);
}
// ── Analytics ─────────────────────────────────────────────────
// ── Analytics ───────────────────────────────────────────────
private void LoadDeckAnalytics(string deckId)
{
ApiClient.Instance.GetRaw($"/api/decks/{deckId}",
@@ -174,13 +234,11 @@ namespace CommanderAILab.UI
int barCount = 8;
float barW = w / barCount * 0.7f;
float gap = w / barCount;
for (int i = 0; i <= 7; i++)
{
string key = i == 7 ? "7+" : i.ToString();
curve.TryGetValue(key, out int count);
float barH = (float)count / max * h * 0.85f;
var bar = new GameObject($"Bar_{key}");
bar.transform.SetParent(manaCurveChartRect, false);
var img = bar.AddComponent<Image>();
@@ -192,14 +250,14 @@ namespace CommanderAILab.UI
}
}
// ── Chat ────────────────────────────────────────────────────────
// ── Chat (Streaming) ─────────────────────────────────────────
private void OnSend()
{
string msg = chatInput.text.Trim();
if (string.IsNullOrEmpty(msg) || _isWaiting) return;
chatInput.text = "";
AddUserBubble(msg);
SendToCoach(msg);
SendToCoachStreaming(msg);
}
private void OnAskCoach()
@@ -209,23 +267,131 @@ namespace CommanderAILab.UI
AddCoachBubble("Please select a deck first.");
return;
}
string msg = $"Analyze my deck and suggest improvements.";
string msg = "Analyze my deck and suggest improvements.";
AddUserBubble(msg);
SendToCoach(msg);
SendToCoachStreaming(msg);
}
private void SendToCoach(string userMessage)
private void SendToCoachStreaming(string userMessage)
{
_isWaiting = true;
sendButton.interactable = false;
typingIndicator.gameObject.SetActive(true);
typingIndicator.text = "Coach is thinking...";
var payload = new ChatRequest
_chatHistory.Add(new ChatMessage { role = "user", content = userMessage });
var messages = new List<object>();
foreach (var m in _chatHistory)
messages.Add(new { role = m.role, content = m.content });
var payload = new {
deck_id = _selectedDeckId ?? "",
messages = messages,
stream = true
};
string body = JsonConvert.SerializeObject(payload);
// Create streaming bubble
_streamingBuffer.Clear();
_streamingBubble = Instantiate(coachBubblePrefab, chatContent);
_streamingText = _streamingBubble.GetComponentInChildren<TMP_Text>();
if (_streamingText != null) _streamingText.text = "";
ScrollToBottom();
StartCoroutine(StreamChatResponse(body));
}
private IEnumerator StreamChatResponse(string jsonBody)
{
string url = ApiClient.Instance.BaseUrl + "/api/coach/chat";
byte[] bodyBytes = Encoding.UTF8.GetBytes(jsonBody);
using var request = new UnityWebRequest(url, "POST");
request.uploadHandler = new UploadHandlerRaw(bodyBytes);
request.downloadHandler = new DownloadHandlerBuffer();
request.SetRequestHeader("Content-Type", "application/json");
request.SetRequestHeader("Accept", "text/event-stream");
var op = request.SendWebRequest();
int lastProcessed = 0;
while (!op.isDone)
{
message = userMessage,
deck_id = _selectedDeckId,
history = _chatHistory
string partial = request.downloadHandler?.text ?? "";
if (partial.Length > lastProcessed)
{
ProcessSSEChunk(partial.Substring(lastProcessed));
lastProcessed = partial.Length;
}
yield return null;
}
// Process remaining data
string final_ = request.downloadHandler?.text ?? "";
if (final_.Length > lastProcessed)
ProcessSSEChunk(final_.Substring(lastProcessed));
if (request.result != UnityWebRequest.Result.Success)
{
if (_streamingBuffer.Length == 0)
{
// Fallback: try non-streaming
Destroy(_streamingBubble);
SendToCoachFallback(_chatHistory[_chatHistory.Count - 1].content);
yield break;
}
}
// Finalize streaming
string fullReply = _streamingBuffer.ToString();
if (!string.IsNullOrEmpty(fullReply))
_chatHistory.Add(new ChatMessage { role = "assistant", content = fullReply });
_isWaiting = false;
sendButton.interactable = true;
typingIndicator.gameObject.SetActive(false);
_streamingBubble = null;
_streamingText = null;
}
private void ProcessSSEChunk(string chunk)
{
var lines = chunk.Split('\n');
foreach (var line in lines)
{
var trimmed = line.Trim();
if (!trimmed.StartsWith("data: ")) continue;
var data = trimmed.Substring(6);
if (data == "[DONE]") continue;
try
{
var obj = JObject.Parse(data);
string content = obj["content"]?.ToString() ?? "";
if (!string.IsNullOrEmpty(content))
{
_streamingBuffer.Append(content);
if (_streamingText != null)
_streamingText.text = _streamingBuffer.ToString();
typingIndicator.gameObject.SetActive(false);
ScrollToBottom();
}
}
catch (JsonException) { }
}
}
// ── Non-streaming fallback ───────────────────────────────────
private void SendToCoachFallback(string userMessage)
{
var messages = new List<object>();
foreach (var m in _chatHistory)
messages.Add(new { role = m.role, content = m.content });
var payload = new {
deck_id = _selectedDeckId ?? "",
messages = messages,
stream = false
};
string body = JsonConvert.SerializeObject(payload);
@@ -237,10 +403,10 @@ namespace CommanderAILab.UI
typingIndicator.gameObject.SetActive(false);
try
{
var resp = JsonConvert.DeserializeObject<ChatResponse>(json);
_chatHistory.Add(new ChatMessage { role = "user", content = userMessage });
_chatHistory.Add(new ChatMessage { role = "assistant", content = resp.reply });
AddCoachBubble(resp.reply);
var obj = JObject.Parse(json);
string reply = obj["content"]?.ToString() ?? "";
_chatHistory.Add(new ChatMessage { role = "assistant", content = reply });
AddCoachBubble(reply);
}
catch (Exception e) { ShowError($"Chat parse error: {e.Message}"); }
},
@@ -275,6 +441,107 @@ namespace CommanderAILab.UI
chatScrollRect.verticalNormalizedPosition = 0f;
}
// ── Precon Refresh ───────────────────────────────────────────
private void LoadPrecons()
{
ApiClient.Instance.GetRaw("/api/lab/precons",
json =>
{
try
{
var result = JObject.Parse(json);
var preconArray = result["precons"] as JArray;
_precons.Clear();
preconDropdown.ClearOptions();
var opts = new List<string> { "-- Compare to Precon --" };
if (preconArray != null)
{
foreach (var p in preconArray)
{
var entry = new PreconEntry
{
fileName = p["fileName"]?.ToString() ?? "",
name = p["name"]?.ToString() ?? p["fileName"]?.ToString() ?? "",
commander = p["commander"]?.ToString() ?? ""
};
_precons.Add(entry);
opts.Add(entry.name);
}
}
preconDropdown.AddOptions(opts);
}
catch (Exception e) { Debug.LogWarning($"Precon load: {e.Message}"); }
},
err => Debug.LogWarning($"Precon load failed: {err}"));
}
private void OnPreconSelected(int index)
{
if (index <= 0 || index > _precons.Count || _selectedDeckId == null)
{
if (preconPanel != null) preconPanel.SetActive(false);
return;
}
var precon = _precons[index - 1];
CompareToPrecon(precon);
}
private void CompareToPrecon(PreconEntry precon)
{
if (preconPanel != null) preconPanel.SetActive(true);
if (preconCompareText != null)
preconCompareText.text = $"Comparing to: {precon.name}\nCommander: {precon.commander}\nLoading comparison...";
// Load deck report for comparison
ApiClient.Instance.GetRaw($"/api/coach/decks/{_selectedDeckId}/report",
json =>
{
try
{
var report = JObject.Parse(json);
var structure = report["structure"];
int landCount = structure?["landCount"]?.Value<int>() ?? 0;
var buckets = structure?["curveBuckets"];
string curveStr = buckets != null ? string.Join(", ", buckets) : "N/A";
if (preconCompareText != null)
preconCompareText.text =
$"Precon Baseline: {precon.name}\n" +
$"Commander: {precon.commander}\n" +
$"\nYour Deck Analysis:\n" +
$" Lands: {landCount}\n" +
$" Curve: [{curveStr}]\n" +
$"\nSelect a precon above to compare structure.";
}
catch
{
if (preconCompareText != null)
preconCompareText.text = $"Precon: {precon.name}\nDeck report not available. Run a coaching session first.";
}
},
err =>
{
if (preconCompareText != null)
preconCompareText.text = $"Precon: {precon.name}\nCould not load deck report: {err}";
});
}
private void OnPreconRefresh()
{
if (preconCompareText != null)
preconCompareText.text = "Refreshing precon database...";
string body = JsonConvert.SerializeObject(new { });
StartCoroutine(ApiClient.Instance.PostRaw("/api/lab/precons/refresh", body,
json =>
{
if (preconCompareText != null)
preconCompareText.text = "Precon database refreshed!";
LoadPrecons();
},
err => ShowError($"Precon refresh failed: {err}")));
}
// ── Export to Simulator ───────────────────────────────────────
private void OnExportToSim()
{
@@ -300,15 +567,17 @@ namespace CommanderAILab.UI
Debug.LogWarning($"[CoachController] {msg}");
}
// ── Data Models ───────────────────────────────────────────────
[Serializable] private class DeckSummary
// ── Data Models ──────────────────────────────────────────────
[Serializable]
private class DeckSummary
{
public string id;
public string name;
public string commander;
}
[Serializable] private class DeckDetail
[Serializable]
private class DeckDetail
{
public float avg_cmc;
public int land_count;
@@ -317,22 +586,19 @@ namespace CommanderAILab.UI
public Dictionary<string, int> card_types;
}
[Serializable] private class ChatMessage
[Serializable]
private class ChatMessage
{
public string role;
public string content;
}
[Serializable] private class ChatRequest
[Serializable]
private class PreconEntry
{
public string message;
public string deck_id;
public List<ChatMessage> history;
}
[Serializable] private class ChatResponse
{
public string reply;
public string fileName;
public string name;
public string commander;
}
}
}