load in json

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import tiktoken
+import json
 
 def count_tokens(text):
     """
@@ -18,8 +19,15 @@ def count_tokens(text):
     # Encode the input text to get the list of token IDs
     tokens = encoding.encode(text)
     
-    # Return the number of tokens and the beautified text
-    return len(tokens), text.replace("\\n", "\n")
+    try:
+        parsed_json = json.loads(text)
+        text = json.dumps(parsed_json, indent=4)
+    except json.JSONDecodeError:
+        pass
+   
+    text = text.replace("\\n", "\n")
+    
+    return len(tokens), text
 
 # Define the Gradio interface
 iface = gr.Interface(