Update app.py

app.py

@@ -6,11 +6,43 @@ from datetime import datetime
 from os import listdir
 from web import Online_Scraper
 import requests
+import google.generativeai as genai
+from time import time as t
 
 app = Flask(__name__)
 
 
 # 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)
 
 
 @app.route('/mistral7b', methods=['POST'])
@@ -87,7 +119,26 @@ def IMGEN():
 
 
     return requests.post(API_URL, headers=headers, json={"inputs": prompt,}).content
+
+@app.route('/generativeai', methods=['POST'])
+def Genration():
+    global model
+    data = request.json
+    prompt = data.get('prompt', '')
+    messages = data.get('messages', [])
+    key = data.get('key', '')
+    
+    C=t()
+    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():