Update app.py

app.py

@@ -17,6 +17,9 @@ from gtts import gTTS
 from googletrans import Translator
 import uuid
 import os
+import time
+
+
 # Load Dataset
 df = pd.read_csv("https://drive.google.com/uc?id=1RCZShB5ohy1HdU-mogcP16TbeVv9txpY")
 df = df.dropna(subset=['instruction', 'response'])
@@ -278,6 +281,35 @@ CORS(app)
 def intents():
     return jsonify({"intents" :list(set(df['intent'].dropna()))})
 
+@app.route("/en_response", methods=["POST"])
+def query_model():
+    data = request.get_json()
+    query = data.get("query", "")
+    # Get current time in seconds
+    if not query:
+        return jsonify({"error": "Query cannot be empty"}), 400
+
+    start = time.time()
+    # Assuming `generate_response` is a function that processes the query
+    response = generate_response(model, query)
+    end = time.time()
+    return jsonify({"english":(response),"response_time":(end-start)})
+
+@app.route("/query", methods=["POST"])
+def query_model():
+    global audio_telugu_response
+    data = request.get_json()
+    query = data.get("query", "")
+    
+    def clean_response(response):
+        return response.replace("<EOS>", "").replace("<SOS>", "").strip()
+    start_te = time.time()
+    response=clean_response(response)
+    telugu_response = translate(model_telugu, response, en_vocab, te_vocab, te_inv_vocab)
+    end_te = time.time()
+    audio_telugu_response=telugu_response
+    return jsonify({"telugu":(telugu_response),"english":(response),"telugu_time":(end_te-start_te)})
+
 @app.route("/query", methods=["POST"])
 def query_model():
     global audio_telugu_response
@@ -287,14 +319,20 @@ def query_model():
     if not query:
         return jsonify({"error": "Query cannot be empty"}), 400
 
+    start_eng = time.time()
     # Assuming `generate_response` is a function that processes the query
     response = generate_response(model, query)
+    end_eng = time.time()
     def clean_response(response):
         return response.replace("<EOS>", "").replace("<SOS>", "").strip()
     response=clean_response(response)
+    start_te = time.time()
     telugu_response = translate(model_telugu, response, en_vocab, te_vocab, te_inv_vocab)
+    end_te = time.time()
     audio_telugu_response=telugu_response
-    return jsonify({"telugu":(telugu_response),"english":(response)})
+    return jsonify({"telugu":(telugu_response),"english":(response),"eng_time":(end_eng-start_eng),"telugu_time":(end_te-start_te)})
+
+    
 @app.route("/audio", methods=["POST"])
 def get_audio():
     data = request.get_json()