Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import json
 from transformers import pipeline
 
 @st.cache_resource
@@ -52,8 +53,13 @@ def main():
         # Clear 'Generating Output' message and display the generated text
         output_placeholder.empty()
         st.write("Generated Text:")
-        st.write(generated_text)
+        try:
+            parsed_json = json.loads(generated_text)
+            beautified_json = json.dumps(parsed_json, indent=4)
+            st.code(beautified_json, language="json")
+        except json.JSONDecodeError:
+            st.write(generated_text)
 
 # Run the app
 if __name__ == "__main__":
-    main()
+    main()