Update app.py

app.py

@@ -24,10 +24,7 @@ def get_api_key_from_secrets():
     """Try to get API key from HuggingFace secrets or environment variables"""
     # Try multiple possible secret names
     possible_keys = [
-        'GOOGLE_API_KEY',
-        'GEMINI_API_KEY', 
-        'GOOGLE_GEMINI_API_KEY',
-        'GOOGLE_AI_API_KEY'
+        'GEMINI_API_KEY'
     ]
     
     for key_name in possible_keys:
@@ -61,39 +58,15 @@ def setup_gemini(api_key: str = None):
         API_KEY = get_api_key_from_secrets()
     
     if not API_KEY:
-        return False, "❌ No API key found. Please:\n1. Set GOOGLE_API_KEY in HuggingFace Spaces secrets, or\n2. Enter your API key in the interface below\n\n🔗 Get your API key from: https://makersuite.google.com/app/apikey"
+        return False, "❌ No API key found. Please:\n1. Set GOOGLE_API_KEY in HuggingFace Spaces secrets"
     
     try:
         # Configure the API
         genai.configure(api_key=API_KEY)
         
-        # Initialize the model with safety settings
-        safety_settings = [
-            {
-                "category": "HARM_CATEGORY_MEDICAL",
-                "threshold": "BLOCK_NONE"
-            },
-            {
-                "category": "HARM_CATEGORY_HARASSMENT",
-                "threshold": "BLOCK_ONLY_HIGH"
-            },
-            {
-                "category": "HARM_CATEGORY_HATE_SPEECH", 
-                "threshold": "BLOCK_ONLY_HIGH"
-            },
-            {
-                "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-                "threshold": "BLOCK_ONLY_HIGH"
-            },
-            {
-                "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-                "threshold": "BLOCK_ONLY_HIGH"
-            }
-        ]
         
         GEMINI_MODEL = genai.GenerativeModel(
-            'gemini-1.5-pro',
-            safety_settings=safety_settings
+            'gemini-1.5-pro'
         )
         
         # Test the connection with a simple query
@@ -401,7 +374,7 @@ def create_interface():
             if initial_api_key:
                 gr.Markdown("""
                 ### ✅ API Key Found in Secrets
-                An API key was found in the environment/secrets. You can still override it below if needed.
+                An API key was found in the environment/secrets.
                 """)
             else:
                 gr.Markdown("""