Update app.py

app.py

@@ -7,6 +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,history,language):
     history.append("USER: "+prompt+"\n-----\n")
     output = client.predict(prompt,language)
@@ -16,7 +17,7 @@ def generate(prompt,history,language):
         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(
@@ -36,6 +37,15 @@ with gr.Blocks(fill_height=True) as demo:
     btn.click(generate, inputs=[prompt,language_selector,history_var], outputs=[history_var,output])
 '''
 
+def generate(prompt,history,language):
+    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  output
+
 with gr.Blocks(fill_height=True) as demo:
     title = gr.Markdown(
     """