Update app.py

app.py

@@ -49,7 +49,6 @@ chatgpt_chain = LLMChain(
     llm=OpenAI(temperature=1), 
     prompt=prompt, 
     verbose=True, 
-    return_intermediate_steps=True
 )
 
 #Helper function as this is what Pebblely API takes in
@@ -114,25 +113,26 @@ def upload_and_process(image, promptText, promptImg, lens_option, n_images, size
     image.save(image_path)
     # Remove backdrop
     responses = chatgpt_chain.predict(human_input=promptText)
-    print(responses["intermediate_steps"])
-    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)
+    print(json.dumps(responses, indent=2))
+    
+    # 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")
@@ -142,7 +142,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>{formatted_json}</h2>"
+    text_output = f"<h2>{json.dumps(responses, indent=2)}</h2>"
     
     return text_output, result_images  # Output should be a list of PIL Image objects or numpy arrays