rdune71's picture
Update app.py imports and send button logic, enhance config detection for HF Spaces, improve HF token validation
86b116d
raw
history blame
1.12 kB
import os
from dotenv import load_dotenv
class Config:
def __init__(self):
load_dotenv()
# Detect if running on HF Spaces
self.is_hf_space = bool(os.getenv("SPACE_ID"))
self.hf_token = os.getenv("HF_TOKEN")
self.hf_api_url = os.getenv("HF_API_ENDPOINT_URL", "https://api-inference.huggingface.co/v1/")
self.use_fallback = os.getenv("USE_FALLBACK", "true").lower() == "true"
# Redis configuration (optional for HF)
self.redis_host = os.getenv("REDIS_HOST", "localhost")
self.redis_port = int(os.getenv("REDIS_PORT", "6379"))
self.redis_username = os.getenv("REDIS_USERNAME", "")
self.redis_password = os.getenv("REDIS_PASSWORD", "")
self.redis_retries = int(os.getenv("REDIS_RETRIES", "3"))
self.redis_retry_delay = int(os.getenv("REDIS_RETRY_DELAY", "1"))
# Local model configuration
self.local_model_name = os.getenv("LOCAL_MODEL_NAME", "mistral:latest")
self.ollama_host = os.getenv("OLLAMA_HOST", "")
# Global config instance
config = Config()