Update app.py

app.py

@@ -119,15 +119,19 @@ def upload_and_process(image, promptText, promptImg, lens_option, n_images, size
     split_pairs = responses.split(" ")
 
     # Initialize an empty dictionary to store the key-value pairs
-    key_value_dict = {}
-    
-    # Iterate over each pair, splitting it further at the ":" character
+    key = None
+    value = ""
     for pair in split_pairs:
-        pair_parts = pair.split(":", 1)
-        if len(pair_parts) == 2:
-            key = pair_parts[0].strip()
-            value = pair_parts[1].strip()
-            key_value_dict[key] = value
+        if ":" in pair:
+            if key is not None:
+                key_value_dict[key.strip()] = value.strip()
+            key, value = pair.split(":", 1)
+        else:
+            value += " " + pair.strip()
+    
+    # Add the last key-value pair
+    if key is not None:
+        key_value_dict[key.strip()] = value.strip()
     
     # Print each key-value pair on a new line
     for key, value in key_value_dict.items():