replicate keyを複数設定できるように修正

.env.example

@@ -1,3 +1,3 @@
-REPLICATE_API_TOKEN=
+REPLICATE_API_TOKEN_LIST=key1,key2
 OPENAI_API_KEY=
 ENV=PRODUCTION

app.py

@@ -17,13 +17,17 @@ from PIL import Image as PIL_Image
 
 from jinja2 import Template
 
-load_dotenv()
-openai.api_key = os.getenv('OPENAI_API_KEY')
-REPLICATE_API_TOKEN = os.getenv("REPLICATE_API_TOKEN")
 ENV = os.getenv("ENV")
 MODEL = "gpt-3.5-turbo"
 # MODEL = "gpt-4"
 
+load_dotenv()
+openai.api_key = os.getenv('OPENAI_API_KEY')
+# REPLICATE_API_TOKEN = os.getenv("REPLICATE_API_TOKEN")
+# REPLICATE_API_TOKEN_LISTをロードし、カンマで分割してリストに変換
+REPLICATE_API_TOKEN_LIST = os.getenv("REPLICATE_API_TOKEN_LIST").split(',')
+REPLICATE_API_TOKEN_INDEX = 0 # トークンのインデックスを初期化
+
 if ENV == "PRODUCTION":
     import replicate
 else:
@@ -41,6 +45,14 @@ class Video:
         self.file_path = f"assets/thread_{index}_video.mp4"
 
     def run_replicate(self):
+        global REPLICATE_API_TOKEN_INDEX
+        start_time = time.time()
+
+        # 現在のトークンを取得し、次のトークンにインデックスを更新
+        token = REPLICATE_API_TOKEN_LIST[REPLICATE_API_TOKEN_INDEX]
+        REPLICATE_API_TOKEN_INDEX = (REPLICATE_API_TOKEN_INDEX + 1) % len(REPLICATE_API_TOKEN_LIST)
+        os.environ['REPLICATE_API_TOKEN'] = token
+        
         start_time = time.time()
 
         self.output_url = replicate.run(

template.md

@@ -1,5 +1,9 @@
 # Movie Title: {{ args.title }}
 
+## Story
+
+{{ args.story }}
+
 ## Generation Time
 
 {{ generation_time }}