Update app.py

app.py

@@ -407,6 +407,18 @@ class SyntheticDataGenerator:
                     timeout=5
                 )
                 return response.status_code == 200
+            elif provider == "OpenAI":
+                try:
+                    client.models.list()
+                    return True
+                except Exception:
+                    return False
+            elif provider == "Groq":
+                try:
+                    client.models.list()
+                    return True
+                except Exception:
+                    return False
             elif provider == "Google":
                 try:
                     if not st.session_state.google_configured:  # Check if google has been configured
@@ -422,7 +434,6 @@ class SyntheticDataGenerator:
 
                         configure(api_key=api_key)  # Configure API Key
                         st.session_state.google_configured = True
-                        # st.write("configuring key")
 
                     genai.GenerativeModel(model_name=self.PROVIDER_CONFIG["Google"]["models"][0]).generate_content(
                         "test")  # Test a generation
@@ -433,8 +444,8 @@ class SyntheticDataGenerator:
                     return False
 
             else:
-                client.models.list()
-                return True
+                return False  # Unknown provider
+
         except Exception:
             return False