back to beautify_prompt

app.py

@@ -162,14 +162,14 @@ with gr.Blocks() as demo:
   
   # image.upload(fn=resize_image, inputs=image, outputs=image, queue=False)
 
-  # 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 beautify_prompt(prompt: str, max_attempts: int = 5) -> str:
+    prompt = giga_generate(prompt, max_attempts=max_attempts)
 
   def enhance_prompt_func(prompt):
-    return giga_generate(prompt, max_attempts=5)
+    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