empty files

test.py

@@ -0,0 +1,230 @@
+"""
+CLI TEST INTERFACE
+==================
+PURPOSE:
+  Command-line interface for testing JARVIS functionality.
+  Provides interactive chat with mode switching and session management.
+  
+WHAT THIS FILE DOES:
+  - Provides interactive CLI for testing both chat modes
+  - Supports switching between general and realtime chat
+  - Displays chat history and session information
+  - Handles user commands (/history, /clear, /quit, etc.)
+  
+USAGE:
+  python test.py
+  
+COMMANDS:
+  1 - Switch to General Chat mode
+  2 - Switch to Realtime Chat mode
+  /history - View chat history
+  /clear - Start new session
+  /quit or /exit - Exit the program
+"""
+
+import requests
+import sys
+from app.utils.time_info import get_greeting, get_time_info
+
+# API configuration
+API_BASE_URL = "http://localhost:8000"
+
+# Chat modes
+MODE_GENERAL = "general"
+MODE_REALTIME = "realtime"
+
+# Global state
+current_mode = MODE_GENERAL
+session_id = None
+
+
+def print_banner():
+    """Print welcome banner."""
+    print("=" * 70)
+    print("  J.A.R.V.I.S - Just A Rather Very Intelligent System")
+    print("  CLI Test Interface")
+    print("=" * 70)
+
+
+def print_time_info():
+    """Print current time information."""
+    time_info = get_time_info()
+    print(f"\n{get_greeting()}!")
+    print(f"Date: {time_info['date']} | Time: {time_info['time']} | Day: {time_info['day']}")
+
+
+def print_mode_info():
+    """Print current chat mode."""
+    mode_name = "General Chat (No Web Search)" if current_mode == MODE_GENERAL else "Realtime Chat (With Web Search)"
+    print(f"\nCurrent Mode: {mode_name}")
+    print(f"Session ID: {session_id if session_id else 'Not started'}")
+
+
+def print_help():
+    """Print available commands."""
+    print("\n" + "=" * 70)
+    print("Available Commands:")
+    print("  1           - Switch to General Chat mode")
+    print("  2           - Switch to Realtime Chat mode")
+    print("  /history    - View chat history")
+    print("  /clear      - Start new session")
+    print("  /quit       - Exit the program")
+    print("  /exit       - Exit the program")
+    print("=" * 70)
+
+
+def send_message(message: str) -> str:
+    """
+    Send message to JARVIS API.
+    
+    Args:
+        message: User message
+        
+    Returns:
+        Assistant response
+    """
+    global session_id
+    
+    # Determine endpoint based on mode
+    endpoint = "/chat" if current_mode == MODE_GENERAL else "/chat/realtime"
+    url = f"{API_BASE_URL}{endpoint}"
+    
+    # Prepare request
+    payload = {
+        "message": message,
+        "session_id": session_id
+    }
+    
+    try:
+        # Send request
+        response = requests.post(url, json=payload, timeout=60)
+        response.raise_for_status()
+        
+        # Parse response
+        data = response.json()
+        session_id = data.get("session_id")
+        return data.get("response", "")
+        
+    except requests.exceptions.ConnectionError:
+        return "Error: Cannot connect to JARVIS server. Make sure it's running (python run.py)"
+    except requests.exceptions.Timeout:
+        return "Error: Request timed out. The server might be busy."
+    except requests.exceptions.HTTPError as e:
+        return f"Error: HTTP {e.response.status_code} - {e.response.text}"
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+
+def get_history():
+    """Get and display chat history."""
+    global session_id
+    
+    if not session_id:
+        print("\nNo active session. Start chatting to create one.")
+        return
+    
+    url = f"{API_BASE_URL}/chat/history/{session_id}"
+    
+    try:
+        response = requests.get(url, timeout=10)
+        response.raise_for_status()
+        
+        data = response.json()
+        messages = data.get("messages", [])
+        
+        if not messages:
+            print("\nNo messages in this session yet.")
+            return
+        
+        print("\n" + "=" * 70)
+        print("Chat History:")
+        print("=" * 70)
+        
+        for msg in messages:
+            role = msg.get("role", "unknown").upper()
+            content = msg.get("content", "")
+            print(f"\n[{role}]")
+            print(content)
+        
+        print("\n" + "=" * 70)
+        
+    except Exception as e:
+        print(f"\nError getting history: {e}")
+
+
+def clear_session():
+    """Clear current session and start new one."""
+    global session_id
+    session_id = None
+    print("\nSession cleared. Your next message will start a new session.")
+
+
+def switch_mode(new_mode: str):
+    """Switch chat mode."""
+    global current_mode
+    current_mode = new_mode
+    mode_name = "General Chat" if new_mode == MODE_GENERAL else "Realtime Chat"
+    print(f"\nSwitched to {mode_name} mode")
+
+
+def main():
+    """Main CLI loop."""
+    global current_mode, session_id
+    
+    # Print banner and info
+    print_banner()
+    print_time_info()
+    print_mode_info()
+    print_help()
+    
+    print("\nType your message and press Enter to chat with JARVIS")
+    print("Type a command (e.g., /help) for available commands\n")
+    
+    # Main loop
+    while True:
+        try:
+            # Get user input
+            user_input = input("\nYou: ").strip()
+            
+            if not user_input:
+                continue
+            
+            # Handle commands
+            if user_input == "/quit" or user_input == "/exit":
+                print("\nGoodbye!")
+                sys.exit(0)
+            
+            elif user_input == "/history":
+                get_history()
+                continue
+            
+            elif user_input == "/clear":
+                clear_session()
+                continue
+            
+            elif user_input == "/help":
+                print_help()
+                continue
+            
+            elif user_input == "1":
+                switch_mode(MODE_GENERAL)
+                continue
+            
+            elif user_input == "2":
+                switch_mode(MODE_REALTIME)
+                continue
+            
+            # Send message to JARVIS
+            print("\nJARVIS: ", end="", flush=True)
+            response = send_message(user_input)
+            print(response)
+            
+        except KeyboardInterrupt:
+            print("\n\nGoodbye!")
+            sys.exit(0)
+        except Exception as e:
+            print(f"\nError: {e}")
+
+
+if __name__ == "__main__":
+    main()