Update app.py

app.py

@@ -4,35 +4,39 @@ from gradio_client import Client
 
 MY_HF_TOKEN_KEY = os.environ["MY_HF_TOKEN_KEY"]
 
-#iface = gr.load(name="Ghana-NLP/Khaya-chat-bot",hf_token=MY_HF_TOKEN_KEY,src ='spaces')
-client = Client("Ghana-NLP/Khaya-chat-backend",hf_token=MY_HF_TOKEN_KEY)
+client = Client("Ghana-NLP/kikuyu-tts",hf_token=MY_HF_TOKEN_KEY)
 
-def generate(prompt,language,history):
-    history.append("USER: "+prompt+"\n-----\n")
-    output = client.predict(prompt,language)
-    history.append("KHAYA: "+output+"\n*****\n")
-    full_output = ""
-    for el in history:
-        full_output = full_output + el
-    return  history,full_output
+def generate(text):
+    output = client.predict(text)
+    return  output
 
-with gr.Blocks() as demo:
-    history_var = gr.State([]) # conversation history
-    title = gr.Markdown(
-    """
-    # African Language Chatbot (Khaya)
-    1. SELECT YOUR LANGUAGE
-    2. TYPE IN SELECTED LANGUAGE!!
-    3. REFRESH PAGE TO START FRESH CONVERSATION \n\n
-    Based on the <a href="https://translation.ghananlp.org/">Khaya AI Translation API</a>, <a href="https://lesan.ai/">Lesan AI</a>, Google Translate API and the Mixtral model. Lesan is used for <b>Amharic</b> and <b>Tigrinya</b>, <a href="https://translation.ghananlp.org/">Khaya AI</a> is used for <b>Twi</b>, <b>Dagbani</b>, <b>Ewe</b>, <b>Ga</b>, <b>Gurene</b>, <b>Fante</b>, <b>Kikuyu</b>, <b>Kimeru</b>, <b>Luo</b>, <b>Yoruba</b> and Google is used for <b>Hausa</b>, <b>Swahili</b> and <b>Shona</b>.
-    """)    
-    language_selector = gr.Dropdown(["Amharic","Twi","Dagbani","Ewe","Ga","Gurene","Fante","Hausa", "Kikuyu", "Kimeru", "Luo","Shona","Swahili","Tigrinya","Yoruba"],value="Twi",label="Choose Language! (default is Twi)", info="Language:")
-    with gr.Row():
-        output = gr.Text(label="OUTPUT WINDOW (Conversation so far):")
-    with gr.Row():
-        prompt = gr.Text(label="Enter Text In Your Local Language:")
-    btn = gr.Button("Chat")
-    btn.click(generate, inputs=[prompt,language_selector,history_var], outputs=[history_var,output])
+title = "GhanaNLP: Speech Synthesis"
 
-if __name__ == "__main__":
-    demo.queue(max_size=20).launch()
+description = """
+<b>How to use:</b> Enter some Kikuyu text and choose a speaker.
+"""
+
+examples = [
+    ["Nao makĩguthũkĩra indo iria maatahĩte, makĩoya ngʼondu, na ngʼombe, na tũcaũ, magĩcithĩnjĩra hau thĩ, na magĩcirĩanĩria na thakame.", "ki_1"],
+    ["Hĩndĩ ĩyo Abimeleku agĩthiĩ kũrĩ Isaaka oimĩte Gerari, marĩ na Ahuzathu ũrĩa wamũtaaraga, na Fikolu mũnene wa ita ciake.", "ki_2"],
+    ["Wee nĩũũĩ ũrĩa matu macio macuurĩtio wega, magegania macio ma ũrĩa ũrĩ ũmenyo mũkinyanĩru?", "ki_3"],
+]
+
+gr.Interface(
+    fn=synthesize,
+    inputs=[
+        gr.Text(label="Input Text"),
+        gr.Radio(label="Speaker", choices=[
+            "ki_1",
+            "ki_2",
+            "ki_3",
+        ],
+        value="ki_1"),
+    ],
+    outputs=[
+        gr.Audio(label="Generated Speech", type="numpy"),
+    ],
+    title=title,
+    description=description,
+    examples=examples,
+).launch()