Update app.py

app.py

@@ -18,4 +18,69 @@ def diff_strings(a, b):
         result.append("  ")
         result.append(line[2:])
       else: 
-    
+        result.append("  ")
+        result.append((""), replacement, "#ffd"))
+        replacement = ""
+        result.append(line[2:])
+    elif line.startswith("- "):
+      if len(replacement) == 0:
+        replacement = line[2:]
+      else:
+        result.append("  ")
+        result.append((""), replacement, "#fdd"))
+        replacement = ""
+      elif line.startswith("+ "):
+        if len(replacement) == 0:
+          result.append((line[2:]), "", "#dfd"))
+        else:
+          result.append("  ")
+          result.append((line[2:], replacement, "#ddf"))
+          replacement = ""
+  return result
+  
+@st.cache(suppress_st_warning=True, allow_output_mutation=True)
+  def get_happy_text(model_name):
+    st.write(f"Loading the HappyTextToText model {model_name}, please wait...")
+    return HappyTextToText("T5", model_name)
+         
+happy_tt = get_happy_text(checkpoint)
+args = TTSettings(num_beams=5, min_length=1)
+
+st.title("Grammar Corrector Two")
+st.markdown("Paste or type text, enter and the machine will attempt to correct your text's grammar.")
+
+st.subheader("Example text: ")
+col1, col2 = st.columns([1, 1])
+with col1:
+  example_1 = st.button("Speed of light is fastest then speed of sound")
+with col2:
+  example_2 = st.button("Who are the president?")
+
+input_text = st.text_area('Paste or type text')
+button = st.button('Submit')
+
+def output(text):
+  with st.spinner('Correcting'):
+    text = "grammar: " + text
+    result = happy_tt.generate_text(text, args=args)
+    diff = diff_strings(text[9:], result.text)
+    annotated_text(*diff)
+
+    copy_button = Button(label="Copy the Result")
+    copy_button.js_on_event("button_click", CustomJS(args=dict(result=result.text), code="""
+      navigator.clipboard.writeText(result);
+      """))
+    streamlit_bokeh_events(
+      copy_button,
+      events="GET_TEXT",
+      key="get_text",
+      refresh_on_update=True,
+      override_height=75,
+      debounce_time=0)
+
+if example_1:
+  output("Speed of light is fastest then speed of sound")
+elif example_2:
+  output("Who are the president?")
+elif input_text:
+  output(input_text)