Update app.py

app.py

@@ -110,13 +110,13 @@ def upload_and_process(image, promptText, promptImg, lens_option, n_images, size
     # Remove backdrop
     responses = chatgpt_chain.predict(human_input=promptText)
     print(responses)
-    print(type(responses))
-    #responses = json.loads(responses)
+    #print(type(responses))
+    response = json.loads(responses)
 
     dall_e_prompt = promptImg + " , " + lens_option
-    primary = "P"
-    headline = "R"
-    description = "D"
+    primary = response['primary']
+    headline = response['headline']
+    description = response['description']
     # Generate variations using DALL-E
     removeBackdrop(image_path, "temp_no_bg.png")
     NoBackImg = "temp_no_bg.png"