Adding Swahili Translation

app.py

@@ -10,6 +10,11 @@ import sys
 llm = OpenAI(temperature=0.0, model="gpt-3.5-turbo")
 client = OpenAIOG()
 
+from langdetect import detect
+from langdetect import DetectorFactory
+DetectorFactory.seed = 0
+from deep_translator import GoogleTranslator
+
 # Load index
 from llama_index.core import VectorStoreIndex
 from llama_index.core import StorageContext
@@ -23,6 +28,12 @@ import gradio as gr
 def nishauri(question: str, conversation_history: list[str]):
     
     context = " ".join([item["user"] + " " + item["chatbot"] for item in conversation_history])
+
+    lang_question = detect(question)
+    
+    if lang_question=="sw":
+        question = GoogleTranslator(source='sw', target='en').translate(question)
+        
     response = query_engine.query(question)
 
     background = ("The person who asked the question is a person living with HIV."
@@ -50,10 +61,15 @@ def nishauri(question: str, conversation_history: list[str]):
         {"role": "user", "content": question_final}
       ]
     )
+
+    reply_to_user = completion.choices[0].message.content
+    
+    if lang_question=="sw":
+        reply_to_user = GoogleTranslator(source='auto', target='sw').translate(reply_to_user)
     
     conversation_history.append({"user": question, "chatbot": response.response})   
 
-    return completion.choices[0].message.content, conversation_history
+    return reply_to_user, conversation_history
 
 demo = gr.Interface(
     title = "Nishauri Chatbot Demo",