Update app.py

app.py

@@ -6,6 +6,7 @@ from datetime import datetime
 from os import listdir
 from web import Online_Scraper
 import requests
+from RT import RealTimeGemini
 from time import time as t
 import os
 
@@ -148,6 +149,65 @@ def Genration():
         return jsonify(result)
 
 
+@app.route('/realtime', methods=['POST'])
+def Genration():
+    try:
+        import google.generativeai as genai
+        generation_config = {
+            "temperature": 0.9,
+            "top_p": 1,
+            "top_k": 1,
+            "max_output_tokens": 300,
+        }
+        
+        safety_settings = [
+            {
+            "category": "HARM_CATEGORY_HARASSMENT",
+            "threshold": "BLOCK_NONE"
+            },
+            {
+            "category": "HARM_CATEGORY_HATE_SPEECH",
+            "threshold": "BLOCK_NONE"
+            },
+            {
+            "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+            "threshold": "BLOCK_NONE"
+            },
+            {
+            "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+            "threshold": "BLOCK_NONE"
+            },
+        ]
+        
+        model = genai.GenerativeModel(
+         model_name="gemini-pro",
+         generation_config=generation_config,
+         safety_settings=safety_settings)
+    
+        data = request.json
+        query = data.get('prompt', "hello ?")
+        messages = data.get('messages', [])
+        key = data.get('key', '')
+        
+        C=t()
+        genai.configure(api_key=key)
+        response = RealTimeGemini(query,messages,model)
+        
+        
+        # Prepare the response
+        result = {
+            'response': response,
+            'execution_time': t()-C
+        }
+        return jsonify(result)
+    except Exception as e:
+        result = {
+            'response': f"{e}",
+            'execution_time': t()-C
+        }
+        return jsonify(result)
+
+
 @app.route('/divyanshpizza', methods=['GET'])
 def get_counters():
     return jsonify(counter),jsonify({"data":str(listdir(r"static/data/"))})