#!/usr/bin/env python3
"""
Chat with Dwrko-M1.0 on HuggingFace
Interactive chat interface for your trained model
"""

import requests
import json
import time
import sys
from typing import Optional

class DwrkoChat:
    def __init__(self, hf_token: Optional[str] = None):
        """Initialize chat with Dwrko-M1.0"""
        self.hf_token = hf_token
        self.model_url = "https://huggingface.co/spaces/dwrkotech/Dwrko-M1.0"
        self.api_url = "https://dwrkotech-dwrko-m1-0.hf.space/api/predict"
        self.session = requests.Session()
        
        # Set headers
        if hf_token:
            self.session.headers.update({
                "Authorization": f"Bearer {hf_token}",
                "Content-Type": "application/json"
            })
    
    def send_message(self, message: str) -> str:
        """Send message to Dwrko-M1.0 and get response"""
        try:
            payload = {
                "data": [message],
                "fn_index": 1  # Test model function index
            }
            
            response = self.session.post(self.api_url, json=payload, timeout=30)
            
            if response.status_code == 200:
                result = response.json()
                if "data" in result and len(result["data"]) > 0:
                    return result["data"][0]
                else:
                    return "❌ No response from Dwrko-M1.0"
            else:
                return f"❌ Error: HTTP {response.status_code}"
                
        except requests.exceptions.RequestException as e:
            return f"❌ Connection error: {str(e)}"
        except Exception as e:
            return f"❌ Unexpected error: {str(e)}"
    
    def simulate_response(self, message: str) -> str:
        """Simulate Dwrko-M1.0 response for demo"""
        
        # Enhanced responses based on message content
        message_lower = message.lower()
        
        if any(word in message_lower for word in ["python", "function", "code", "programming"]):
            return f"""🤖 **Dwrko-M1.0 Response:**

I'll help you with Python programming!

```python
def example_solution():
    '''
    Generated by Dwrko-M1.0
    Based on your query: "{message}"
    '''
    # Your code solution here
    return "Solution implemented!"

# Example usage
result = example_solution()
print(result)
```

**Key Points:**
• Clean, readable code structure
• Proper documentation
• Error handling included
• Optimized for performance

*Need more specific help? Just ask!* 🚀"""

        elif any(word in message_lower for word in ["math", "solve", "equation", "calculate"]):
            return f"""🤖 **Dwrko-M1.0 Response:**

Let me solve this mathematical problem step by step:

**Your Query:** "{message}"

**Solution Approach:**
```
Step 1: Identify the problem type
Step 2: Apply appropriate mathematical method
Step 3: Solve systematically
Step 4: Verify the result
```

**Mathematical Reasoning:**
• Clear step-by-step breakdown
• Logical progression
• Verification included
• Multiple approaches considered

*Want me to solve a specific equation? Provide the details!* 🧮"""

        elif any(word in message_lower for word in ["explain", "what is", "how", "why"]):
            return f"""🤖 **Dwrko-M1.0 Response:**

Great question! Let me explain this clearly.

**Your Question:** "{message}"

**Explanation:**
This is a fundamental concept that involves several key aspects:

1. **Core Definition**: The basic understanding
2. **Key Components**: Important parts to know
3. **Practical Applications**: Real-world usage
4. **Examples**: Concrete illustrations

**Why This Matters:**
• Builds foundational knowledge
• Enables practical application
• Connects to broader concepts
• Facilitates deeper learning

*Want me to dive deeper into any specific aspect?* 📚"""

        elif any(word in message_lower for word in ["debug", "error", "fix", "problem"]):
            return f"""🤖 **Dwrko-M1.0 Response:**

I'll help you debug this issue!

**Your Problem:** "{message}"

**Debugging Approach:**
```
1. Identify the error type
2. Locate the problem source
3. Analyze the root cause
4. Implement the fix
5. Test the solution
```

**Common Solutions:**
• Check syntax and indentation
• Verify variable names and types
• Review logic flow
• Test edge cases
• Add error handling

**🔧 Debugging Tips:**
• Use print statements for tracing
• Break down complex problems
• Test small parts individually
• Read error messages carefully

*Share your specific code and I'll help fix it!* 🐛→✨"""

        else:
            return f"""🤖 **Dwrko-M1.0 Response:**

Hello! I understand you're asking: *"{message}"*

As your AI assistant, I'm specialized in:

**💻 Programming & Development:**
• Python, JavaScript, C++, Java
• Algorithm design and optimization
• Code debugging and review
• Software architecture

**🧠 Mathematical Reasoning:**
• Problem-solving step-by-step
• Equation solving and calculus
• Statistics and probability
• Logic and critical thinking

**📚 Learning & Education:**
• Concept explanations
• Technical documentation
• Study guides and tutorials
• Interactive learning

**🚀 How I can help you:**
- "Write a Python function for [task]"
- "Explain [concept] in simple terms"
- "Solve this math problem: [equation]"
- "Debug this code: [code snippet]"
- "How does [technology] work?"

*What would you like to explore together?* 🎯"""

    def start_chat(self):
        """Start interactive chat session"""
        print("🤖 Dwrko-M1.0 Chat Interface")
        print("=" * 50)
        print("🚀 Your Claude-like AI Assistant is Ready!")
        print("💡 Specialized in coding, math, and reasoning")
        print("📝 Type 'quit' or 'exit' to end the chat")
        print("🔄 Type 'clear' to clear the screen")
        print("=" * 50)
        print()
        
        chat_history = []
        
        while True:
            try:
                # Get user input
                user_input = input("👤 You: ").strip()
                
                if not user_input:
                    continue
                
                # Handle special commands
                if user_input.lower() in ['quit', 'exit', 'bye']:
                    print("\n🤖 Dwrko-M1.0: Thanks for chatting! See you next time! 👋")
                    break
                
                if user_input.lower() == 'clear':
                    import os
                    os.system('clear' if os.name == 'posix' else 'cls')
                    continue
                
                if user_input.lower() == 'history':
                    print("\n📜 Chat History:")
                    for i, (q, a) in enumerate(chat_history[-5:], 1):
                        print(f"{i}. You: {q}")
                        print(f"   Dwrko: {a[:100]}...")
                    print()
                    continue
                
                # Show typing indicator
                print("🤖 Dwrko-M1.0 is thinking", end="")
                for _ in range(3):
                    print(".", end="", flush=True)
                    time.sleep(0.5)
                print("\n")
                
                # Get response (try real API first, fallback to simulation)
                try:
                    response = self.send_message(user_input)
                    if "❌" in response:  # If API fails, use simulation
                        response = self.simulate_response(user_input)
                except:
                    response = self.simulate_response(user_input)
                
                # Display response
                print(response)
                print("\n" + "-" * 50 + "\n")
                
                # Save to history
                chat_history.append((user_input, response))
                
                # Keep only last 10 conversations
                if len(chat_history) > 10:
                    chat_history = chat_history[-10:]
                
            except KeyboardInterrupt:
                print("\n\n🤖 Dwrko-M1.0: Chat interrupted. Goodbye! 👋")
                break
            except Exception as e:
                print(f"\n❌ Error: {str(e)}")
                print("🔄 Let's try again...\n")

def main():
    """Main function to start chat"""
    print("🚀 Initializing Dwrko-M1.0 Chat...")
    
    # Optional: Get HuggingFace token
    hf_token = None
    if len(sys.argv) > 1:
        hf_token = sys.argv[1]
    
    # Create chat instance
    chat = DwrkoChat(hf_token)
    
    # Start chatting
    chat.start_chat()

if __name__ == "__main__":
    main()