Update app.py

app.py

@@ -91,55 +91,63 @@ def IMGEN():
 
 @app.route('/generativeai', methods=['POST'])
 def Genration():
-    import google.generativeai as genai
-    generation_config = {
-     "temperature": 0.7,
-     "top_p": 1,
-     "top_k": 1,
-     "max_output_tokens": 300,
-    }
-    
-    safety_settings = [
-     {
-        "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"
-     },
-    ]
+    try:
+        import google.generativeai as genai
+        generation_config = {
+         "temperature": 0.7,
+         "top_p": 1,
+         "top_k": 1,
+         "max_output_tokens": 300,
+        }
+        
+        safety_settings = [
+         {
+            "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"
+         },
+        ]
+        
+        model = genai.GenerativeModel(
+         model_name="gemini-pro",
+         generation_config=generation_config,
+         safety_settings=safety_settings)
     
-    model = genai.GenerativeModel(
-     model_name="gemini-pro",
-     generation_config=generation_config,
-     safety_settings=safety_settings)
+        data = request.json
+        prompt = data.get('prompt', '')
+        messages = data.get('messages', [])
+        key = data.get('key', '')
+        
+        C=t()
+        genai.configure(api_key=key)
+        genai.configure(api_key=key)
+        response = model.generate_content(messages)
+        
+        
+        # Prepare the response
+        result = {
+            'response': response.text,
+            'execution_time': t()-C
+        }
+        return jsonify(result)
+    except:
+        result = {
+            'response': "I'm here to provide information and assistance in a respectful manner. Inappropriate or offensive language is not acceptable. If you have a different question or topic you'd like help with, please feel free to ask, and I'll do my best to assist you.",
+            'execution_time': t()-C
+        }
+        return jsonify(result)
 
-    data = request.json
-    prompt = data.get('prompt', '')
-    messages = data.get('messages', [])
-    key = data.get('key', '')
-    
-    C=t()
-    genai.configure(api_key=key)
-    genai.configure(api_key=key)
-    response = model.generate_content(messages)
-    
-    
-    # Prepare the response
-    result = {
-        'response': response.text,
-        'execution_time': t()-C
-    }
-    return jsonify(result)
 
 @app.route('/divyanshpizza', methods=['GET'])
 def get_counters():