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

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)