trial

app.py

@@ -6,7 +6,6 @@ title = """<h1 align="center">Random Prompt Generator</h1>
 <p><center>
 <a href="https://x.com/gokayfem" target="_blank">[X gokaygokay]</a>
 <a href="https://github.com/gokayfem" target="_blank">[Github gokayfem]</a>
-<a href="https://github.com/dagthomas/comfyui_dagthomas" target="_blank">[comfyui_dagthomas]</a>
 <p align="center">Generate random prompts using powerful LLMs from Hugging Face, Groq, and SambaNova.</p>
 </center></p>
 """
@@ -67,21 +66,42 @@ def create_interface():
                 generate_button = gr.Button("Generate Random Prompt with LLM")
                 text_output = gr.Textbox(label="LLM Generated Text", lines=10, show_copy_button=True)
 
-        # Initialize Models based on provider
+        # Updated Models based on provider
         def update_model_choices(provider):
             provider_models = {
-                "Hugging Face": ["meta-llama/Meta-Llama-3.1-70B-Instruct", "another-model-hf"],
-                "Groq": ["llama-3.1-70b-versatile", "mixtral-8x7b-32768", "gemma2-9b-it"],
-                "SambaNova": ["Meta-Llama-3.1-70B-Instruct", "Meta-Llama-3.1-405B-Instruct", "Meta-Llama-3.1-8B-Instruct"],
+                "Hugging Face": [
+                    "Qwen/Qwen2.5-72B-Instruct",
+                    "meta-llama/Meta-Llama-3.1-70B-Instruct",
+                    "mistralai/Mixtral-8x7B-Instruct-v0.1",
+                    "mistralai/Mistral-7B-Instruct-v0.3"
+                ],
+                "Groq": [
+                    "llama-3.1-70b-versatile",
+                    "mixtral-8x7b-32768",
+                    "gemma2-9b-it"
+                ],
+                "SambaNova": [
+                    "Meta-Llama-3.1-70B-Instruct",
+                    "Meta-Llama-3.1-405B-Instruct",
+                    "Meta-Llama-3.1-8B-Instruct"
+                ],
             }
             models = provider_models.get(provider, [])
-            return gr.Dropdown.update(choices=models, value=models[0] if models else "")
+            return gr.Dropdown(choices=models, value=models[0] if models else "")
         
         def update_api_key_visibility(provider):
             return gr.update(visible=False)  # No API key required for selected providers
 
-        llm_provider.change(update_model_choices, inputs=[llm_provider], outputs=[model])
-        llm_provider.change(update_api_key_visibility, inputs=[llm_provider], outputs=[api_key])
+        llm_provider.change(
+            update_model_choices, 
+            inputs=[llm_provider], 
+            outputs=[model]
+        )
+        llm_provider.change(
+            update_api_key_visibility, 
+            inputs=[llm_provider], 
+            outputs=[api_key]
+        )
 
         # **Unified Function to Generate Prompt and Text**
         def generate_random_prompt_with_llm(custom_input, prompt_type, long_talk, compress, compression_level, custom_base_prompt, provider, api_key, model_selected):