Update app.py

app.py

@@ -116,30 +116,19 @@ def upload_and_process(image, promptText, promptImg, lens_option, n_images, size
     # Remove backdrop
     responses = chatgpt_chain.predict(human_input=promptText)
     print(responses)
-    input_string = responses.replace("\\n", "\n")
+    split_pairs = string.split(" ")
 
+    # Initialize an empty dictionary to store the key-value pairs
+    key_value_dict = {}
     
-    # Remove unnecessary backslashes
-    input_string = input_string.replace("\\", "")
-    
-    # Extract the desired string value
-    output_string = input_string[input_string.find("Output :") + 9:]
-    
-    # Remove leading/trailing whitespace and double quotes
-    output_string = output_string.strip().strip('"')
-    
-    # Split the string into key-value pairs
-    output_list = [pair.strip() for pair in output_string.split("\\n")]
-    
-    # Create a dictionary from the key-value pairs
-    output_dict = {}
-    for pair in output_list:
+    # Iterate over each pair, splitting it further at the ":" character
+    for pair in split_pairs:
         key, value = pair.split(":", 1)
-        output_dict[key.strip()] = value.strip()
+        key_value_dict[key.strip()] = value.strip()
     
-    # Convert the dictionary to JSON format
-    json_output = json.dumps(output_dict, indent=4)
-    print(json_output)
+    # Print each key-value pair on a new line
+    for key, value in key_value_dict.items():
+        print("He",key + ": " + value)
     # json_string = responses.strip()
 
     # # Add indentation and new lines after each key-value pair