Spaces:

Mistral-AI-Game-Jam
/

ai-interactive-comic-book

Running

ai-interactive-comic-book / server /core /setup.py

tfrere HF Staff

update

d9ce58c over 1 year ago

924 Bytes

from services.mistral_client import MistralClient
from core.generators.universe_generator import UniverseGenerator
from core.story_generator import StoryGenerator

# Initialize generators with None - they will be set up when needed
universe_generator = None

def setup_game(api_key: str, model_name: str = "mistral-small"):
    """Setup all game components with the provided API key."""
    global universe_generator
    
    mistral_client = MistralClient(api_key=api_key, model_name=model_name)
    universe_generator = UniverseGenerator(mistral_client)
    # StoryGenerator is a singleton, so we just need to initialize it
    StoryGenerator(api_key=api_key, model_name=model_name)

def get_universe_generator() -> UniverseGenerator:
    """Get the universe generator instance."""
    if universe_generator is None:
        raise RuntimeError("Game not initialized. Call setup_game first.")
    return universe_generator 

	from services.mistral_client import MistralClient
	from core.generators.universe_generator import UniverseGenerator
	from core.story_generator import StoryGenerator

	# Initialize generators with None - they will be set up when needed
	universe_generator = None

	def setup_game(api_key: str, model_name: str = "mistral-small"):
	"""Setup all game components with the provided API key."""
	global universe_generator

	mistral_client = MistralClient(api_key=api_key, model_name=model_name)
	universe_generator = UniverseGenerator(mistral_client)
	# StoryGenerator is a singleton, so we just need to initialize it
	StoryGenerator(api_key=api_key, model_name=model_name)

	def get_universe_generator() -> UniverseGenerator:
	"""Get the universe generator instance."""
	if universe_generator is None:
	raise RuntimeError("Game not initialized. Call setup_game first.")
	return universe_generator