Update app.py

app.py

@@ -114,6 +114,24 @@ def upload_and_process(image, promptText, promptImg, lens_option, n_images, size
     # Remove backdrop
     responses = chatgpt_chain.predict(human_input=promptText)
     print(responses)
+    json_string = responses.strip()
+
+    # Add indentation and new lines after each key-value pair
+    formatted_json = ''
+    indentation_level = 0
+    for char in json_string:
+        if char == '{' or char == '[':
+            formatted_json += char + '\n' + '    ' * indentation_level
+            indentation_level += 1
+        elif char == '}' or char == ']':
+            indentation_level -= 1
+            formatted_json += '\n' + '    ' * indentation_level + char
+        elif char == ',':
+            formatted_json += char + '\n' + '    ' * indentation_level
+        else:
+            formatted_json += char
+
+    print(formatted_json)
     dall_e_prompt = promptImg + " , " + lens_option
     # Generate variations using DALL-E
     removeBackdrop(image_path, "temp_no_bg.png")
@@ -123,7 +141,7 @@ def upload_and_process(image, promptText, promptImg, lens_option, n_images, size
     # Get and return the images
     result_images = outputGenImages(image_path, response_data)
     
-    text_output = f"<h2>{responses}</h2>"
+    text_output = f"<h2>{formatted_json}</h2>"
     
     return text_output, result_images  # Output should be a list of PIL Image objects or numpy arrays