Update app_dialogue.py

app_dialogue.py

@@ -68,26 +68,34 @@ from PIL import Image
 from PIL import Image
 import tempfile
 
+    filename = f"{uuid.uuid4()}.jpg"
+    local_path = f"{filename}" 
+    image.save(local_path)
+    img_str =  f"![](/file={local_path})"
+    return img_str
+
 def convert_to_rgb_pil(image):
     print(f"***** convert_to_rgb_pil ******")
     print(f"params: image is - {image}")
     #if image.mode == "RGB":
     #    return image
     # Save the converted image to a temporary file
-    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".jpg")
-    temp_file_path = temp_file.name
-
+    #temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".jpg")
+    #temp_file_path = temp_file.name
+    filename = f"{uuid.uuid4()}.jpg"
+    local_path = f"{filename}"
+    
     if image.mode != "RGB":
         image_rgba = image.convert("RGBA")
         background = Image.new("RGBA", image_rgba.size, (255, 255, 255))
         alpha_composite = Image.alpha_composite(background, image_rgba)
         alpha_composite = alpha_composite.convert("RGB")
-        alpha_composite.save(temp_file_path)
+        alpha_composite.save(local_path)
     else:
-        image.save(temp_file_path)
+        image.save(local_path)
         
     temp_file.close()
-    print(f"# Return the path to the saved image as - {temp_file_path}")
+    print(f"# Return the path to the saved image as - {local_path}")
     return temp_file_path  # Return the path to the saved image
     
 
@@ -99,6 +107,7 @@ def convert_to_rgb(filepath_or_pilimg):
     
     if isinstance(filepath_or_pilimg, PIL.Image.Image):
         return convert_to_rgb_pil(filepath_or_pilimg)
+        
     with Image.open(filepath_or_pilimg) as image:
         # Check if the image is already in the RGB format
         if image.mode == "RGB":
@@ -117,12 +126,14 @@ def convert_to_rgb(filepath_or_pilimg):
         alpha_composite = alpha_composite.convert("RGB")
 
         # Save the converted image to a temporary file
-        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".jpg")
-        temp_file_path = temp_file.name
-        alpha_composite.save(temp_file_path)
+        filename = f"{uuid.uuid4()}.jpg"
+        local_path = f"{filename}"
+        #temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".jpg")
+        #temp_file_path = temp_file.name
+        alpha_composite.save(local_path)
         temp_file.close()
 
-        print(f"# Return the path to the saved image as - {temp_file_path}")
+        print(f"# Return the path to the saved image as - {local_path}")
         return temp_file_path  # Return the path to the saved image
 
 def pil_to_markdown_im(image):
@@ -811,6 +822,11 @@ And so, the story of Mulan and Shrek's romance came to an end, leaving a lasting
         penalty_alpha,
     ):
         # global processor, model, tokenizer
+        print("***********Model_inference*************")
+        print(f"Inside Model_inference, user_prompt_str is - {user_prompt_str} ")
+        print(f"Inside Model_inference, chat_history is - {chat_history} ")
+        print(f"Inside Model_inference, image type is - {type(image)} ")
+        print(f"Inside Model_inference, image is - {image} ")
 
         force_words = ""
         hide_special_tokens = False
@@ -821,6 +837,9 @@ And so, the story of Mulan and Shrek's romance came to an end, leaving a lasting
             history=chat_history,
         )
 
+        print(f"formated_prompt_list (or resulting_list)  is {formated_prompt_list}")
+        print(f"user_prompt_list (or [current_user_prompt_str]) is {user_prompt_list}")
+        
         generated_text = model_generation(
             prompt_list=formated_prompt_list,
             processor=processor,
@@ -848,6 +867,7 @@ And so, the story of Mulan and Shrek's romance came to an end, leaving a lasting
             chat_history.append(
                 (user_prompt_list_to_markdown(user_prompt_list), generated_text.strip("<end_of_utterance>"))
             )
+            print(f"chat_history (IF image is None or is with fake token) is -{chat_history}")
         else:
             # Case where the image is passed through the Image Box.
             # Convert the image into base64 for both passing it through the chat history and
@@ -858,9 +878,11 @@ And so, the story of Mulan and Shrek's romance came to an end, leaving a lasting
                     generated_text.strip("<end_of_utterance>"),
                 )
             )
+            int(f"chat_history (ELSE IF image is available) is -{chat_history}")
         return "", None, chat_history
 
     def process_example(message, image):
+        print("********* process_example **********")
         clear_msg, image_value, chat = model_inference(
             user_prompt_str=message,
             chat_history=[],