Update app.py

app.py

@@ -1,5 +1,28 @@
 import gradio as gr
 import json
+import jsonfix
+import json5
+
+# Function to attempt to fix JSON using jsonfix and json5
+def fix_json(json_data):
+    try:
+        # First, try parsing with json5 which is more lenient
+        parsed_data = json5.loads(json_data)
+        
+        # Try to pretty-print (indent) the parsed data
+        fixed_json = json.dumps(parsed_data, indent=4)
+        return fixed_json, "JSON fixed using json5."
+    
+    except Exception as e:
+        # If json5 fails, try jsonfix to attempt to auto-correct
+        try:
+            fixed_json = jsonfix.fix_data(json_data)
+            parsed_data = json.loads(fixed_json)  # Validate it as valid JSON
+            fixed_json = json.dumps(parsed_data, indent=4)
+            return fixed_json, "JSON fixed using jsonfix."
+        
+        except Exception as e:
+            return None, f"Failed to fix JSON: {str(e)}"
 
 # Function to remove duplicates and fix JSON
 def process_json(json_data):
@@ -7,9 +30,14 @@ def process_json(json_data):
     if not json_data.strip():
         return None, "Error: The uploaded file is empty.", None
 
+    # Attempt to fix the JSON
+    fixed_json, fix_message = fix_json(json_data)
+    if fixed_json is None:
+        return None, fix_message, None
+
     try:
-        # Parse the JSON string to a list of dictionaries
-        data = json.loads(json_data)
+        # Parse the fixed JSON string to a list of dictionaries
+        data = json.loads(fixed_json)
         
         # Check if data is a list of dictionaries
         if not isinstance(data, list):
@@ -21,10 +49,10 @@ def process_json(json_data):
         # Convert the list back to JSON
         cleaned_json = json.dumps(unique_data, indent=4)
         
-        return cleaned_json, None, cleaned_json
+        return cleaned_json, f"JSON is valid and cleaned. {fix_message}", cleaned_json
 
     except json.JSONDecodeError as e:
-        return None, f"Invalid JSON: {e}", None
+        return None, f"Invalid JSON after fixing: {e}", None
 
 # Function to handle file uploads and processing
 def process_file(uploaded_file):
@@ -32,10 +60,10 @@ def process_file(uploaded_file):
     json_data = uploaded_file  # This is already the content of the file as a string
     
     # Process the JSON data
-    cleaned_json, error, download_json = process_json(json_data)
+    cleaned_json, message, download_json = process_json(json_data)
     
-    if error:
-        return None, error, None
+    if message:
+        return cleaned_json, message, download_json
     
     # Return the cleaned JSON for display and as a downloadable file
     return cleaned_json, "JSON is valid and cleaned.", download_json
@@ -45,8 +73,8 @@ iface = gr.Interface(
     fn=process_file,
     inputs=gr.File(label="Upload your JSON file"),
     outputs=[gr.JSON(label="Fixed JSON"), "text", gr.File(label="Download cleaned JSON file")],
-    title="JSON Cleaner and Fixer",
-    description="Upload a JSON file to remove duplicates, fix its structure, and download the cleaned version."
+    title="Advanced JSON Cleaner and Fixer",
+    description="Upload a JSON file to automatically fix, remove duplicates, and download the cleaned version."
 )
 
 iface.launch()