change name for streamlit_app.py

app.py

@@ -1,25 +1,42 @@
-import numpy as np
-from flask import Flask, request, jsonify, render_template
-import pickle
+import streamlit as st
+from happytransformer import HappyTextToText, TTSettings
 
-from happytransformer import TTSettings
+st.set_page_config(page_title="Grammar Correction Tool", layout="centered")
 
-app = Flask(__name__)
+checkpoint = "team-writing-assistant/t5-base-c4jfleg"
 
-@app.route('/')
-def home():
-    return render_template('index.html')
+@st.cache_resource
+def get_happy_text(model_name):
+    return HappyTextToText("T5", model_name)
 
-@app.route('/predict',methods=['POST'])
-def predict():
+happy_tt = get_happy_text(checkpoint)
+args = TTSettings(num_beams=5, min_length=1)
 
-    int_features = str(request.form['Input_text'])
-    beam_settings = TTSettings(num_beams=5, min_length=1, max_length=20)
+st.title(" NLP - Grammar Correction Tool")
+st.subheader("Software - Group 2")
+st.markdown("""
+Simply enter your text below or use one of the example sentences to get started!
+""")
 
-    model = pickle.load(open('model.pkl', 'rb'))
-    output = model.generate_text(int_features, args=beam_settings)
+col1, col2 = st.columns(2)
 
-    return render_template('index.html', prediction_text='{}'.format(output.text))
+with col1:
+    if st.button("Example 1: Incorrect Grammar"):
+        st.session_state.input_text = "Speed of light is fastest then speed of sound"
+with col2:
+    if st.button("Example 2: Common Mistake"):
+        st.session_state.input_text = "Who are the president?"
 
-if __name__ == "__main__":
-    app.run(host='0.0.0.0', debug=True)
+input_text = st.text_area("Enter your text here:", st.session_state.get("input_text", ""))
+
+if st.button("Correct Grammar"):
+    if input_text.strip():
+        with st.spinner("Correcting grammar..."):
+            formatted_input = "grammar: " + input_text
+            result = happy_tt.generate_text(formatted_input, args=args)
+            st.markdown("### Corrected Text:")
+            st.write(result.text.strip())
+    else:
+        st.warning("Please enter text to correct.")
+
+st.markdown("---")

app_flask.py

@@ -0,0 +1,25 @@
+import numpy as np
+from flask import Flask, request, jsonify, render_template
+import pickle
+
+from happytransformer import TTSettings
+
+app = Flask(__name__)
+
+@app.route('/')
+def home():
+    return render_template('index.html')
+
+@app.route('/predict',methods=['POST'])
+def predict():
+
+    int_features = str(request.form['Input_text'])
+    beam_settings = TTSettings(num_beams=5, min_length=1, max_length=20)
+
+    model = pickle.load(open('model.pkl', 'rb'))
+    output = model.generate_text(int_features, args=beam_settings)
+
+    return render_template('index.html', prediction_text='{}'.format(output.text))
+
+if __name__ == "__main__":
+    app.run(host='0.0.0.0', debug=True)

streamlit_app.py

@@ -1,42 +0,0 @@
-import streamlit as st
-from happytransformer import HappyTextToText, TTSettings
-
-st.set_page_config(page_title="Grammar Correction Tool", layout="centered")
-
-checkpoint = "team-writing-assistant/t5-base-c4jfleg"
-
-@st.cache_resource
-def get_happy_text(model_name):
-    return HappyTextToText("T5", model_name)
-
-happy_tt = get_happy_text(checkpoint)
-args = TTSettings(num_beams=5, min_length=1)
-
-st.title(" NLP - Grammar Correction Tool")
-st.subheader("Software - Group 2")
-st.markdown("""
-Simply enter your text below or use one of the example sentences to get started!
-""")
-
-col1, col2 = st.columns(2)
-
-with col1:
-    if st.button("Example 1: Incorrect Grammar"):
-        st.session_state.input_text = "Speed of light is fastest then speed of sound"
-with col2:
-    if st.button("Example 2: Common Mistake"):
-        st.session_state.input_text = "Who are the president?"
-
-input_text = st.text_area("Enter your text here:", st.session_state.get("input_text", ""))
-
-if st.button("Correct Grammar"):
-    if input_text.strip():
-        with st.spinner("Correcting grammar..."):
-            formatted_input = "grammar: " + input_text
-            result = happy_tt.generate_text(formatted_input, args=args)
-            st.markdown("### Corrected Text:")
-            st.write(result.text.strip())
-    else:
-        st.warning("Please enter text to correct.")
-
-st.markdown("---")