Update app.py

app.py

@@ -7,7 +7,7 @@ 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)
 
-def generate(prompt,language,history):
+def generate(prompt,history,language):
     history.append("USER: "+prompt+"\n-----\n")
     output = client.predict(prompt,language)
     history.append("KHAYA: "+output+"\n*****\n")
@@ -16,6 +16,7 @@ def generate(prompt,language,history):
         full_output = full_output + el
     return  history,full_output
 
+'''
 with gr.Blocks(fill_height=True) as demo:
     history_var = gr.State([]) # conversation history
     title = gr.Markdown(
@@ -33,6 +34,19 @@ with gr.Blocks(fill_height=True) as demo:
         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])
+'''
+
+with gr.Blocks(fill_height=True) as demo:
+    title = gr.Markdown(
+    """
+    # African Language Chatbot (Khaya)
+    SELECT YOUR LANGUAGE BELOW!
+    """)
+    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:")
+    main_block = gr.ChatInterface(fn=generate, additional_inputs = language_selector,
+        description = 'Select an African language above and type messages below in selected language',
+        retry_btn=None,
+        undo_btn=None)
 
 if __name__ == "__main__":
     demo.queue(max_size=20).launch()