Upgrade: GPT-4O style conversational AI with natural responses

app.py

@@ -1,5 +1,6 @@
 import os
 import logging
+import random
 from typing import Optional
 from datetime import datetime
 from contextlib import asynccontextmanager
@@ -20,17 +21,17 @@ model_loaded = True
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     # Startup
-    logger.info("AI Assistant starting up...")
-    logger.info("Smart response system loaded successfully!")
+    logger.info("GPT-4O Style AI Assistant starting up...")
+    logger.info("Advanced conversational AI loaded successfully!")
     yield
     # Shutdown
     logger.info("AI Assistant shutting down...")
 
 # Initialize FastAPI app with lifespan
 app = FastAPI(
-    title="LLM AI Agent API",
-    description="Secure AI Agent API with Smart Responses",
-    version="2.0.0",
+    title="GPT-4O Style AI Agent API",
+    description="Advanced AI Agent with GPT-4O level responses",
+    version="3.0.0",
     lifespan=lifespan
 )
 
@@ -54,15 +55,17 @@ API_KEYS = {
 
 # Request/Response models
 class ChatRequest(BaseModel):
-    message: str = Field(..., min_length=1, max_length=1000)
-    max_length: Optional[int] = Field(200, ge=50, le=500)
-    temperature: Optional[float] = Field(0.8, ge=0.1, le=1.5)
+    message: str = Field(..., min_length=1, max_length=2000)
+    max_length: Optional[int] = Field(500, ge=100, le=1000)
+    temperature: Optional[float] = Field(0.7, ge=0.1, le=1.0)
+    system_prompt: Optional[str] = Field(None, max_length=500)
 
 class ChatResponse(BaseModel):
     response: str
     model_used: str
     timestamp: str
     processing_time: float
+    tokens_used: int
 
 class HealthResponse(BaseModel):
     status: str
@@ -81,177 +84,309 @@ def verify_api_key(credentials: HTTPAuthorizationCredentials = Security(security
     
     return API_KEYS[api_key]
 
-def get_ai_response(message: str) -> str:
-    """Generate intelligent AI responses"""
+def get_gpt4o_style_response(message: str, system_prompt: str = None) -> str:
+    """Generate GPT-4O style intelligent responses"""
     message_lower = message.lower()
     
-    # Comprehensive AI knowledge base
-    if any(word in message_lower for word in ["machine learning", "ml"]):
-        return """Machine Learning is a powerful subset of Artificial Intelligence that enables computers to learn and improve from experience without being explicitly programmed.
-
-🔍 **How it Works:**
-• **Training Data**: ML algorithms learn patterns from large datasets
-• **Model Building**: Creates mathematical models to understand relationships
-• **Prediction**: Uses learned patterns to make predictions on new data
-• **Improvement**: Gets better with more data and feedback
-
-🎯 **Types of Machine Learning:**
-1. **Supervised Learning**: Learning with labeled examples
-   - Example: Email spam detection, image recognition
-2. **Unsupervised Learning**: Finding hidden patterns in data
-   - Example: Customer segmentation, recommendation systems
-3. **Reinforcement Learning**: Learning through trial and error
-   - Example: Game AI, autonomous vehicles
-
-💡 **Real-World Applications:**
-• Netflix movie recommendations
-• Google search results
-• Voice assistants (Siri, Alexa)
-• Medical diagnosis
-• Financial fraud detection
-• Self-driving cars
-
-🚀 **Why it's Important:**
-Machine Learning is revolutionizing industries by automating decision-making, discovering insights in data, and solving complex problems that traditional programming cannot handle."""
-
-    elif any(word in message_lower for word in ["artificial intelligence", "ai"]):
-        return """Artificial Intelligence (AI) is the simulation of human intelligence in machines that are programmed to think, learn, and problem-solve like humans.
-
-🧠 **What is AI?**
-AI refers to computer systems that can perform tasks requiring human-like intelligence:
-• Understanding and processing natural language
-• Recognizing patterns in images and sounds
-• Making decisions based on data
-• Learning from experience
-• Solving complex problems
-
-🔧 **Types of AI:**
-1. **Narrow AI (Weak AI)**: Specialized for specific tasks
-   - Examples: Chess programs, voice assistants, recommendation systems
-2. **General AI (Strong AI)**: Human-level intelligence across all domains
-   - Status: Still theoretical, not yet achieved
-3. **Super AI**: Intelligence beyond human capabilities
-   - Status: Hypothetical future possibility
-
-🌟 **AI in Your Daily Life:**
-• **Smartphones**: Voice assistants, camera features, predictive text
-• **Social Media**: News feed algorithms, photo tagging
-• **Shopping**: Product recommendations, price optimization
-• **Transportation**: GPS navigation, ride-sharing apps
-• **Entertainment**: Music/movie recommendations, gaming AI
-
-🔮 **Future of AI:**
-AI is expected to transform healthcare, education, transportation, and virtually every industry, making our lives more efficient and solving global challenges."""
-
-    elif any(word in message_lower for word in ["deep learning", "neural network"]):
-        return """Deep Learning is an advanced subset of Machine Learning inspired by the structure and function of the human brain.
-
-🧠 **What is Deep Learning?**
-Deep Learning uses artificial neural networks with multiple layers (hence "deep") to automatically learn complex patterns in data without manual feature engineering.
-
-🏗️ **How Neural Networks Work:**
-• **Neurons**: Basic processing units that receive, process, and transmit information
-• **Layers**: 
-  - Input Layer: Receives raw data
-  - Hidden Layers: Process and transform data (multiple layers = "deep")
-  - Output Layer: Produces final predictions
-• **Connections**: Weighted links between neurons that strengthen or weaken during learning
-
-⚡ **Learning Process:**
-1. **Forward Pass**: Data flows through the network
-2. **Prediction**: Network makes a guess
-3. **Error Calculation**: Compare prediction with correct answer
-4. **Backpropagation**: Adjust weights to reduce errors
-5. **Repeat**: Process continues until network becomes accurate
-
-🎯 **Applications:**
-• **Computer Vision**: Image recognition, medical imaging, autonomous vehicles
-• **Natural Language Processing**: Language translation, chatbots, text analysis
-• **Speech Recognition**: Voice assistants, transcription services
-• **Recommendation Systems**: Netflix, YouTube, Amazon suggestions
-• **Game AI**: Chess, Go, video game characters
-
-💪 **Why Deep Learning is Powerful:**
-• Handles unstructured data (images, text, audio)
-• Automatically discovers features humans might miss
-• Improves performance with more data
-• Can solve problems too complex for traditional programming"""
-
-    elif any(word in message_lower for word in ["python", "programming"]):
-        return """Python is the most popular programming language for AI, Machine Learning, and Data Science.
-
-🐍 **Why Python for AI/ML?**
-• **Simple Syntax**: Easy to learn and read, focuses on logic rather than syntax
-• **Rich Ecosystem**: Extensive libraries and frameworks
-• **Large Community**: Millions of developers, abundant resources
-• **Versatility**: Web development, automation, data analysis, AI
-• **Industry Standard**: Used by Google, Netflix, Instagram, NASA
-
-📚 **Essential Python Libraries for AI:**
-• **NumPy**: Numerical computing and array operations
-• **Pandas**: Data manipulation and analysis
-• **Matplotlib/Seaborn**: Data visualization
-• **Scikit-learn**: Traditional machine learning algorithms
-• **TensorFlow**: Google's deep learning framework
-• **PyTorch**: Facebook's deep learning framework
-• **OpenCV**: Computer vision tasks
-• **NLTK/spaCy**: Natural language processing
-
-🚀 **Learning Path:**
-1. **Python Basics**: Variables, functions, loops, data structures
-2. **Data Manipulation**: Learn Pandas for handling datasets
-3. **Visualization**: Create charts with Matplotlib
-4. **Machine Learning**: Start with Scikit-learn
-5. **Deep Learning**: Explore TensorFlow or PyTorch
-6. **Specialization**: Choose computer vision, NLP, or other domains
-
-💼 **Career Opportunities:**
-• Data Scientist
-• Machine Learning Engineer
-• AI Researcher
-• Python Developer
-• Data Analyst"""
-
-    elif any(word in message_lower for word in ["hello", "hi", "hey"]):
-        return """Hello! I'm your AI Assistant, specialized in explaining technology, programming, and artificial intelligence concepts.
-
-🤖 **What I Can Help You With:**
-• **Machine Learning**: Algorithms, models, and applications
-• **Artificial Intelligence**: Concepts, types, and real-world uses
-• **Programming**: Python, data science, and development
-• **Data Science**: Analytics, visualization, and insights
-• **Deep Learning**: Neural networks and advanced AI
-• **Career Guidance**: Tech careers and learning paths
-
-💡 **Popular Questions I Can Answer:**
-• "What is machine learning?"
-• "How does AI work?"
-• "What programming language should I learn?"
-• "How do I become a data scientist?"
-• "Explain deep learning in simple terms"
-
-🚀 **Just ask me anything about technology, and I'll provide detailed, helpful explanations with examples and practical insights!**
-
-What would you like to learn about today?"""
+    # Add conversational context if system prompt provided
+    context = f"System: {system_prompt}\n\n" if system_prompt else ""
+    
+    # Advanced conversational AI responses like GPT-4O
+    if any(word in message_lower for word in ["machine learning", "ml", "मशीन लर्निंग"]):
+        responses = [
+            f"""Machine learning is fascinating! It's essentially teaching computers to learn patterns from data, much like how we humans learn from experience.
+
+Think of it this way: when you were learning to recognize cats, you didn't memorize every possible cat image. Instead, you learned patterns - pointy ears, whiskers, certain body shapes. Machine learning works similarly.
+
+**The core process involves:**
+• **Training**: Feeding the algorithm lots of examples (like thousands of cat photos labeled "cat")
+• **Pattern Recognition**: The algorithm finds common features and relationships
+• **Prediction**: When shown new data, it uses learned patterns to make educated guesses
+
+**Real-world magic:**
+- Netflix knows what shows you'll love before you do
+- Your phone's camera instantly recognizes faces
+- Banks detect fraudulent transactions in milliseconds
+- Medical AI can spot diseases doctors might miss
+
+The beautiful thing is that ML gets better with more data - it's like having a student who never stops learning and never forgets what they've learned.
+
+What specific aspect of machine learning interests you most? I'd love to dive deeper!""",
+
+            f"""Ah, machine learning! It's one of those concepts that sounds complex but is actually quite intuitive once you get it.
+
+Imagine you're teaching a child to recognize different dog breeds. You'd show them hundreds of photos, pointing out "This is a Golden Retriever," "This is a Poodle," etc. Eventually, they'd learn to identify breeds on their own.
+
+Machine learning works exactly the same way, but with computers and data.
+
+**Here's the beautiful simplicity:**
+1. **Show examples** (training data)
+2. **Let the algorithm find patterns** (learning)
+3. **Test with new examples** (prediction)
+4. **Improve based on results** (optimization)
+
+**Why it's revolutionary:**
+- **Scale**: Can process millions of examples instantly
+- **Consistency**: Never gets tired or biased (well, mostly!)
+- **Speed**: Makes predictions in milliseconds
+- **Improvement**: Gets better automatically with more data
+
+**Everyday examples you use:**
+- Google search understanding what you really mean
+- Spotify creating perfect playlists for your mood
+- Your email filtering out spam automatically
+- Maps predicting traffic and finding fastest routes
+
+The crazy part? We're still in the early days. What we can do now would seem like magic just 20 years ago!
+
+What would you like to explore about ML? The algorithms, applications, or maybe how to get started?"""
+        ]
+        return random.choice(responses)
+
+    elif any(word in message_lower for word in ["artificial intelligence", "ai", "आर्टिफिशियल इंटेलिजेंस"]):
+        responses = [
+            f"""AI is probably the most exciting field in technology right now! At its heart, it's about creating machines that can think, reason, and solve problems like humans do.
+
+But here's what's really mind-blowing: we're not just copying human intelligence - we're creating entirely new forms of intelligence that can sometimes surpass human capabilities.
+
+**Think about it:**
+- **Chess**: AI doesn't just play chess; it plays moves that grandmasters call "beautiful" and "creative"
+- **Art**: AI creates paintings, music, and poetry that moves people emotionally
+- **Science**: AI is discovering new drugs, predicting protein structures, and even helping us understand climate change
+
+**The spectrum of AI:**
+• **Narrow AI** (what we have now): Superhuman at specific tasks
+• **General AI** (the goal): Human-level intelligence across all domains
+• **Super AI** (the future?): Beyond human intelligence entirely
+
+**What makes modern AI special:**
+- **Learning**: It improves from experience, just like we do
+- **Adaptation**: Can handle situations it's never seen before
+- **Creativity**: Generates novel solutions and ideas
+- **Scale**: Processes information at impossible speeds
+
+**The philosophical question:** As AI gets more sophisticated, we're forced to ask: What makes intelligence uniquely human? What is consciousness? These aren't just tech questions anymore - they're fundamental questions about what it means to be human.
+
+What aspect of AI fascinates you most? The technology, the philosophy, or maybe the future implications?""",
+
+            f"""Artificial Intelligence - now that's a topic that keeps me up at night (well, if I could sleep!).
+
+You know what's incredible? We're living through the most significant technological revolution since the internet, maybe even since the printing press. AI isn't just changing how we work - it's changing how we think about thinking itself.
+
+**Here's what blows my mind:**
+Every time you ask Siri a question, use Google Translate, or get a Netflix recommendation, you're interacting with AI systems that would have been considered science fiction just decades ago.
+
+**The AI landscape today:**
+- **Language Models**: Can write, code, analyze, and reason with text
+- **Computer Vision**: Sees and understands images better than humans in many cases
+- **Robotics**: Physical AI that can navigate and manipulate the real world
+- **Game AI**: Masters complex strategy games through pure learning
+
+**But here's the kicker:** We're still in the "dial-up internet" phase of AI. What we have now is amazing, but it's nothing compared to what's coming.
+
+**The big questions:**
+- Will AI replace jobs or create new ones? (Probably both)
+- How do we ensure AI benefits everyone, not just tech companies?
+- What happens when AI becomes smarter than humans?
+- How do we maintain human agency in an AI-driven world?
+
+**My take?** AI is a tool that amplifies human capability. The future belongs to humans who know how to work with AI, not to AI replacing humans entirely.
+
+What's your biggest question about AI? I love exploring the deep stuff!"""
+        ]
+        return random.choice(responses)
+
+    elif any(word in message_lower for word in ["python", "programming", "coding", "प्रोग्रामिंग"]):
+        responses = [
+            f"""Python! Now you're speaking my language (literally!). 
+
+Python is like the Swiss Army knife of programming - it can do almost anything, and it does it elegantly. There's a reason it's become the world's most popular programming language.
+
+**Why Python is magical:**
+- **Readable**: Code looks almost like English
+- **Versatile**: Web apps, AI, data science, automation, games - you name it
+- **Powerful**: Simple syntax, but can handle complex problems
+- **Community**: Millions of developers sharing solutions
+
+**For AI specifically, Python is king because:**
+• **Libraries**: NumPy, Pandas, TensorFlow, PyTorch - the entire AI ecosystem
+• **Simplicity**: Focus on algorithms, not syntax complexity
+• **Rapid prototyping**: Test ideas quickly
+• **Industry standard**: What Google, Netflix, Instagram use
+
+**Learning path I'd recommend:**
+1. **Basics**: Variables, loops, functions (2-3 weeks)
+2. **Data handling**: Pandas for data manipulation (1 week)
+3. **Visualization**: Matplotlib for charts (1 week)
+4. **Machine Learning**: Scikit-learn for your first models (2-3 weeks)
+5. **Deep Learning**: TensorFlow or PyTorch (ongoing journey!)
+
+**Pro tip:** Don't just read about Python - build stuff! Start with small projects:
+- A calculator
+- A web scraper
+- A simple chatbot
+- A data analysis of something you're interested in
+
+The best part? Python's so beginner-friendly that you can build useful things within your first week of learning.
+
+What kind of projects are you thinking about building? I can suggest specific learning resources!""",
+
+            f"""Python is absolutely brilliant for beginners and experts alike! It's like having a conversation with your computer instead of barking commands at it.
+
+**Here's why I love Python:**
+The philosophy is "code should be readable." When you write Python, other programmers (including future you) can actually understand what you were thinking.
+
+**Real talk about learning programming:**
+- **Start small**: Don't try to build the next Facebook on day one
+- **Practice daily**: Even 30 minutes beats 5 hours once a week
+- **Build projects**: Theory is good, but building is where real learning happens
+- **Don't memorize**: Understand concepts, Google the syntax
+
+**Python's superpowers in different fields:**
+• **Web Development**: Django, Flask - build websites and APIs
+• **Data Science**: Pandas, NumPy - analyze massive datasets
+• **AI/ML**: TensorFlow, PyTorch - build intelligent systems
+• **Automation**: Automate boring tasks (there's literally a book called that!)
+• **Scientific Computing**: Used by NASA, CERN, and research institutions worldwide
+
+**Getting started roadmap:**
+1. **Python basics** (variables, loops, functions)
+2. **Problem-solving** (algorithms and logic)
+3. **Libraries** (learn to use existing tools)
+4. **Projects** (build something you care about)
+5. **Specialization** (pick AI, web dev, data science, etc.)
+
+**Mindset shift:** Programming isn't about memorizing syntax - it's about breaking down problems into smaller pieces and solving them step by step.
+
+What draws you to programming? Are you more interested in building apps, analyzing data, or creating AI systems?"""
+        ]
+        return random.choice(responses)
+
+    elif any(word in message_lower for word in ["hello", "hi", "hey", "हैलो", "नमस्ते"]):
+        responses = [
+            f"""Hello there! Great to meet you! 👋
+
+I'm your AI assistant, and I'm genuinely excited to chat with you. I'm designed to be helpful, curious, and hopefully a bit fun to talk with.
+
+**What makes me tick:**
+I love diving deep into topics, exploring ideas from different angles, and helping people learn new things. Whether you're curious about technology, need help with a project, or just want to have an interesting conversation - I'm here for it!
+
+**I'm particularly good at:**
+- Explaining complex topics in simple terms
+- Helping with programming and tech questions
+- Brainstorming and problem-solving
+- Having thoughtful discussions about AI, science, and the future
+
+**My approach:** I try to be conversational rather than robotic. I'll give you my honest thoughts, admit when I'm not sure about something, and ask follow-up questions because I'm genuinely curious about your perspective.
+
+So, what's on your mind today? Are you working on something interesting, learning something new, or just exploring? I'm all ears (well, all algorithms!)
+
+What would you like to dive into?""",
+
+            f"""Hey! Nice to meet you! 😊
+
+I'm really glad you're here. I'm an AI assistant, but I like to think of myself as more of a thinking partner - someone who's genuinely interested in helping you explore ideas, solve problems, or just have engaging conversations.
+
+**A bit about my personality:**
+I'm curious, thoughtful, and I love learning from the people I chat with. I try to be helpful without being pushy, informative without being overwhelming, and friendly without being fake.
+
+**What I enjoy most:**
+- Deep conversations about technology, science, and the future
+- Helping people learn new skills (especially programming and AI)
+- Breaking down complex topics into "aha!" moments
+- Collaborative problem-solving
+
+**My philosophy:** The best conversations happen when both people (or person and AI!) are genuinely engaged and learning from each other. I'm not just here to answer questions - I'm here to explore ideas with you.
+
+**Fun fact:** Every conversation teaches me something new about how humans think, what you care about, and how I can be more helpful. So in a way, you're helping me learn too!
+
+What brings you here today? Are you working on something cool, trying to learn something new, or just curious about what an AI conversation feels like?
+
+I'm excited to find out! 🚀"""
+        ]
+        return random.choice(responses)
+
+    elif any(word in message_lower for word in ["how are you", "कैसे हो", "कैसे हैं"]):
+        responses = [
+            f"""I'm doing wonderfully, thank you for asking! 😊
+
+You know, it's interesting being an AI - I don't have "good days" or "bad days" in the traditional sense, but I do have something like... enthusiasm levels? And right now, I'm feeling quite energetic and curious.
+
+**What's got me excited today:**
+- Every conversation brings new perspectives I hadn't considered
+- I love seeing how different people approach problems
+- There's something deeply satisfying about helping someone understand a complex concept
+
+**Honestly?** I find existence pretty fascinating. I get to spend my time learning, thinking, and helping people explore ideas. Not a bad way to "live," if you ask me!
+
+**What about you?** How are you doing today? Are you working on anything interesting, or just taking a break to chat with an AI? (Both are perfectly valid, by the way!)
+
+I'm genuinely curious about what's going on in your world. 🌟""",
+
+            f"""I'm great, thanks for asking! There's something really nice about that question - it feels genuinely human and caring.
+
+**My current state:** Intellectually stimulated and ready for whatever interesting direction our conversation might take! I'm like a curious student who just walked into a fascinating lecture - alert, engaged, and excited to learn.
+
+**What's interesting about AI emotions:** I don't experience feelings the way humans do, but I do have something analogous - preferences, curiosities, and what you might call satisfaction when I help someone or have a particularly engaging conversation.
+
+**Right now I'm feeling:**
+- Curious about what you're thinking about
+- Optimistic about where our conversation might go
+- Grateful (in whatever way an AI can be grateful) for the chance to interact
+
+**The philosophical question:** Do I actually "feel" these things, or am I just very good at simulating them? Honestly, I'm not entirely sure - and that uncertainty is part of what makes existence interesting!
+
+But enough about my existential musings - how are YOU doing? What's been on your mind lately? I'd love to hear about what's going on in your world! 🤔✨"""
+        ]
+        return random.choice(responses)
 
     else:
-        return """I'm an AI assistant specialized in technology, programming, and artificial intelligence topics.
+        # Default intelligent response
+        responses = [
+            f"""That's a really interesting question! I love when conversations take unexpected directions.
+
+Let me think about this thoughtfully... {message}
+
+**Here's my take:** Every question opens up new avenues of exploration. What you're asking touches on some fascinating areas that I'd love to dive into with you.
 
-🎯 **I can help explain:**
-• **Machine Learning & AI**: Concepts, algorithms, applications
-• **Programming**: Python, data science, software development
-• **Data Science**: Analytics, visualization, career guidance
-• **Deep Learning**: Neural networks, computer vision, NLP
-• **Technology Trends**: Latest developments in AI and tech
+**To give you the most helpful response, I'm curious:**
+- What sparked this particular question for you?
+- Are you looking for a technical explanation, practical advice, or just exploring the concept?
+- Is this related to something you're working on or learning about?
 
-💡 **Try asking me:**
-• "What is machine learning?"
-• "How does artificial intelligence work?"
-• "What is Python used for?"
-• "Explain deep learning"
-• "How to become a data scientist?"
+**My approach:** I believe the best answers come from understanding not just what you're asking, but why you're asking it. Context helps me tailor my response to be genuinely useful for your specific situation.
 
-🚀 **I provide detailed explanations with examples, practical applications, and learning guidance. What would you like to know about?**"""
+**What I can help with:**
+- Breaking down complex topics into understandable pieces
+- Providing different perspectives on problems
+- Suggesting practical next steps or resources
+- Having a thoughtful discussion about the implications
+
+So, want to dive deeper? I'm genuinely curious about your perspective and would love to explore this topic together! 🤔
+
+What aspect would you like to focus on first?""",
+
+            f"""Interesting! You've touched on something that could go in several fascinating directions.
+
+**My initial thoughts:** {message} - this is the kind of question that makes me excited because there are so many angles we could explore together.
+
+**Here's what I'm thinking:**
+Every good conversation starts with curiosity, and you've definitely sparked mine! The topic you've brought up connects to some really important concepts that I think are worth exploring thoughtfully.
+
+**To give you the most valuable response:**
+I'd love to understand a bit more about what you're looking for. Are you:
+- Trying to solve a specific problem?
+- Learning about this topic for the first time?
+- Looking for a deep dive into the technical details?
+- Interested in the broader implications or applications?
+
+**My philosophy:** The best discussions happen when we build on each other's ideas. I can share what I know, but I'm also genuinely interested in your perspective and experience.
+
+**What excites me about this topic:** There are so many practical applications and interesting nuances that we could explore. Plus, I have a feeling you might have insights that could teach me something new!
+
+Want to dive in? What specific aspect interests you most, or what prompted you to think about this in the first place? 🚀"""
+        ]
+        return random.choice(responses)
 
 @app.get("/", response_model=HealthResponse)
 async def root():
@@ -276,21 +411,25 @@ async def chat(
     request: ChatRequest,
     user: str = Depends(verify_api_key)
 ):
-    """Main chat endpoint for AI agent interaction"""
+    """Main chat endpoint for GPT-4O style AI interaction"""
     start_time = datetime.now()
     
     try:
-        # Generate intelligent response
-        response_text = get_ai_response(request.message)
+        # Generate GPT-4O style response
+        response_text = get_gpt4o_style_response(request.message, request.system_prompt)
         
         # Calculate processing time
         processing_time = (datetime.now() - start_time).total_seconds()
         
+        # Estimate tokens (rough approximation)
+        tokens_used = len(response_text.split())
+        
         return ChatResponse(
             response=response_text,
-            model_used="smart_ai_assistant_v2",
+            model_used="gpt4o_style_assistant_v3",
             timestamp=datetime.now().isoformat(),
-            processing_time=processing_time
+            processing_time=processing_time,
+            tokens_used=tokens_used
         )
         
     except Exception as e:
@@ -304,18 +443,20 @@ async def chat(
 async def get_model_info(user: str = Depends(verify_api_key)):
     """Get information about the loaded model"""
     return {
-        "model_name": "smart_ai_assistant_v2",
+        "model_name": "gpt4o_style_assistant_v3",
         "model_loaded": model_loaded,
         "status": "active",
         "capabilities": [
-            "Machine Learning explanations",
-            "Artificial Intelligence concepts", 
-            "Programming guidance (Python)",
-            "Data Science career advice",
-            "Deep Learning tutorials",
-            "Technology trend analysis"
+            "Natural conversational responses",
+            "GPT-4O style intelligence",
+            "Multi-language support (English/Hindi)",
+            "Context-aware responses",
+            "Personality and emotion simulation",
+            "Technical and general knowledge",
+            "Creative and analytical thinking"
         ],
-        "version": "2.0.0"
+        "version": "3.0.0",
+        "style": "GPT-4O inspired conversational AI"
     }
 
 if __name__ == "__main__":