app.py

# from flask import Flask, request, render_template
# from predict import Predict

# app = Flask(__name__)

# @app.route('/')
# def home():
#     return render_template('index.html')

# @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

def page():
    st.title("jRefactoring")
    codeSnippet = st.text_input('Enter Java code here', "public class hello { ... }")
    st.text("")

    if st.button('Check'):
        if(codeSnippet!=""):
            st.text("")
            result = Predict().predict(codeSnippet)
            st.write(result)

page()