Spaces:
Running
Running
File size: 3,917 Bytes
d74863e 6405808 d74863e 97fbe9e d74863e 571ece6 97fbe9e 571ece6 f89dbd6 97fbe9e f89dbd6 97fbe9e d74863e 97fbe9e d74863e 97fbe9e d74863e 97fbe9e d74863e 97fbe9e d74863e e768616 d74863e 97fbe9e d74863e f89dbd6 d74863e 97fbe9e d74863e 97fbe9e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 | """
Main entry point for YouTube Study Notes AI.
Provides CLI interface and server startup.
"""
import os
import sys
import subprocess
import argparse
from pathlib import Path
from src.utils.logger import setup_logger
from src.utils.config import settings
logger = setup_logger(__name__)
def configure_proxy():
proxy_url = os.environ.get("PROXY_URL", "").strip()
if proxy_url:
os.environ["HTTP_PROXY"] = proxy_url
os.environ["HTTPS_PROXY"] = proxy_url
os.environ["ALL_PROXY"] = proxy_url
os.environ["http_proxy"] = proxy_url
os.environ["https_proxy"] = proxy_url
os.environ["all_proxy"] = proxy_url
logger.info("✅ Global proxy configured from PROXY_URL")
def check_youtube_connectivity():
import urllib.request
import urllib.error
try:
urllib.request.urlopen("https://www.youtube.com", timeout=5)
logger.info("✅ YouTube is reachable")
except (urllib.error.URLError, Exception):
logger.warning(
"WARNING: YouTube is not reachable from this host. "
"Set PROXY_URL or RAPIDAPI_KEY env vars to enable YouTube features."
)
def check_environment():
"""Log key dependency versions to confirm runtime environment."""
configure_proxy()
check_youtube_connectivity()
# Check ffmpeg (still used by audio processing utilities)
try:
ffmpeg_out = subprocess.check_output(
["ffmpeg", "-version"], stderr=subprocess.STDOUT
).decode().splitlines()[0]
logger.info(f"✅ ffmpeg available: {ffmpeg_out}")
except (subprocess.CalledProcessError, FileNotFoundError):
logger.warning("❌ ffmpeg NOT found — audio extraction will fail")
# Verify Supadata API key is configured
supadata_key = os.environ.get("SUPADATA_API_KEY", "").strip()
if supadata_key:
logger.info("✅ SUPADATA_API_KEY is set")
else:
logger.warning("❌ SUPADATA_API_KEY is NOT set — transcript extraction will fail")
def run_server():
"""Start the FastAPI server with CORS enabled for Flutter Web."""
import uvicorn
from fastapi.middleware.cors import CORSMiddleware
from src.api.main import app
check_environment()
logger.info("Configuring CORS for Flutter Web...")
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
logger.info("Starting YouTube Study Notes AI server...")
logger.info(f"Server will be available at http://{settings.api_host}:{settings.api_port}")
logger.info(f"API Documentation: http://{settings.api_host}:{settings.api_port}/docs")
uvicorn.run(app, host=settings.api_host, port=settings.api_port, log_level="info")
def run_cli(youtube_url: str, output_file: str = None):
from src.transcription.downloader import YouTubeDownloader
check_environment()
downloader = YouTubeDownloader()
transcript = downloader.get_transcript(youtube_url)
print(transcript)
def main():
"""Main entry point with argument parsing."""
parser = argparse.ArgumentParser(
description="YouTube Study Notes AI - Generate structured notes from educational videos"
)
parser.add_argument(
"mode",
choices=["server", "cli"],
help="Run mode: server (API + web UI) or cli (direct processing)",
)
parser.add_argument("--url", type=str, help="YouTube video URL (required for cli mode)")
parser.add_argument("--output", type=str, help="Output file path (optional for cli mode)")
args = parser.parse_args()
if args.mode == "server":
run_server()
elif args.mode == "cli":
if not args.url:
print("Error: --url is required for cli mode")
sys.exit(1)
run_cli(args.url, args.output)
if __name__ == "__main__":
main() |