Update app.py

app.py

@@ -281,34 +281,47 @@ CORS(app)
 def intents():
     return jsonify({"intents" :list(set(df['intent'].dropna()))})
 
-@app.route("/en_response", methods=["POST"])
-def query_response():
-    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("/translate", methods=["POST"])
-def translate_model():
-    global audio_telugu_response
+def translate_text():
+    data = request.get_json()
+    text = data.get("text", "")
+    if not text:
+        return jsonify({"error": "Text cannot be empty"}), 400
+
+    # First generate English response
+    english_response = text
+    start=time.time()
+    # Then translate to Telugu
+    telugu_response = translate(model_telugu, english_response, en_vocab, te_vocab, te_inv_vocab)
+    end=time.time()
+    return jsonify({
+        "english": english_response,
+        "telugu": telugu_response,
+        "time": end-start
+    })
+
+@app.route("/generate", methods=["POST"])
+def generate_text():
     data = request.get_json()
     query = data.get("query", "")
-    
+
+    if not query:
+        return jsonify({"error": "Query cannot be empty"}), 400
+    start=time.time()
+    response = generate_response(model, query)
+    end=time.time()
+    # Clean the response
     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),"translate_time":(end_te-start_te)})
+    
+    response = clean_response(response)
+    
+    return jsonify({
+        "response": response,
+        "time": end-start
+    })
 
 @app.route("/query", methods=["POST"])
 def query_model():