Update app.py

app.py

@@ -379,7 +379,9 @@ def build_prompts(snippets: List[str], prompt_instruction: str, custom_prompt: O
     return "\n\n".join(prompts)
 
 def send_to_model(prompt, model_selection, hf_model_choice, hf_custom_model, hf_api_key,
-                 groq_model_choice, groq_api_key, openai_api_key, openai_model_choice):
+                 groq_model_choice, groq_api_key, openai_api_key, openai_model_choice,
+                 cohere_api_key=None, cohere_model=None, glhf_api_key=None, glhf_model=None, 
+                 glhf_custom_model=None):
     """Wrapper function for send_to_model_impl with comprehensive error handling."""
     
     logging.info("send to model starting...")
@@ -391,7 +393,7 @@ def send_to_model(prompt, model_selection, hf_model_choice, hf_custom_model, hf_
         logging.info("sending to model preparation.")
         
         # Basic input validation
-        valid_selections = ["Clipboard only", "HuggingFace Inference", "Groq API", "OpenAI ChatGPT", "Cohere API"]
+        valid_selections = ["Clipboard only", "HuggingFace Inference", "Groq API", "OpenAI ChatGPT", "Cohere API", "GLHF API"]
         if model_selection not in valid_selections:
             return "Error: Invalid model selection", None
             
@@ -405,11 +407,11 @@ def send_to_model(prompt, model_selection, hf_model_choice, hf_custom_model, hf_
                 return "Error: GLHF API key required", None
             if glhf_model == "Use HuggingFace Model":
                 model_id = hf_custom_model if hf_model_choice == "Custom Model" else model_registry.hf_models[hf_model_choice]
-                summary = send_to_glhf(prompt, True, model_id, "", glhf_api_key)
             else:
                 if not glhf_custom_model.strip():
                     return "Error: Custom model ID required", None
-                summary = send_to_glhf(prompt, False, "", glhf_custom_model.strip(), glhf_api_key)
+                model_id = glhf_custom_model.strip()
+            summary = send_to_glhf(prompt, glhf_model == "Use HuggingFace Model", model_id, glhf_custom_model, glhf_api_key)
             
         # Call implementation with error handling
         try:
@@ -423,7 +425,12 @@ def send_to_model(prompt, model_selection, hf_model_choice, hf_custom_model, hf_
                 groq_model_choice=groq_model_choice,
                 groq_api_key=groq_api_key,
                 openai_api_key=openai_api_key,
-                openai_model_choice=openai_model_choice
+                openai_model_choice=openai_model_choice,
+                cohere_api_key=cohere_api_key,
+                cohere_model=cohere_model,
+                glhf_api_key=glhf_api_key,
+                glhf_model=glhf_model,
+                glhf_custom_model=glhf_custom_model
             )
             logging.info("summary received:", summary)
             
@@ -445,8 +452,6 @@ def send_to_model(prompt, model_selection, hf_model_choice, hf_custom_model, hf_
             error_msg = "Unknown error occurred"
         logging.error(f"Error in send_to_model: {error_msg}")
         return f"Error: {error_msg}", None
-    finally:
-        logging.info("send to model completed.")
 
 def send_to_model_impl(prompt, model_selection, hf_model_choice, hf_custom_model, hf_api_key,
                       groq_model_choice, groq_api_key, openai_api_key, openai_model_choice,
@@ -1020,11 +1025,11 @@ with gr.Blocks(css="""
 
     def toggle_model_options(choice):
         return (
-            gr.update(visible=choice == "HuggingFace Inference"),
-            gr.update(visible=choice == "Groq API"),
-            gr.update(visible=choice == "OpenAI ChatGPT"),
-            gr.update(visible=choice == "Cohere API"),
-            gr.update(visible=choice == "GLHF API")
+            gr.update(visible=choice == "HuggingFace Inference"),  # hf_options
+            gr.update(visible=choice == "Groq API"),               # groq_options
+            gr.update(visible=choice == "OpenAI ChatGPT"),         # openai_options
+            gr.update(visible=choice == "Cohere API"),            # cohere_options
+            gr.update(visible=choice == "GLHF API")               # glhf_options
         )
 
     def refresh_groq_models_list():
@@ -1245,8 +1250,9 @@ with gr.Blocks(css="""
     )
 
     # Model processing
+    # Model processing button
     send_to_model_btn.click(
-        send_to_model,
+        fn=send_to_model,
         inputs=[
             generated_prompt,
             model_choice,