Update app.py

app.py

@@ -212,35 +212,54 @@ def send_to_model(*args, **kwargs):  # Correct the outputs here
             return f"Error: {str(e)}", None  # Return error message and None for the file
 
 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): # Added openai_model_choice
-        try:
-            if model_selection == "Clipboard only":
-                return "Use copy/paste for processing", None
-
-            if model_selection == "HuggingFace Inference":
-                # ... (Existing logic)
-
-            elif model_selection == "Groq API":
-                # ... (Existing logic)
-
-            elif model_selection == "OpenAI ChatGPT":
-                if not openai_api_key:
-                    return "Error: OpenAI API key required", None
-                summary = send_to_openai(prompt, openai_api_key, model=openai_model_choice) # Use openai_model_choice
+                    groq_model_choice, groq_api_key, openai_api_key, openai_model_choice):
+    try:
+        if model_selection == "Clipboard only":
+            return "Use copy/paste for processing", None
+
+        elif model_selection == "HuggingFace Inference":
+            if not hf_api_key:
+                return "Error: HuggingFace API key required", None
+            if not hf_model_choice:
+                return "Error: Select a HuggingFace model", None
+            model_id = hf_custom_model if hf_model_choice == "Custom Model" else model_registry.hf_models[hf_model_choice]
+            try:
+                summary = send_to_hf_inference(prompt, model_id, hf_api_key)
+            except Exception as e:
+                return f"Error with HuggingFace Inference: {e}", None
+
+
+        elif model_selection == "Groq API":
+            if not groq_api_key:
+                return "Error: Groq API key required", None
+            if not groq_model_choice:
+                return "Error: Select a Groq model", None
+            try:
+                summary = send_to_groq(prompt, groq_model_choice, groq_api_key)
+            except Exception as e:
+                return f"Error with Groq API: {e}", None
+
+        elif model_selection == "OpenAI ChatGPT":
+            if not openai_api_key:
+                return "Error: OpenAI API key required", None
+            try:
+                summary = send_to_openai(prompt, openai_api_key, model=openai_model_choice)
+            except Exception as e:
+                return f"Error with OpenAI API: {e}", None
 
-            else:
-                return "Error: Invalid model selection", None
+        else:
+            return "Error: Invalid model selection", None
 
-            with tempfile.NamedTemporaryFile(delete=False, mode='w', suffix='.txt') as f:
-                f.write(summary)
-                download_file = f.name
+        with tempfile.NamedTemporaryFile(delete=False, mode='w', suffix='.txt') as f:
+            f.write(summary)
+            download_file = f.name
 
-            return summary, download_file
+        return summary, download_file
 
-        except Exception as e:
-            error_msg = f"Error processing request: {str(e)}"
-            logging.error(error_msg)
-            return error_msg, None # Return the error message
+    except Exception as e:  # Outer exception handler
+        error_msg = f"An unexpected error occurred: {str(e)}"
+        logging.error(error_msg)
+        return error_msg, None
     
 def send_to_hf_inference(prompt: str, model_name: str, api_key: str) -> str:
    try: