add basic function beautify_prompt

app.py

@@ -1,3 +1,4 @@
+# исправленная версия (чтобы не потерялась)
 import subprocess
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 # subprocess.run('pip install bitsandbytes', shell=True)
@@ -21,6 +22,8 @@ import uuid
 import random
 from huggingface_hub import hf_hub_download
 
+from src.gigachat import giga_generate
+
 #gradio.helpers.CACHED_FOLDER = '/data/cache'
 
 # pipe = StableVideoDiffusionPipeline.from_pretrained(
@@ -134,7 +137,15 @@ with gr.Blocks() as demo:
             placeholder="Enter your prompt",
             container=False,
         )
-        #TODO ? можно так принять входное значение 
+        #TODO нужен здесь Row или нет, можно сразу с Markdown
+        with gr.Row():
+            #TODO давать ссылку на гигачат?
+            #TODO заменить текст)
+            gr.Markdown(
+                "✨Upon pressing the enhanced prompt button, we will use [GigaChat Model](https://github.com/THUDM/GLM-4) to polish the prompt and overwrite the original one."
+            )
+            enhance_button = gr.Button("✨ Enhance Prompt(Optional)")
+
         resolution = gr.Dropdown(
             label="Video resolution",
             choices=["1:1", "9:16", "16:9", "1:2", "2:1"],
@@ -148,10 +159,49 @@ with gr.Blocks() as demo:
   #     randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
   #     motion_bucket_id = gr.Slider(label="Motion bucket id", info="Controls how much motion to add/remove from the image", value=127, minimum=1, maximum=255)
   #     fps_id = gr.Slider(label="Frames per second", info="The length of your video in seconds will be 25/fps", value=6, minimum=5, maximum=30)
-
+  
   # image.upload(fn=resize_image, inputs=image, outputs=image, queue=False)
-  #TODO функция так принимает аргументы?
+
+  #TODO изменить под гигачат
+  def beautify_prompt(prompt: str, retry_times: int = 3) -> str:
+    prompt = giga_generate(prompt)
+
+
+    # if not os.environ.get("OPENAI_API_KEY"):
+    #     return prompt
+    # client = OpenAI()
+    # text = prompt.strip()
+
+    # for i in range(retry_times):
+    #     response = client.chat.completions.create(
+    #         messages=[
+    #             {"role": "system", "content": sys_prompt},
+    #             {
+    #                 "role": "user",
+    #                 "content": 'Create an imaginative video descriptive caption or modify an earlier caption for the user input : "a girl is on the beach"',
+    #             },
+    #             {
+    #                 "role": "assistant",
+    #                 "content": "A radiant woman stands on a deserted beach, arms outstretched, wearing a beige trench coat, white blouse, light blue jeans, and chic boots, against a backdrop of soft sky and sea. Moments later, she is seen mid-twirl, arms exuberant, with the lighting suggesting dawn or dusk. Then, she runs along the beach, her attire complemented by an off-white scarf and black ankle boots, the tranquil sea behind her. Finally, she holds a paper airplane, her pose reflecting joy and freedom, with the ocean's gentle waves and the sky's soft pastel hues enhancing the serene ambiance.",
+    #             }
+    #         ],
+    #         model="glm-4-plus",
+    #         temperature=0.01,
+    #         top_p=0.7,
+    #         stream=False,
+    #         max_tokens=200,
+    #     )
+    #     if response.choices:
+    #         return response.choices[0].message.content
+    return prompt
+
+
   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=[
   #       "images/blink_meme.png",