Update app.py

app.py

@@ -106,6 +106,10 @@ MODEL_CONTEXT_SIZES = {
         "c4ai-aya-expanse-32b": 131072,
     },
     "GLHF API": {
+        "mistralai/Mixtral-8x7B-Instruct-v0.1": 32768,
+        "mistralai/Mixtral-8x7B-Instruct-v0.2": 32768,
+        "NousResearch/Nous-Hermes-2-Solar-10.7B": 32768,
+        "01-ai/Yi-34B-Chat": 32768,
         "mistralai/Mistral-7B-Instruct-v0.3": 32768,
         "microsoft/phi-3-mini-4k-instruct": 4096,
         "microsoft/Phi-3.5-mini-instruct": 4096,
@@ -522,14 +526,14 @@ def send_to_model_impl(prompt, model_selection, hf_model_choice, hf_custom_model
             if not glhf_api_key:
                 return "Error: GLHF API key required", None
                 
-            # Determine the actual model ID to use
-            if glhf_model == "Use HuggingFace Model":
-                model_id = f"hf:{hf_custom_model if hf_model_choice == 'Custom Model' else model_registry.hf_models[hf_model_choice]}"
-            else:
+            # Handle model selection
+            if glhf_model == "Custom Model":
                 model_id = f"hf:{glhf_custom_model}"
+            else:
+                model_id = f"hf:{glhf_model}"
                 
             summary = send_to_glhf(prompt, glhf_api_key, model_id, use_rate_limits)
-        
+            
         else:
             return "Error: Invalid model selection", None
 
@@ -956,9 +960,9 @@ with gr.Blocks(css="""
                         label="🔑 GLHF API Key",
                         type="password"
                     )
-                    glhf_model = gr.Radio(
-                        choices=["Use HuggingFace Model", "Custom Model"],
-                        value="Use HuggingFace Model",
+                    glhf_model = gr.Dropdown(
+                        choices=list(MODEL_CONTEXT_SIZES["GLHF API"].keys()) + ["Custom Model"],
+                        value="mistralai/Mistral-7B-Instruct-v0.3",
                         label="Model Selection"
                     )
                     glhf_custom_model = gr.Textbox(
@@ -1122,18 +1126,17 @@ with gr.Blocks(css="""
                 ]
             elif choice == "GLHF API":
                 # Always show HuggingFace models for GLHF since they're used in both cases
-                model_choices = list(model_registry.hf_models.keys())
+                # model_choices = list(model_registry.hf_models.keys())
                 return [
-                    gr.update(visible=True),   # hf_options - Keep visible for model selection
+                    gr.update(visible=False),  # hf_options - hide completely
                     gr.update(visible=False),  # groq_options
                     gr.update(visible=False),  # openai_options
                     gr.update(visible=False),  # cohere_options
                     gr.update(visible=True),   # glhf_options
                     gr.update(value=ctx_size), # context_size
-                    gr.Dropdown(choices=model_choices, value="Mixtral 7B")  # Use HF models
+                    gr.Dropdown(choices=[])    # not used
                 ]
 
-        
         # Default return for "Clipboard only" or other options
         return [
             gr.update(visible=False),  # hf_options