Update app.py

app.py

@@ -1,8 +1,10 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 import requests
+from googletrans import Translator
 
 client = InferenceClient("mistralai/Mistral-Nemo-Instruct-2407")
+translator = Translator()
 
 def get_internet_data(query: str) -> str:
     """
@@ -20,7 +22,6 @@ def get_internet_data(query: str) -> str:
         response = requests.get(url, params=params, timeout=5)
         response.raise_for_status()
         data = response.json()
-        # Retrieve a summary from the API's result
         result = data.get("AbstractText", "")
         if not result:
             result = "Wala koy nakuha nga impormasyon gikan sa internet."
@@ -28,24 +29,43 @@ def get_internet_data(query: str) -> str:
         result = "Naay problema sa pagkuha sa impormasyon gikan sa internet."
     return result
 
+def translate_to_bisaya(text: str) -> str:
+    """
+    Translates the given text to Bisaya (Cebuano) using googletrans.
+    """
+    try:
+        # 'ceb' is the ISO 639-3 code for Cebuano
+        translation = translator.translate(text, dest='ceb')
+        return translation.text
+    except Exception:
+        return text
+
 def respond(message, history: list[tuple[str, str]]):
-    system_message = "Ikaw usa ka buotan nga Chatbot. Tubaga lang sa binisaya, ug ayaw gamita ang english nga pinulungan. Gamita ang pinakabag-ong impormasyon gikan sa internet kung kinahanglan."
+    system_message = (
+        "Ikaw usa ka buotan nga Chatbot. Tubaga lang sa binisaya, "
+        "ug ayaw gamita ang english nga pinulungan. Gamita ang pinakabag-ong "
+        "impormasyon gikan sa internet ug isalin kini sa binisaya."
+    )
     max_tokens = 4096
     temperature = 0.6
     top_p = 0.95
 
     messages = [{"role": "system", "content": system_message}]
     for user_text, assistant_text in history:
-        # Ensure alternating roles: user then assistant
         if user_text:
             messages.append({"role": "user", "content": user_text})
         if assistant_text:
             messages.append({"role": "assistant", "content": assistant_text})
     
-    # Get search results based on the user's query.
+    # Automatically perform an internet search for the query.
     search_result = get_internet_data(message)
-    # Enrich the user message with the search result.
-    enriched_message = f"{message}\n\nMga resulta gikan sa internet: {search_result}"
+    # Translate the search result to Bisaya.
+    translated_result = translate_to_bisaya(search_result)
+    # Enrich the user query with the translated internet search result.
+    enriched_message = (
+        f"{message}\n\nMga resulta gikan sa internet (isinalin sa binisaya): "
+        f"{translated_result}"
+    )
     messages.append({"role": "user", "content": enriched_message})
 
     response = ""
@@ -61,11 +81,9 @@ def respond(message, history: list[tuple[str, str]]):
         if not token:
             break
         response += token
-        # Yield only if new content was added.
         if response != previous_response:
             yield response
             previous_response = response
-        # Optional: break if the response is too long.
         if len(response) > 3000:
             break