Update app.py

app.py

@@ -149,12 +149,13 @@ def upload_and_process(image, promptText, promptImg, lens_option, n_images, size
     for key, value in key_value_dict.items():
         print(key + ": " + value)
     items = list(key_value_dict.items())
-    description = items[0][0]
-    print(description)
-    primary = items[1][0]
-    print(primary)
-    headline = items[2][0]
-    print(headline)
+    if(items[0][0]=='primary')
+        primary=items[0][1]
+    elif(items[0][0]=='headline')
+        headline=items[0][1])
+    elif(items[0][0]=='description')
+        description=items[0][1])
+    
     
     # json_string = responses.strip()