streamlit integration

app.py

@@ -1,18 +1,34 @@
 
-from flask import Flask, request, render_template
-from predict import Predict
+# from flask import Flask, request, render_template
+# from predict import Predict
+
+# app = Flask(__name__)
+
+# @app.route('/')
+# def home():
+#     return render_template('index.html')
 
-app = Flask(__name__)
+# @app.route('/predict', methods=['POST'])
+# def predict():
+#     feature_list = request.form.to_dict()
+#     result = Predict().predict(feature_list['code'])
+#     return render_template('index.html', prediction_text=result)
+
+# if __name__ == "__main__":
+#     app.run(debug=True, use_reloader=False, port='8080')
+
+import streamlit as st
+from predict import Predict
 
-@app.route('/')
-def home():
-    return render_template('index.html')
+def page():
+    st.title("jRefactoring")
+    codeSnippet = st.text_input('Enter Java code here', "public class hello { ... }")
+    st.text("")
 
-@app.route('/predict', methods=['POST'])
-def predict():
-    feature_list = request.form.to_dict()
-    result = Predict().predict(feature_list['code'])
-    return render_template('index.html', prediction_text=result)
+    if st.button('Check'):
+        if(codeSnippet!=""):
+            st.text("")
+            result = Predict().predict(codeSnippet)
+            st.write(result)
 
-if __name__ == "__main__":
-    app.run(debug=True, use_reloader=False, port='8080')
+page()