example_client.py

"""
Simple example client to interact with Acharya API
"""

import requests
import json
from typing import Optional


class AcharyaClient:
    """Simple client for Acharya API."""
    
    def __init__(self, base_url: str = "http://localhost:8000"):
        """Initialize client."""
        self.base_url = base_url
        self.session_id: Optional[str] = None
        self.student_id: Optional[int] = None
    
    def start_conversation(
        self,
        name: str,
        email: str,
        conversation_type: str = "profiling"
    ) -> dict:
        """Start a new conversation."""
        response = requests.post(
            f"{self.base_url}/api/conversation/start",
            json={
                "student_name": name,
                "student_email": email,
                "conversation_type": conversation_type
            }
        )
        response.raise_for_status()
        data = response.json()
        
        self.session_id = data["session_id"]
        print(f"\n[Chanakya]: {data['message']}\n")
        
        return data
    
    def send_message(self, message: str, use_voice: bool = False) -> dict:
        """Send a message in the conversation."""
        if not self.session_id:
            raise ValueError("No active session. Start a conversation first.")
        
        response = requests.post(
            f"{self.base_url}/api/conversation/continue",
            json={
                "session_id": self.session_id,
                "message": message,
                "use_voice": use_voice
            }
        )
        response.raise_for_status()
        data = response.json()
        
        print(f"\n[Chanakya]: {data['message']}\n")
        
        if data.get("extracted_data"):
            print(f"[Extracted Data]: {json.dumps(data['extracted_data'], indent=2)}\n")
        
        return data
    
    def interactive_session(self):
        """Run an interactive conversation session."""
        print("=" * 60)
        print("Acharya - AI Career Mentor")
        print("=" * 60)
        
        name = input("Enter your name: ")
        email = input("Enter your email: ")
        
        self.start_conversation(name, email)
        
        print("Type 'quit' to end the conversation\n")
        
        while True:
            user_input = input("[You]: ")
            
            if user_input.lower() in ['quit', 'exit', 'bye']:
                print("\n[Acharya]: Thank you for the conversation! Keep learning and growing!")
                break
            
            if not user_input.strip():
                continue
            
            self.send_message(user_input)
    
    def generate_roadmap(
        self,
        student_id: int,
        career_goal: str,
        timeline_months: int = 12
    ) -> dict:
        """Generate a career roadmap."""
        response = requests.post(
            f"{self.base_url}/api/roadmap/generate",
            json={
                "student_id": student_id,
                "career_goal": career_goal,
                "current_level": "beginner",
                "target_level": "professional",
                "timeline_months": timeline_months
            }
        )
        response.raise_for_status()
        data = response.json()
        
        print(f"\n✓ Career Roadmap Generated!")
        print(f"Career Goal: {career_goal}")
        print(f"Timeline: {timeline_months} months")
        print(f"Milestones: {data['milestones_count']}")
        print(f"Resources: {data['resources_count']}")
        print(f"Projects: {data['projects_count']}")
        print(f"\nDownload PDF: {self.base_url}{data['pdf_url']}\n")
        
        return data
    
    def start_simulation(
        self,
        student_id: int,
        career_role: str,
        scenario_type: str = "interview"
    ) -> dict:
        """Start a career simulation."""
        response = requests.post(
            f"{self.base_url}/api/simulation/start",
            json={
                "student_id": student_id,
                "career_role": career_role,
                "scenario_type": scenario_type
            }
        )
        response.raise_for_status()
        data = response.json()
        
        print(f"\n=== Career Simulation: {career_role} ===")
        print(f"Scenario Type: {scenario_type}\n")
        print(f"[Scenario]:\n{data['scenario']}\n")
        
        return data
    
    def respond_to_simulation(
        self,
        simulation_id: int,
        response_text: str
    ) -> dict:
        """Submit response to simulation."""
        response = requests.post(
            f"{self.base_url}/api/simulation/respond",
            json={
                "simulation_id": simulation_id,
                "student_response": response_text
            }
        )
        response.raise_for_status()
        data = response.json()
        
        print(f"\n[Mentor Feedback]:\n{data['feedback']}\n")
        print(f"Score: {data['score']}/100\n")
        
        return data


def main():
    """Example usage."""
    client = AcharyaClient()
    
    print("\n" + "="*60)
    print("Acharya API Client - Example Usage")
    print("="*60 + "\n")
    
    # Example 1: Interactive conversation
    print("Example 1: Starting interactive conversation...\n")
    client.interactive_session()
    
    # Uncomment below for more examples:
    
    # # Example 2: Generate roadmap
    # print("\nExample 2: Generating career roadmap...\n")
    # client.generate_roadmap(
    #     student_id=1,
    #     career_goal="Software Engineer",
    #     timeline_months=12
    # )
    
    # # Example 3: Career simulation
    # print("\nExample 3: Starting career simulation...\n")
    # sim = client.start_simulation(
    #     student_id=1,
    #     career_role="Product Manager",
    #     scenario_type="meeting"
    # )
    # 
    # response = input("\n[Your Response]: ")
    # client.respond_to_simulation(
    #     simulation_id=sim["simulation_id"],
    #     response_text=response
    # )


if __name__ == "__main__":
    main()