Changed chatbot to basic blocks

app.py

@@ -6,7 +6,7 @@ model = TFT5ForConditionalGeneration.from_pretrained('Yati05/TF-CodeT5-base')
 # load saved tokenizer
 tokenizer = RobertaTokenizer.from_pretrained('Yati05/TF-CodeT5-base')
 
-def chat(chat_history, user_input):
+def predict(user_input):
     query = "Generate Python: " + user_input
     encoded_text = tokenizer(query, return_tensors='tf', padding='max_length', truncation=True, max_length=48)
     
@@ -19,17 +19,16 @@ def chat(chat_history, user_input):
     # decode generated tokens
     decoded_code = tokenizer.decode(generated_code.numpy()[0], skip_special_tokens=True)
     
-    return chat_history + [(user_input, decoded_code)]
-#   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)]
-     
+    return decoded_code
 
-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="Text Prompt",placeholder="Enter the prompt to generate code")
-    message.submit(chat, [chatbot, message], chatbot)
+     
 
-demo.queue().launch(debug = True)
+with gr.Blocks(title="Python Code Generation",theme=gr.themesMonochrome()) as demo:
+    gr.HTML(value="<style>h1 {text-align: center;}</style><h1>Python Code Generation</h1>")  
+    #gr.Markdown("# Python Code Generation")
+    input = gr.Textbox(label="Text Prompt",placeholder="Enter the prompt to generate code")
+    greet_btn = gr.Button("Generate")
+    output = gr.Textbox(label="Code")
+    greet_btn.click(fn=predict, inputs=input, outputs=output)
+ 
+demo.launch(debug=True)