Update app.py

app.py

@@ -47,19 +47,30 @@ def generate_response(message, selected_topics):
     inputs = tokenizer.encode(message, return_tensors="pt")
     outputs = model.generate(inputs, max_length=50, do_sample=True)
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    #response = f"Echo: {message}. Selected topics: {', '.join(selected_topics)}"
     return response
 
 
+def main():
+    with gr.Blocks() as demo:
+        gr.Markdown("### Child-Safe Chatbot BETA")
+        with gr.Row():
+            message_input = gr.Textbox(label="Your Message")
+            topics_dropdown = gr.Dropdown(choices=topics_list, label="Select Topics", multiselect=True)
+            submit_btn = gr.Button("Send")
+        response_output = gr.Textbox(label="Bot Response")
 
+        def combined_function(message, selected_topics):
+            return generate_response(message, [selected_topics])
 
-dropdown = gr.components.Dropdown(choices=topics_list, label="Select Topics", multiselect=True)
+        submit_btn.click(
+            fn=combined_function, 
+            inputs=[message_input, topics_dropdown], 
+            outputs=response_output
+        )
+
+    demo.launch()
 
-iface = gr.ChatInterface(
-    fn=generate_response, 
-    title="Child-Safe Chatbot BETA", 
-    additional_inputs=[dropdown],  
-    description="A chatbot that stays on topic and filters inappropriate content. Select relevant topics.",
-                    )
 
 # Run the app
 if __name__ == "__main__":