giga_generate inside enhance_prompt_func

app.py

@@ -162,10 +162,14 @@ with gr.Blocks() as demo:
   
   # image.upload(fn=resize_image, inputs=image, outputs=image, queue=False)
 
-  #TODO изменить под гигачат
-  def beautify_prompt(prompt: str, retry_times: int = 3) -> str:
-    prompt = giga_generate(prompt, max_attempts=5)
+  # def beautify_prompt(prompt: str, max_attempts: int = 5) -> str:
+  #   prompt = giga_generate(prompt, max_attempts=max_attempts)
 
+  # def enhance_prompt_func(prompt):
+  #   return beautify_prompt(prompt, max_attempts=5)
+
+  def enhance_prompt_func(prompt):
+    return giga_generate(prompt, max_attempts=5)
 
     # if not os.environ.get("OPENAI_API_KEY"):
     #     return prompt
@@ -197,10 +201,9 @@ with gr.Blocks() as demo:
 
 
   generate_btn.click(fn=sample, inputs=[prompt, resolution], outputs=[video], api_name="video")
-  #TODO
-  def enhance_prompt_func(prompt):
-    return beautify_prompt(prompt, retry_times=1)
-  #TODO
+  
+
+  
   enhance_button.click(enhance_prompt_func, inputs=[prompt], outputs=[prompt])
   # gr.Examples(
   #   examples=[