UPD: Made some UI Changes and QoL Changes

main.py

@@ -143,7 +143,7 @@ def save_image(prompt, image):
         repo_id = "xqt/fashion_model_generator",
         commit_message = f"ADD: prompt {temp_id}.json",
     )
-    gradio.Info(f"Image and prompt saved successfully at https://huggingface.co/datasets/xqt/fashion_model_generator/tree/main/images/{year}/{month}/{day}/{hour}/{minute}/{temp_id}.png", duration = 5)
+    gradio.Info(f"Image and prompt saved successfully <a href = \"https://huggingface.co/datasets/xqt/fashion_model_generator/blob/main/images/{year}/{month}/{day}/{hour}/{minute}/{temp_id}.png\"> here <\a>" , duration = 5)
     os.remove(f"{temp_id}.png")
     os.remove(f"{temp_id}.json")
     return
@@ -159,10 +159,10 @@ with gradio.Blocks(fill_width = False) as app:
     api_token_input = gradio.Textbox(label = "Together AI API Key (key is never stored and it uses free models only)", placeholder = "Enter your Together AI API Key here.", type = "password")
     
     with gradio.Row(equal_height = True):
-        with gradio.Column(scale = 3):
+        with gradio.Column(scale = 4):
             prompt_input = gradio.Textbox(label = "Prompt", placeholder = "Enter your prompt here.")
-        with gradio.Column():
-            prompt_assist = gradio.Button(value = "Assisted Prompt Generation")
+        with gradio.Column(scale = 1):
+            prompt_assist = gradio.Button(value = "Prompt Assist", icon = "assets/wand-magic-sparkles-solid.svg")
     
     image_output = gradio.Image(label="Generated Image")
     
@@ -176,4 +176,3 @@ with gradio.Blocks(fill_width = False) as app:
 
 if __name__ == "__main__":
     app.launch()
-