Update app.py

app.py

@@ -29,6 +29,7 @@ from sklearn.metrics.pairwise import cosine_similarity
 import torch
 import gradio as gr
 from transformers import pipeline
+from langdetect import detect
 
 audio_model_id = "Sandiago21/whisper-large-v2-greek"  # update with your model id
 audio_pipe = pipeline("automatic-speech-recognition", model=audio_model_id)
@@ -823,6 +824,9 @@ def tool_executor(state: AgentState):
     """
 
     try:
+        user_input = state["messages"][-1].content
+        useer_input_language = detect(user_input)
+        
         webpage_result = ""
         action = Action.model_validate(state["proposed_action"])
     
@@ -838,7 +842,7 @@ def tool_executor(state: AgentState):
             query_arg_embeddings = sentence_transformer_model.encode_query(state["proposed_action"]["args"]["query"]).reshape(1, -1)
             score = float(cosine_similarity(query_embeddings, query_arg_embeddings)[0][0])
     
-            if score > 0.80:
+            if score > 0.80 or useer_input_language != "en":
                 results = web_search(**action.args)
             else:
                 logger.info(f"Overwriting user query because the Agent suggested query had score: {state["proposed_action"]["args"]["query"]} - {score}")