Update app.py

app.py

@@ -11,7 +11,36 @@ from time import time as t
 
 app = Flask(__name__)
 
-
+generation_config = {
+ "temperature": 0.7,
+ "top_p": 1,
+ "top_k": 1,
+ "max_output_tokens": 512,
+}
+
+safety_settings = [
+ {
+    "category": "HARM_CATEGORY_HARASSMENT",
+    "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+ },
+ {
+    "category": "HARM_CATEGORY_HATE_SPEECH",
+    "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+ },
+ {
+    "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+    "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+ },
+ {
+    "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+    "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+ },
+]
+
+model = genai.GenerativeModel(
+ model_name="gemini-pro",
+ generation_config=generation_config,
+ safety_settings=safety_settings)
 
 
 
@@ -92,44 +121,14 @@ def IMGEN():
 
 @app.route('/generativeai', methods=['POST'])
 def Genration():
-    global model
+    global model,safety_settings,generation_config
     data = request.json
     prompt = data.get('prompt', '')
     messages = data.get('messages', [])
     key = data.get('key', '')
     
     C=t()
-    # Tracking API usage
-    generation_config = {
-     "temperature": 0.7,
-     "top_p": 1,
-     "top_k": 1,
-     "max_output_tokens": 350,
-    }
-    
-    safety_settings = [
-     {
-        "category": "HARM_CATEGORY_HARASSMENT",
-        "threshold": "BLOCK_MEDIUM_AND_ABOVE"
-     },
-     {
-        "category": "HARM_CATEGORY_HATE_SPEECH",
-        "threshold": "BLOCK_MEDIUM_AND_ABOVE"
-     },
-     {
-        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-        "threshold": "BLOCK_MEDIUM_AND_ABOVE"
-     },
-     {
-        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-        "threshold": "BLOCK_MEDIUM_AND_ABOVE"
-     },
-    ]
-    
-    model = genai.GenerativeModel(
-     model_name="gemini-pro",
-     generation_config=generation_config,
-     safety_settings=safety_settings)
+    genai.configure(api_key=key)
     genai.configure(api_key=key)
     response = model.generate_content(messages)