mirror of
https://github.com/wassname/Clover-Edition.git
synced 2026-09-25 13:10:22 +08:00
46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
import os
|
|
from story.utils import *
|
|
from google.cloud import storage
|
|
import json
|
|
from story.story_manager import *
|
|
# from generator.web.web_generator import *
|
|
# from generator.ctrl.ctrl_generator import *
|
|
from generator.simple.simple_generator import *
|
|
import tensorflow as tf
|
|
import textwrap
|
|
import sys
|
|
CRED_FILE = "./AI-Adventure-2bb65e3a4e2f.json"
|
|
|
|
|
|
def play_unconstrained():
|
|
generator = SimpleGenerator()
|
|
prompt = get_story_start("zombies")
|
|
context = get_context("zombies")
|
|
story_manager = UnconstrainedStoryManager(generator)
|
|
story_manager.start_new_story(prompt, context=context)
|
|
|
|
print("\n")
|
|
print(context)
|
|
print(str(story_manager.story))
|
|
while True:
|
|
action = input("> ")
|
|
|
|
if action != "":
|
|
action = action.strip()
|
|
|
|
action = action[0].upper() + action[1:]
|
|
|
|
action = "\n> " + action + ".\n"
|
|
action = remove_profanity(action)
|
|
#action = first_to_second_person(action)
|
|
|
|
result = story_manager.act(action)
|
|
print(result)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
play_unconstrained()
|
|
|
|
|
|
|