keeping json and indentation

app.py

@@ -46,7 +46,7 @@ tax_deductions = '''Extract the following information in the given format:
 
 def format_nested_dict(data, indent=0):
     formatted_str = ""
-    indent_str = "    " * indent  # Indentation for the current level
+    indent_str = "        " * indent  # Indentation for the current level
     
     for key, value in data.items():
         # If value is a dictionary, recurse deeper
@@ -115,14 +115,15 @@ def process_document(image):
         image_path = tmp_file.name  # Get the path of the saved file
 
     # Process the image with your model
-    one = format_nested_dict(process_function(image_path, other_benifits))
-    two = format_nested_dict(process_function(image_path, tax_deductions))
-
+    one = process_function(image_path, other_benifits)
+    two = process_function(image_path, tax_deductions)
+    three = format_nested_dict(one)
+    four = format_nested_dict(two)
 
     # Optionally, you can delete the temporary file after use
     os.remove(image_path)
 
-    return one, two
+    return one, three, two, four
 
 
 
@@ -133,7 +134,9 @@ demo = gr.Interface(
     fn=process_document,
     inputs="image",  # Gradio will handle the image input
     outputs=[
-        gr.Textbox(label="Tax Deductions Information"),  # First output box with heading
+        gr.JSON(label="Tax Deductions Information"),  # First output box with heading
+        gr.Textbox(label="Other Benefits and Information"),  # Second output box with heading
+        gr.JSON(label="Tax Deductions Information"),  # First output box with heading
         gr.Textbox(label="Other Benefits and Information")  # Second output box with heading
     ],
     title="<div style='text-align: center;'>Information Extraction From PaySlip</div>",