Update ChatBot.py

ChatBot.py

@@ -56,26 +56,30 @@ def find_question_and_answer(json_file, question, input_ids):
     generated_response = tokenizer.decode(generated_output[0], skip_special_tokens=True)
     return generated_response
 
-@app.route("/")
-def index():
-    return render_template('index.html')
+# @app.route("/")
+# def index():
+#     return render_template('index.html')
 
-@app.route('/ask', methods=['POST'])
-def ask():
-    user_input = request.form['user_input']
-    if not user_input:
-        response = ""
-    else:
-        input_ids = tokenizer.encode(user_input, return_tensors='pt')
-        response = find_question_and_answer(json_file_path, user_input, input_ids)
+# @app.route('/ask', methods=['POST'])
+# def ask():
+#     user_input = request.form['user_input']
+#     if not user_input:
+#         response = ""
+#     else:
+#         input_ids = tokenizer.encode(user_input, return_tensors='pt')
+#         response = find_question_and_answer(json_file_path, user_input, input_ids)
 
-    if isinstance(response, dict):
-        response_type = "mapping"
-    else:
-        response_type = "string"
+#     if isinstance(response, dict):
+#         response_type = "mapping"
+#     else:
+#         response_type = "string"
 
-    return render_template('index.html', user_input=user_input, response=response, response_type=response_type)
+#     return render_template('index.html', user_input=user_input, response=response, response_type=response_type)
 
 
 if __name__ == '__main__':
-    app.run(debug=True)
+    user_input = st.text_area("Ask Haven AI: ")
+    response = find_question_and_answer(json_file_path, user_input)
+    st.write(response)
+    app.run(debug=True)
+