updates

dvc.yaml

@@ -36,11 +36,14 @@ stages:
       - models
       - src/models/evaluate_model.py
     metrics:
-      - reports/metrics.txt:
+      - reports/evaluation_metrics.txt:
           cache: false
   visualize:
     cmd: streamlit run src/visualization/visualize.py
     deps:
       - models
       - src/visualization/visualize.py
+    metrics:
+      - reports/visualization_metrics.txt:
+          cache: false
 

src/visualization/visualize.py

@@ -13,9 +13,12 @@ def visualize():
     text = st.text_area("Enter text here")
     if st.button("Generate Summary"):
         with st.spinner("Connecting the Dots..."):
-            text = predict_model(text=text)
+            sumtext = predict_model(text=text)
         st.write("# Generated Summary:")
-        st.write("{}".format(text))
+        st.write("{}".format(sumtext))
+        with open("reports/visualization_metrics.txt", "w") as file1:
+            file1.writelines(text)
+            file1.writelines(sumtext)
 
 
 if __name__ == "__main__":