Update app.py

app.py

@@ -105,8 +105,8 @@ def main():
                     try:
                         st.info("Rewriting the text. This takes time.", icon="ℹ️")
                         input_txt, paraphrased_txt = inference_long_text(input_txt, n_sents)
-                        input_txt = "<br><br>".join(input_txt)
-                        paraphrased_txt = "<br><br>".join(paraphrased_txt)
+                        # input_txt = "<br><br>".join(input_txt)
+                        # paraphrased_txt = "<br><br>".join(paraphrased_txt)
                     except Exception as e:
                         paraphrased_txt = None
                         paraphrase_error = str(e)
@@ -131,7 +131,9 @@ def main():
                 if paraphrase_error is not None:
                     st.error(f"Paraphrasing Error:  \n{paraphrase_error}", icon="🚨")
                 else:
-                    st.markdown(f"<h3>Scraped Text ({n_sents} sentences)</h3><p>{input_txt}</p><br><h3>Rewritten Text ({n_sents} sentences)</h3><p>{paraphrased_txt}</p>", unsafe_allow_html=True)
+                    result = [f"<b>Scraped Sentence:</b> {scraped}<br><b>Rewritten Sentence:</b> {paraphrased}" for scraped, paraphrased in zip(input_txt, paraphrased_txt)]
+                    result = "<br><br>".join(result)
+                    st.markdown(f"{result}", unsafe_allow_html=True)
 
     else:
         input_txt = st.text_area("Enter the text. (Ensure the text is grammatically correct and has punctuations at the right places):", "", height=150)
@@ -165,7 +167,9 @@ def main():
             if paraphrase_error is not None:
                 st.error(f"Paraphrasing Error:  \n{paraphrase_error}", icon="🚨")
             else:
-                st.markdown(f"<h3>Scraped Text ({n_sents} sentences)</h3><p>{input_txt}</p><br><h3>Rewritten Text ({n_sents} sentences)</h3><p>{paraphrased_txt}</p>", unsafe_allow_html=True)
+                result = [f"<b>Scraped Sentence:</b> {scraped}<br><b>Rewritten Sentence:</b> {paraphrased}" for scraped, paraphrased in zip(input_txt, paraphrased_txt)]
+                result = "<br><br>".join(result)
+                st.markdown(f"{result}", unsafe_allow_html=True)