add timer and game phases

app_streamlit_old.py

@@ -3,6 +3,10 @@ import asyncio
 from quickstart import WebSocketHandler, AsyncHumeClient, ChatConnectOptions, MicrophoneInterface, SubscribeEvent
 import os
 from dotenv import load_dotenv
+from dataclasses import dataclass, field
+from typing import List, Dict
+import time
+from timer import AsyncTimer
 
 # Page config
 st.set_page_config(
@@ -21,6 +25,18 @@ if 'messages' not in st.session_state:
     st.session_state.messages = []
     st.session_state.recording = False
 
+@dataclass
+class GameState:
+    round: int = 1
+    chat_group_id: str = None
+    user_decisions: List[str] = field(default_factory=list)
+    ai_decisions: List[str] = field(default_factory=list)
+    conversation_history: List[Dict] = field(default_factory=list)
+    emotion_history: List[Dict] = field(default_factory=list)
+    scores: Dict[str, int] = field(default_factory=lambda: {"user": 0, "ai": 0})
+    phase: str = "INIT"  # INIT, CONVERSATION, USER_DECISION, AI_DECISION, RESULTS, NEXT_ROUND
+    timer_start: float = None
+
 class StreamlitWebSocketHandler(WebSocketHandler):
     async def on_message(self, message: SubscribeEvent):
         await super().on_message(message)
@@ -75,6 +91,21 @@ async def run_chat():
         
         await microphone_task
 
+async def handle_conversation_phase():
+    """Handle the CONVERSATION phase with timer and chat"""
+    timer_placeholder = st.empty()
+    
+    async def on_timer_complete():
+        st.session_state.game.phase = "USER_DECISION"
+        st.session_state.game.timer_start = None
+        st.rerun()
+    
+    timer = AsyncTimer(30, on_timer_complete, timer_placeholder)
+    await asyncio.gather(
+        timer.start(),
+        run_chat()
+    )
+
 # Display welcome message
 if len(st.session_state.messages) == 0:
     st.info("Welcome to the Hume.ai Voice Chat Demo! Click the button below to start chatting.")
@@ -98,3 +129,51 @@ with col2:
         st.session_state.messages = []
         st.session_state.recording = False
         st.rerun()
+
+# Initialize session state
+if 'game' not in st.session_state:
+    st.session_state.game = GameState()
+
+# Game UI based on phase
+if st.session_state.game.phase == "INIT":
+    if st.button("Start Game"):
+        st.session_state.game.phase = "CONVERSATION"
+        st.session_state.game.timer_start = time.time()
+        st.rerun()
+
+elif st.session_state.game.phase == "CONVERSATION":
+    asyncio.run(handle_conversation_phase())
+
+elif st.session_state.game.phase == "USER_DECISION":
+    col1, col2 = st.columns(2)
+    with col1:
+        if st.button("Cooperate"):
+            st.session_state.game.user_decisions.append("C")
+            st.session_state.game.phase = "AI_DECISION"
+            st.rerun()
+    with col2:
+        if st.button("Defect"):
+            st.session_state.game.user_decisions.append("D")
+            st.session_state.game.phase = "AI_DECISION"
+            st.rerun()
+
+elif st.session_state.game.phase == "AI_DECISION":
+    # Run AI decision chat
+    asyncio.run(run_chat())
+
+elif st.session_state.game.phase == "RESULTS":
+    # Display round results
+    # Update scores
+    if st.button("Next Round"):
+        st.session_state.game.round += 1
+        st.session_state.game.phase = "CONVERSATION"
+        st.session_state.game.timer_start = time.time()
+        st.rerun()
+
+def check_timer():
+    if st.session_state.game.timer_start:
+        elapsed = time.time() - st.session_state.game.timer_start
+        if elapsed >= 30:
+            st.session_state.game.phase = "USER_DECISION"
+            st.session_state.game.timer_start = None
+            st.rerun()

timer.py

@@ -0,0 +1,52 @@
+import asyncio
+import time
+import streamlit as st
+from typing import Callable, Optional
+
+class AsyncTimer:
+    def __init__(self, duration: int, on_complete: Callable, placeholder: Optional[st.empty] = None):
+        """
+        Initialize async timer
+        
+        Args:
+            duration: Duration in seconds
+            on_complete: Callback function to run when timer completes
+            placeholder: Optional streamlit placeholder to display timer
+        """
+        self.duration = duration
+        self.on_complete = on_complete
+        self.placeholder = placeholder or st.empty()
+        self.start_time = None
+        self.is_running = False
+
+    def format_time(self, seconds: int) -> str:
+        """Format seconds into MM:SS"""
+        mm, ss = seconds//60, seconds%60
+        return f"{mm:02d}:{ss:02d}"
+
+    async def start(self):
+        """Start the countdown timer"""
+        self.start_time = time.time()
+        self.is_running = True
+        
+        while self.is_running:
+            elapsed = int(time.time() - self.start_time)
+            remaining = max(0, self.duration - elapsed)
+            
+            # Update display
+            self.placeholder.metric(
+                "Time Remaining", 
+                self.format_time(remaining)
+            )
+            
+            # Check if timer completed
+            if remaining <= 0:
+                self.is_running = False
+                await self.on_complete()
+                break
+                
+            await asyncio.sleep(0.1)  # Small delay to prevent excessive updates
+
+    def stop(self):
+        """Stop the timer"""
+        self.is_running = False