Carlexx commited on
Commit
2155275
Β·
verified Β·
1 Parent(s): b25d7e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -100,7 +100,7 @@ def run_storyboard_generation(num_fragments: int, prompt: str, initial_image_pat
100
  with open(os.path.join(os.path.dirname(__file__), prompt_file), "r", encoding="utf-8") as f: template = f.read()
101
  director_prompt = template.format(user_prompt=prompt, num_fragments=int(num_fragments), image_metadata=exif_metadata)
102
  genai.configure(api_key=GEMINI_API_KEY)
103
- model = genai.GenerativeModel('gemini-1.5-flash'); img = Image.open(initial_image_path)
104
  print("Gerando roteiro com anΓ‘lise de visΓ£o integrada...")
105
  response = model.generate_content([director_prompt, img])
106
  try:
@@ -117,7 +117,7 @@ def get_dreamo_prompt_for_transition(previous_image_path: str, target_scene_desc
117
  prompt_file = "prompts/img2img_evolution_prompt.txt"
118
  with open(os.path.join(os.path.dirname(__file__), prompt_file), "r", encoding="utf-8") as f: template = f.read()
119
  director_prompt = template.format(target_scene_description=target_scene_description)
120
- model = genai.GenerativeModel('gemini-1.5-flash'); img = Image.open(previous_image_path)
121
  response = model.generate_content([director_prompt, "Previous Image:", img])
122
  return response.text.strip().replace("\"", "")
123
 
@@ -171,7 +171,7 @@ def run_keyframe_generation(storyboard, ref_images_tasks, progress=gr.Progress()
171
  def get_initial_motion_prompt(user_prompt: str, start_image_path: str, destination_image_path: str, dest_scene_desc: str):
172
  if not GEMINI_API_KEY: raise gr.Error("Chave da API Gemini nΓ£o configurada!")
173
  try:
174
- genai.configure(api_key=GEMINI_API_KEY); model = genai.GenerativeModel('gemini-1.5-flash'); prompt_file = "prompts/initial_motion_prompt.txt"
175
  with open(os.path.join(os.path.dirname(__file__), prompt_file), "r", encoding="utf-8") as f: template = f.read()
176
  cinematographer_prompt = template.format(user_prompt=user_prompt, destination_scene_description=dest_scene_desc)
177
  start_img, dest_img = Image.open(start_image_path), Image.open(destination_image_path)
@@ -183,7 +183,7 @@ def get_initial_motion_prompt(user_prompt: str, start_image_path: str, destinati
183
  def get_dynamic_motion_prompt(user_prompt, story_history, memory_image_path, path_image_path, destination_image_path, path_scene_desc, dest_scene_desc):
184
  if not GEMINI_API_KEY: raise gr.Error("Chave da API Gemini nΓ£o configurada!")
185
  try:
186
- genai.configure(api_key=GEMINI_API_KEY); model = genai.GenerativeModel('gemini-1.5-flash'); prompt_file = "prompts/dynamic_motion_prompt.txt"
187
  with open(os.path.join(os.path.dirname(__file__), prompt_file), "r", encoding="utf-8") as f: template = f.read()
188
  cinematographer_prompt = template.format(user_prompt=user_prompt, story_history=story_history, midpoint_scene_description=path_scene_desc, destination_scene_description=dest_scene_desc)
189
  mem_img, path_img, dest_img = Image.open(memory_image_path), Image.open(path_image_path), Image.open(destination_image_path)
 
100
  with open(os.path.join(os.path.dirname(__file__), prompt_file), "r", encoding="utf-8") as f: template = f.read()
101
  director_prompt = template.format(user_prompt=prompt, num_fragments=int(num_fragments), image_metadata=exif_metadata)
102
  genai.configure(api_key=GEMINI_API_KEY)
103
+ model = genai.GenerativeModel('gemini-2.5-flash'); img = Image.open(initial_image_path)
104
  print("Gerando roteiro com anΓ‘lise de visΓ£o integrada...")
105
  response = model.generate_content([director_prompt, img])
106
  try:
 
117
  prompt_file = "prompts/img2img_evolution_prompt.txt"
118
  with open(os.path.join(os.path.dirname(__file__), prompt_file), "r", encoding="utf-8") as f: template = f.read()
119
  director_prompt = template.format(target_scene_description=target_scene_description)
120
+ model = genai.GenerativeModel('gemini-2.5-flash'); img = Image.open(previous_image_path)
121
  response = model.generate_content([director_prompt, "Previous Image:", img])
122
  return response.text.strip().replace("\"", "")
123
 
 
171
  def get_initial_motion_prompt(user_prompt: str, start_image_path: str, destination_image_path: str, dest_scene_desc: str):
172
  if not GEMINI_API_KEY: raise gr.Error("Chave da API Gemini nΓ£o configurada!")
173
  try:
174
+ genai.configure(api_key=GEMINI_API_KEY); model = genai.GenerativeModel('gemini-2.5-flash'); prompt_file = "prompts/initial_motion_prompt.txt"
175
  with open(os.path.join(os.path.dirname(__file__), prompt_file), "r", encoding="utf-8") as f: template = f.read()
176
  cinematographer_prompt = template.format(user_prompt=user_prompt, destination_scene_description=dest_scene_desc)
177
  start_img, dest_img = Image.open(start_image_path), Image.open(destination_image_path)
 
183
  def get_dynamic_motion_prompt(user_prompt, story_history, memory_image_path, path_image_path, destination_image_path, path_scene_desc, dest_scene_desc):
184
  if not GEMINI_API_KEY: raise gr.Error("Chave da API Gemini nΓ£o configurada!")
185
  try:
186
+ genai.configure(api_key=GEMINI_API_KEY); model = genai.GenerativeModel('gemini-2.5-flash'); prompt_file = "prompts/dynamic_motion_prompt.txt"
187
  with open(os.path.join(os.path.dirname(__file__), prompt_file), "r", encoding="utf-8") as f: template = f.read()
188
  cinematographer_prompt = template.format(user_prompt=user_prompt, story_history=story_history, midpoint_scene_description=path_scene_desc, destination_scene_description=dest_scene_desc)
189
  mem_img, path_img, dest_img = Image.open(memory_image_path), Image.open(path_image_path), Image.open(destination_image_path)