Update run.py

run.py

@@ -5,8 +5,8 @@ Provides CLI interface and server startup.
 
 import sys
 import argparse
+import os
 from pathlib import Path
-import os  # Added to read environment variables
 
 # Import necessary modules for server and middleware
 from src.utils.logger import setup_logger
@@ -32,8 +32,11 @@ def run_server():
         allow_headers=["*"],  # Allows all headers
     )
 
-    # Use port 7860 for Hugging Face, or default to 8000 locally
-    # This ensures compatibility with Hugging Face Spaces
+    # ---------------------------------------------------------
+    # TWEAK FOR HUGGING FACE SPACES
+    # Hugging Face sets the 'PORT' environment variable to 7860.
+    # We use it if available, otherwise fallback to 7860 default.
+    # ---------------------------------------------------------
     port = int(os.environ.get("PORT", 7860))
 
     logger.info("Starting YouTube Study Notes AI server...")
@@ -41,17 +44,13 @@ def run_server():
     logger.info(f"API Documentation: http://0.0.0.0:{port}/docs")
 
     # Run the server using the app object directly
-    # IMPORTANT: host must be "0.0.0.0" and port must be 7860 for Hugging Face
+    # Host must be "0.0.0.0" to be accessible externally
     uvicorn.run(app, host="0.0.0.0", port=port, log_level="info")
 
 
 def run_cli(youtube_url: str, output_file: str = None):
     """
     Run note generation from command line.
-
-    Args:
-        youtube_url: YouTube video URL
-        output_file: Optional output file path
     """
     from src.audio.downloader import YouTubeDownloader
     from src.transcription.whisper_transcriber import WhisperTranscriber