Carlexx commited on
Commit
605d568
·
verified ·
1 Parent(s): da57b7d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -110,7 +110,7 @@ def run_storyboard_generation(num_fragments: int, prompt: str, initial_image_pat
110
  director_prompt = template.format(user_prompt=prompt, num_fragments=int(num_fragments), image_metadata=exif_metadata)
111
 
112
  genai.configure(api_key=GEMINI_API_KEY)
113
- model = genai.GenerativeModel('gemini-pro-vision')
114
  img = Image.open(initial_image_path)
115
  print("Gerando roteiro com análise de visão integrada...")
116
  response = model.generate_content([director_prompt, img])
@@ -131,7 +131,7 @@ def get_dreamo_prompt_for_transition(previous_image_path: str, target_scene_desc
131
  prompt_file = "prompts/img2img_evolution_prompt.txt"
132
  with open(os.path.join(os.path.dirname(__file__), prompt_file), "r", encoding="utf-8") as f: template = f.read()
133
  director_prompt = template.format(target_scene_description=target_scene_description)
134
- model = genai.GenerativeModel('gemini-pro-vision'); img = Image.open(previous_image_path)
135
  response = model.generate_content([director_prompt, "Previous Image:", img])
136
  return response.text.strip().replace("\"", "")
137
 
@@ -193,7 +193,7 @@ def get_dynamic_motion_prompt(user_prompt: str, story_history: str, memory_image
193
  if not GEMINI_API_KEY: raise gr.Error("Chave da API Gemini não configurada!")
194
  try:
195
  genai.configure(api_key=GEMINI_API_KEY)
196
- model = genai.GenerativeModel('gemini-pro-vision')
197
  prompt_file = "prompts/dynamic_motion_prompt.txt"
198
  with open(os.path.join(os.path.dirname(__file__), prompt_file), "r", encoding="utf-8") as f: template = f.read()
199
  cinematographer_prompt = template.format(user_prompt=user_prompt, story_history=story_history, midpoint_scene_description=path_scene_desc, destination_scene_description=dest_scene_desc)
 
110
  director_prompt = template.format(user_prompt=prompt, num_fragments=int(num_fragments), image_metadata=exif_metadata)
111
 
112
  genai.configure(api_key=GEMINI_API_KEY)
113
+ model = genai.GenerativeModel('gemini-2.0-flash')
114
  img = Image.open(initial_image_path)
115
  print("Gerando roteiro com análise de visão integrada...")
116
  response = model.generate_content([director_prompt, img])
 
131
  prompt_file = "prompts/img2img_evolution_prompt.txt"
132
  with open(os.path.join(os.path.dirname(__file__), prompt_file), "r", encoding="utf-8") as f: template = f.read()
133
  director_prompt = template.format(target_scene_description=target_scene_description)
134
+ model = genai.GenerativeModel('gemini-2.0-flash'); img = Image.open(previous_image_path)
135
  response = model.generate_content([director_prompt, "Previous Image:", img])
136
  return response.text.strip().replace("\"", "")
137
 
 
193
  if not GEMINI_API_KEY: raise gr.Error("Chave da API Gemini não configurada!")
194
  try:
195
  genai.configure(api_key=GEMINI_API_KEY)
196
+ model = genai.GenerativeModel('gemini-2.0-flash')
197
  prompt_file = "prompts/dynamic_motion_prompt.txt"
198
  with open(os.path.join(os.path.dirname(__file__), prompt_file), "r", encoding="utf-8") as f: template = f.read()
199
  cinematographer_prompt = template.format(user_prompt=user_prompt, story_history=story_history, midpoint_scene_description=path_scene_desc, destination_scene_description=dest_scene_desc)