Update app.py

app.py

@@ -117,7 +117,13 @@ def upload_and_process(image, promptText, promptImg, lens_option, n_images, size
     responses = chatgpt_chain.predict(human_input=promptText)
     print(responses)
     split_pairs = responses.split(" ")
-
+    pk =""
+    hk =""
+    dk = ""
+    primary = ""
+    headline = ""
+    description = ""
+    
     # Initialize an empty dictionary to store the key-value pairs
     key_value_dict = {}
     key = None
@@ -136,7 +142,7 @@ def upload_and_process(image, promptText, promptImg, lens_option, n_images, size
     
     # Print each key-value pair on a new line
     for key, value in key_value_dict.items():
-        print("He",key + ": " + value)
+        print(key + ": " + value)
     # json_string = responses.strip()
 
     # # Add indentation and new lines after each key-value pair
@@ -164,7 +170,7 @@ def upload_and_process(image, promptText, promptImg, lens_option, n_images, size
     # # Get and return the images
     # result_images = outputGenImages(image_path, response_data)
     
-    text_output = f"<h2>{responses}</h2>"
+    text_output = f"<h2>Headline:{key_value_dict['headline']}</h2><h3>Primary:{key_value_dict['primary']}</h3><p>Description:{key_value_dict['description']}</p>"
     
     return text_output#, result_images  # Output should be a list of PIL Image objects or numpy arrays