telegram.video.summary.bot/src/config.py

46 lines
1.7 KiB
Python

import os
from dotenv import load_dotenv
import logging
logging.basicConfig(
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
level=logging.INFO
)
logger = logging.getLogger(__name__)
# Załaduj zmienne środowiskowe z pliku .env
dotenv_path = os.path.join(os.path.dirname(__file__), '..', '.env')
if os.path.exists(dotenv_path):
load_dotenv(dotenv_path)
logger.info(f"Załadowano zmienne środowiskowe z: {dotenv_path}")
else:
logger.warning(f"Nie znaleziono pliku .env w {dotenv_path}, korzystanie ze zmiennych systemowych.")
TELEGRAM_BOT_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
DATABASE_URL = os.getenv("DATABASE_URL")
YOUTUBE_TRANSCRIPT_API_TOKEN = os.getenv("YOUTUBE_TRANSCRIPT_API_TOKEN")
if not TELEGRAM_BOT_TOKEN:
logger.error("Nie znaleziono TELEGRAM_BOT_TOKEN w zmiennych środowiskowych.")
raise ValueError("Brak TELEGRAM_BOT_TOKEN")
if not OPENAI_API_KEY:
logger.error("Nie znaleziono OPENAI_API_KEY w zmiennych środowiskowych.")
raise ValueError("Brak OPENAI_API_KEY")
if not DATABASE_URL:
logger.error("Nie znaleziono DATABASE_URL w zmiennych środowiskowych.")
raise ValueError("Brak DATABASE_URL")
if not YOUTUBE_TRANSCRIPT_API_TOKEN:
logger.warning("Nie znaleziono YOUTUBE_TRANSCRIPT_API_TOKEN w zmiennych środowiskowych.")
logger.warning("Funkcje transkrypcji mogą nie działać poprawnie.")
# Inne ustawienia
TRANSCRIPT_LANGUAGES = ['pl', 'en'] # Priorytet języków transkrypcji
SUMMARY_PROMPT = """Streść poniższy transkrypt filmu z YouTube w zwięzły sposób w języku polskim.
Skup się na głównych tematach i wnioskach.
Transkrypt:
{transcript}"""