Change UI to chatbot

app.py

@@ -1,12 +1,12 @@
 import gradio as gr
 from transformers import TFT5ForConditionalGeneration, RobertaTokenizer
-
+import regex as re
 # load saved finetuned model
 model = TFT5ForConditionalGeneration.from_pretrained('Yati05/TF-CodeT5-base')
 # load saved tokenizer
 tokenizer = RobertaTokenizer.from_pretrained('Yati05/TF-CodeT5-base')
 
-def predict(user_input):
+def chat(chat_history, user_input):
     query = "Generate Python: " + user_input
     encoded_text = tokenizer(query, return_tensors='tf', padding='max_length', truncation=True, max_length=48)
     
@@ -19,15 +19,18 @@ def predict(user_input):
     # decode generated tokens
     decoded_code = tokenizer.decode(generated_code.numpy()[0], skip_special_tokens=True)
     
-    return decoded_code
-
+    return chat_history + [(user_input, re.sub(r"( {4})", " ", bot_response))]
+#   response = ""
+#   for letter in decoded_code: #[decoded_code[i:i+1] for i in range(0, len(decoded_code), 1)]:
+#       response += letter + ""
+#       yield chat_history + [(user_input, response)]
      
 
-with gr.Blocks(title="Python Code Generation",theme=gr.themes.Monochrome()) as demo:
-    gr.HTML(value="<style>h1 {text-align: center;}</style><h1>Python Code Generation</h1>")  
-    input = gr.Textbox(label="Write a python script to:",placeholder="e.g. Check if a number is prime")
-    greet_btn = gr.Button("Generate")
-    output = gr.Textbox(label="Code")
-    greet_btn.click(fn=predict, inputs=input, outputs=output)
- 
+with gr.Blocks(title="Python Code Generation") as demo:
+    gr.HTML(value="<style>h1 {text-align: center;}</style><h1>Python Code Generation</h1>") 
+    chatbot = gr.Chatbot([], elem_id="chatbot")
+    message = gr.Textbox(label="Write a python script to..",placeholder="Eg. Check if a number is prime")
+    message.submit(chat, [chatbot, message], chatbot)
+
+
 demo.launch(enable_queue=True)