Upload backend/chuyenapichatbot.py with huggingface_hub

backend/chuyenapichatbot.py

@@ -0,0 +1,172 @@
+#!/usr/bin/env python3
+"""
+Script để chuyển đổi API chatbot giữa các mode:
+- api: Gọi HF Spaces API (mặc định)
+- local: Dùng local model
+- llama_cpp: Dùng llama.cpp model
+- openai: Dùng OpenAI API
+- anthropic: Dùng Anthropic Claude API
+- ollama: Dùng Ollama local
+"""
+import os
+import sys
+from pathlib import Path
+
+# Colors for terminal output
+class Colors:
+    GREEN = '\033[92m'
+    YELLOW = '\033[93m'
+    RED = '\033[91m'
+    BLUE = '\033[94m'
+    CYAN = '\033[96m'
+    RESET = '\033[0m'
+    BOLD = '\033[1m'
+
+def print_colored(text: str, color: str = Colors.RESET):
+    """Print colored text."""
+    print(f"{color}{text}{Colors.RESET}")
+
+def get_env_file():
+    """Get .env file path."""
+    backend_dir = Path(__file__).parent
+    env_file = backend_dir / ".env"
+    return env_file
+
+def read_env_file():
+    """Read .env file and return as dict."""
+    env_file = get_env_file()
+    env_vars = {}
+    
+    if env_file.exists():
+        with open(env_file, 'r', encoding='utf-8') as f:
+            for line in f:
+                line = line.strip()
+                if line and not line.startswith('#') and '=' in line:
+                    key, value = line.split('=', 1)
+                    env_vars[key.strip()] = value.strip()
+    
+    return env_vars
+
+def write_env_file(env_vars: dict):
+    """Write .env file from dict."""
+    env_file = get_env_file()
+    
+    # Read existing file to preserve comments and order
+    existing_lines = []
+    if env_file.exists():
+        with open(env_file, 'r', encoding='utf-8') as f:
+            existing_lines = f.readlines()
+    
+    # Update or add LLM_PROVIDER and HF_API_BASE_URL
+    new_lines = []
+    llm_provider_set = False
+    hf_api_base_set = False
+    
+    for line in existing_lines:
+        stripped = line.strip()
+        if stripped.startswith('LLM_PROVIDER='):
+            new_lines.append(f"LLM_PROVIDER={env_vars.get('LLM_PROVIDER', 'api')}\n")
+            llm_provider_set = True
+        elif stripped.startswith('HF_API_BASE_URL='):
+            new_lines.append(f"HF_API_BASE_URL={env_vars.get('HF_API_BASE_URL', 'https://davidtran999-hue-portal-backend.hf.space/api')}\n")
+            hf_api_base_set = True
+        else:
+            new_lines.append(line)
+    
+    # Add if not found
+    if not llm_provider_set:
+        new_lines.append(f"LLM_PROVIDER={env_vars.get('LLM_PROVIDER', 'api')}\n")
+    if not hf_api_base_set and env_vars.get('LLM_PROVIDER') == 'api':
+        new_lines.append(f"HF_API_BASE_URL={env_vars.get('HF_API_BASE_URL', 'https://davidtran999-hue-portal-backend.hf.space/api')}\n")
+    
+    with open(env_file, 'w', encoding='utf-8') as f:
+        f.writelines(new_lines)
+
+def show_current():
+    """Show current LLM provider configuration."""
+    env_vars = read_env_file()
+    provider = env_vars.get('LLM_PROVIDER', 'api')
+    api_url = env_vars.get('HF_API_BASE_URL', 'https://davidtran999-hue-portal-backend.hf.space/api')
+    
+    print_colored("\n📊 Cấu hình hiện tại:", Colors.BOLD)
+    print_colored(f"   Provider: {provider}", Colors.CYAN)
+    if provider == 'api':
+        print_colored(f"   API URL: {api_url}", Colors.CYAN)
+    print()
+
+def switch_provider(provider: str, api_url: str = None):
+    """Switch LLM provider."""
+    env_vars = read_env_file()
+    
+    valid_providers = ['api', 'local', 'llama_cpp', 'openai', 'anthropic', 'ollama', 'huggingface']
+    
+    if provider not in valid_providers:
+        print_colored(f"❌ Provider không hợp lệ: {provider}", Colors.RED)
+        print_colored(f"   Các provider hợp lệ: {', '.join(valid_providers)}", Colors.YELLOW)
+        return False
+    
+    env_vars['LLM_PROVIDER'] = provider
+    
+    if provider == 'api':
+        if api_url:
+            env_vars['HF_API_BASE_URL'] = api_url
+        elif 'HF_API_BASE_URL' not in env_vars:
+            env_vars['HF_API_BASE_URL'] = 'https://davidtran999-hue-portal-backend.hf.space/api'
+        print_colored(f"✅ Đã chuyển sang API mode (HF Spaces)", Colors.GREEN)
+        print_colored(f"   API URL: {env_vars['HF_API_BASE_URL']}", Colors.CYAN)
+    elif provider == 'local':
+        print_colored(f"✅ Đã chuyển sang Local model mode", Colors.GREEN)
+    elif provider == 'llama_cpp':
+        print_colored(f"✅ Đã chuyển sang llama.cpp mode", Colors.GREEN)
+    elif provider == 'openai':
+        print_colored(f"✅ Đã chuyển sang OpenAI mode", Colors.GREEN)
+    elif provider == 'anthropic':
+        print_colored(f"✅ Đã chuyển sang Anthropic Claude mode", Colors.GREEN)
+    elif provider == 'ollama':
+        print_colored(f"✅ Đã chuyển sang Ollama mode", Colors.GREEN)
+    elif provider == 'huggingface':
+        print_colored(f"✅ Đã chuyển sang Hugging Face Inference API mode", Colors.GREEN)
+    
+    write_env_file(env_vars)
+    print_colored("\n⚠️  Cần restart backend server để áp dụng thay đổi!", Colors.YELLOW)
+    return True
+
+def main():
+    """Main function."""
+    if len(sys.argv) < 2:
+        print_colored("\n🔧 Script chuyển đổi API Chatbot", Colors.BOLD)
+        print_colored("=" * 50, Colors.CYAN)
+        print_colored("\nCách sử dụng:", Colors.BOLD)
+        print_colored("  python chuyenapichatbot.py <provider> [api_url]", Colors.YELLOW)
+        print_colored("\nCác provider:", Colors.BOLD)
+        print_colored("  api         - Gọi HF Spaces API (mặc định)", Colors.GREEN)
+        print_colored("  local       - Dùng local model", Colors.CYAN)
+        print_colored("  llama_cpp   - Dùng llama.cpp model", Colors.CYAN)
+        print_colored("  openai      - Dùng OpenAI API", Colors.CYAN)
+        print_colored("  anthropic   - Dùng Anthropic Claude API", Colors.CYAN)
+        print_colored("  ollama      - Dùng Ollama local", Colors.CYAN)
+        print_colored("  huggingface - Dùng Hugging Face Inference API", Colors.CYAN)
+        print_colored("\nVí dụ:", Colors.BOLD)
+        print_colored("  python chuyenapichatbot.py api", Colors.YELLOW)
+        print_colored("  python chuyenapichatbot.py api https://custom-api.hf.space/api", Colors.YELLOW)
+        print_colored("  python chuyenapichatbot.py local", Colors.YELLOW)
+        print_colored("  python chuyenapichatbot.py current  # Xem cấu hình hiện tại", Colors.YELLOW)
+        print()
+        show_current()
+        return
+    
+    command = sys.argv[1].lower()
+    
+    if command == 'current' or command == 'show':
+        show_current()
+        return
+    
+    provider = command
+    api_url = sys.argv[2] if len(sys.argv) > 2 else None
+    
+    switch_provider(provider, api_url)
+
+if __name__ == "__main__":
+    main()
+
+