Base code
This commit is contained in:
Kunthawat Greethong
241
backend/start_linkedin_service.py
Executable file
241
backend/start_linkedin_service.py
Executable file
@@ -0,0 +1,241 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
LinkedIn Content Generation Service Startup Script
|
||||
|
||||
This script helps users quickly start the LinkedIn content generation service
|
||||
with proper configuration and validation.
|
||||
"""
|
||||
|
||||
import os
|
||||
import sys
|
||||
import subprocess
|
||||
import time
|
||||
from pathlib import Path
|
||||
|
||||
def print_banner():
|
||||
"""Print service banner."""
|
||||
print("""
|
||||
╔═══════════════════════════════════════════════════════════════╗
|
||||
║ ║
|
||||
║ 🚀 LinkedIn Content Generation Service ║
|
||||
║ ║
|
||||
║ FastAPI-based AI content generation for LinkedIn ║
|
||||
║ Migrated from Streamlit to robust backend service ║
|
||||
║ ║
|
||||
╚═══════════════════════════════════════════════════════════════╝
|
||||
""")
|
||||
|
||||
def check_dependencies():
|
||||
"""Check if required dependencies are installed."""
|
||||
print("🔍 Checking dependencies...")
|
||||
|
||||
required_packages = [
|
||||
'fastapi', 'uvicorn', 'pydantic', 'loguru',
|
||||
'sqlalchemy', 'google-genai'
|
||||
]
|
||||
|
||||
missing_packages = []
|
||||
|
||||
for package in required_packages:
|
||||
try:
|
||||
__import__(package.replace('-', '_'))
|
||||
print(f" ✅ {package}")
|
||||
except ImportError:
|
||||
print(f" ❌ {package}")
|
||||
missing_packages.append(package)
|
||||
|
||||
if missing_packages:
|
||||
print(f"\n⚠️ Missing packages: {', '.join(missing_packages)}")
|
||||
print("💡 Install with: pip install -r requirements.txt")
|
||||
return False
|
||||
|
||||
print("✅ All dependencies installed!")
|
||||
return True
|
||||
|
||||
def check_environment():
|
||||
"""Check environment configuration."""
|
||||
print("\n🔍 Checking environment configuration...")
|
||||
|
||||
# Check API keys
|
||||
gemini_key = os.getenv('GEMINI_API_KEY')
|
||||
if not gemini_key:
|
||||
print(" ❌ GEMINI_API_KEY not set")
|
||||
print(" Set with: export GEMINI_API_KEY='your_api_key'")
|
||||
return False
|
||||
elif not gemini_key.startswith('AIza'):
|
||||
print(" ⚠️ GEMINI_API_KEY format appears invalid (should start with 'AIza')")
|
||||
print(" Please verify your API key")
|
||||
return False
|
||||
else:
|
||||
print(" ✅ GEMINI_API_KEY configured")
|
||||
|
||||
# Check database
|
||||
db_url = os.getenv('DATABASE_URL', 'sqlite:///./alwrity.db')
|
||||
print(f" ✅ Database URL: {db_url}")
|
||||
|
||||
# Check log level
|
||||
log_level = os.getenv('LOG_LEVEL', 'INFO')
|
||||
print(f" ✅ Log level: {log_level}")
|
||||
|
||||
return True
|
||||
|
||||
def check_file_structure():
|
||||
"""Check if all required files exist."""
|
||||
print("\n🔍 Checking file structure...")
|
||||
|
||||
required_files = [
|
||||
'models/linkedin_models.py',
|
||||
'services/linkedin_service.py',
|
||||
'routers/linkedin.py',
|
||||
'app.py'
|
||||
]
|
||||
|
||||
missing_files = []
|
||||
|
||||
for file_path in required_files:
|
||||
if os.path.exists(file_path):
|
||||
print(f" ✅ {file_path}")
|
||||
else:
|
||||
print(f" ❌ {file_path}")
|
||||
missing_files.append(file_path)
|
||||
|
||||
if missing_files:
|
||||
print(f"\n⚠️ Missing files: {', '.join(missing_files)}")
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
def validate_service():
|
||||
"""Run structure validation."""
|
||||
print("\n🔍 Validating service structure...")
|
||||
|
||||
try:
|
||||
result = subprocess.run(
|
||||
[sys.executable, 'validate_linkedin_structure.py'],
|
||||
capture_output=True,
|
||||
text=True,
|
||||
timeout=30
|
||||
)
|
||||
|
||||
if result.returncode == 0:
|
||||
print(" ✅ Structure validation passed")
|
||||
return True
|
||||
else:
|
||||
print(" ❌ Structure validation failed")
|
||||
print(result.stdout)
|
||||
print(result.stderr)
|
||||
return False
|
||||
|
||||
except subprocess.TimeoutExpired:
|
||||
print(" ⚠️ Validation timeout")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f" ❌ Validation error: {e}")
|
||||
return False
|
||||
|
||||
def start_server(host="0.0.0.0", port=8000, reload=True):
|
||||
"""Start the FastAPI server."""
|
||||
print(f"\n🚀 Starting LinkedIn Content Generation Service...")
|
||||
print(f" Host: {host}")
|
||||
print(f" Port: {port}")
|
||||
print(f" Reload: {reload}")
|
||||
print(f" URL: http://localhost:{port}")
|
||||
print(f" Docs: http://localhost:{port}/docs")
|
||||
print(f" LinkedIn API: http://localhost:{port}/api/linkedin")
|
||||
|
||||
try:
|
||||
cmd = [
|
||||
sys.executable, '-m', 'uvicorn',
|
||||
'app:app',
|
||||
'--host', host,
|
||||
'--port', str(port)
|
||||
]
|
||||
|
||||
if reload:
|
||||
cmd.append('--reload')
|
||||
|
||||
print(f"\n⚡ Executing: {' '.join(cmd)}")
|
||||
print(" Press Ctrl+C to stop the server")
|
||||
print("=" * 60)
|
||||
|
||||
# Start the server
|
||||
subprocess.run(cmd)
|
||||
|
||||
except KeyboardInterrupt:
|
||||
print("\n\n👋 Server stopped by user")
|
||||
except Exception as e:
|
||||
print(f"\n❌ Error starting server: {e}")
|
||||
|
||||
def print_usage_examples():
|
||||
"""Print usage examples."""
|
||||
print("""
|
||||
📚 Quick Start Examples:
|
||||
|
||||
1. Health Check:
|
||||
curl http://localhost:8000/api/linkedin/health
|
||||
|
||||
2. Generate LinkedIn Post:
|
||||
curl -X POST "http://localhost:8000/api/linkedin/generate-post" \\
|
||||
-H "Content-Type: application/json" \\
|
||||
-d '{
|
||||
"topic": "AI in Healthcare",
|
||||
"industry": "Healthcare",
|
||||
"tone": "professional",
|
||||
"include_hashtags": true,
|
||||
"research_enabled": true
|
||||
}'
|
||||
|
||||
3. Interactive Documentation:
|
||||
Open http://localhost:8000/docs in your browser
|
||||
|
||||
4. Available Endpoints:
|
||||
- POST /api/linkedin/generate-post
|
||||
- POST /api/linkedin/generate-article
|
||||
- POST /api/linkedin/generate-carousel
|
||||
- POST /api/linkedin/generate-video-script
|
||||
- POST /api/linkedin/generate-comment-response
|
||||
- GET /api/linkedin/content-types
|
||||
- GET /api/linkedin/usage-stats
|
||||
""")
|
||||
|
||||
def main():
|
||||
"""Main startup function."""
|
||||
print_banner()
|
||||
|
||||
# Check system requirements
|
||||
checks_passed = True
|
||||
|
||||
if not check_dependencies():
|
||||
checks_passed = False
|
||||
|
||||
if not check_environment():
|
||||
checks_passed = False
|
||||
|
||||
if not check_file_structure():
|
||||
checks_passed = False
|
||||
|
||||
if checks_passed and not validate_service():
|
||||
checks_passed = False
|
||||
|
||||
if not checks_passed:
|
||||
print("\n❌ Pre-flight checks failed!")
|
||||
print(" Please resolve the issues above before starting the service.")
|
||||
sys.exit(1)
|
||||
|
||||
print("\n✅ All pre-flight checks passed!")
|
||||
|
||||
# Show usage examples
|
||||
print_usage_examples()
|
||||
|
||||
# Ask user if they want to start the server
|
||||
try:
|
||||
response = input("\n🚀 Start the LinkedIn Content Generation Service? [Y/n]: ").strip().lower()
|
||||
if response in ['', 'y', 'yes']:
|
||||
start_server()
|
||||
else:
|
||||
print("👋 Service not started. Run 'uvicorn app:app --reload' when ready.")
|
||||
except KeyboardInterrupt:
|
||||
print("\n👋 Goodbye!")
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user