Update app.py

app.py

@@ -118,6 +118,7 @@ def upload_and_process(image, promptText, promptImg, lens_option, n_images, size
     print(responses)
     input_string = responses.replace("\\n", "\n")
 
+    
     # Remove unnecessary backslashes
     input_string = input_string.replace("\\", "")
     
@@ -131,7 +132,10 @@ def upload_and_process(image, promptText, promptImg, lens_option, n_images, size
     output_list = [pair.strip() for pair in output_string.split("\\n")]
     
     # Create a dictionary from the key-value pairs
-    output_dict = dict(pair.split(":") for pair in output_list)
+    output_dict = {}
+    for pair in output_list:
+        key, value = pair.split(":", 1)
+        output_dict[key.strip()] = value.strip()
     
     # Convert the dictionary to JSON format
     json_output = json.dumps(output_dict, indent=4)