UI Improvements

app.py

@@ -155,7 +155,7 @@ def ui_main():
 
         with gr.Row():
             llm_model = gr.Textbox(
-                label="LLM Model", placeholder="model_provider/model_name", value="huggingface/meta-llama/Meta-Llama-3.1-8B-Instruct"
+                label="LLM Model", placeholder="model_provider/model_name", value="huggingface/mistralai/Mistral-7B-Instruct-v0.3"
             )
             temperature = gr.Slider(
                 label="Temperature", minimum=0.0, maximum=1.0, step=0.1, value=0.5
@@ -190,21 +190,24 @@ def ui_main():
             num_entries = gr.Number(label="Number of Entries", value=1000)
 
         with gr.Row():
-            hf_token = gr.Textbox(
-                label="Hugging Face Token",
-                placeholder="Your HF Token",
-                type="password",
-                value="hf_1234566789912345677889",
+            hf_token = gr.LoginButton(
+                label="Hugging Face Login",
+                value="Login with Hugging Face",
             )
             hf_repo_name = gr.Textbox(
                 label="Hugging Face Repo Name",
                 placeholder="organization_or_user_name/dataset_name",
-                value="shekswess/synthgenai-dataset",
+                value="Shekswess/synthgenai-dataset",
             )
             llm_env_vars = gr.Textbox(
                 label="LLM Environment Variables",
                 placeholder="Comma-separated environment variables (e.g., KEY1=VALUE1, KEY2=VALUE2)",
-                value="HUGGINGFACE_API_KEY=hf_1234566789912345677889",
+                value="HUGGINGFACE_API_KEY=hf_1234566789912345677889, OPENAI_API_KEY=sk-1234566789912345677889",
+            )
+            gr.Markdown(
+                """
+                For more information on which LLMs are allowed and how they can be used, please refer to the [documentation](https://shekswess.github.io/synthgenai/llm_providers/).
+                """
             )
 
         generate_button = gr.Button("Generate Dataset")