Update app.py

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 import os
 import re
 from groq import Groq
-from groq.errors import AuthenticationError, APIError
 
 def validate_api_key(api_key):
     """Validate if the API key has the correct format."""
@@ -26,12 +25,12 @@ def test_api_connection(api_key):
             max_tokens=5
         )
         return True, "API connection successful"
-    except AuthenticationError:
-        return False, "Authentication failed: Invalid API key"
-    except APIError as e:
-        return False, f"API Error: {str(e)}"
     except Exception as e:
-        return False, f"Error connecting to Groq API: {str(e)}"
+        # Handle all exceptions since Groq might not expose specific error types
+        if "authentication" in str(e).lower() or "api key" in str(e).lower():
+            return False, "Authentication failed: Invalid API key"
+        else:
+            return False, f"Error connecting to Groq API: {str(e)}"
 
 def chat_with_groq(api_key, model, user_message, temperature, max_tokens, top_p, chat_history):
     """
@@ -85,9 +84,7 @@ def clear_conversation():
 models = [
     "deepseek-r1-distill-llama-70b",
     "llama3-70b-8192",
-    "llama3-8b-8192",
-    "mixtral-8x7b-32768",
-    "gemma-7b-it"
+    "llama3-8b-8192"
 ]
 
 # Create the Gradio interface