return the best image

Author: liangchen76

app.py

@@ -102,8 +102,8 @@ def create_image(prompt: str) -> str:
 
         # Download the image and save it with a filename based on the prompt and current time
         current_time = datetime.now().strftime("%Y%m%d%H%M%S")
-        shortened_prompt = prompt[:50]        
-        image_file_path = f"{OUTPUT_FOLDER}/{shortened_prompt}_{current_time}.png"
+        shortened_prompt = format_filename_or_dir(prompt[:50])        
+        image_file_path = f"{OUTPUT_FOLDER}/{current_time}_{shortened_prompt}.png"
 
         response = requests.get(image_url)
         if response.status_code == 200:
@@ -153,7 +153,7 @@ def review_image(image_file_path: str, prompt: str):
 graphic_critic = AssistantAgent(
     name="graphic_critic",
     # system_message="You are an AI image critique, you will use img_review function to review the image generated by the text_to_img_prompt_expert against the original prompt, and provide feedback on how to improve the prompt.",
-     system_message="""Critic. You are a helpful assistant highly skilled in evaluating the quality of a given image code by providing a score from 1 (bad) - 10 (good). Specifically, you can carefully evaluate the image across the following dimensions
+     system_message="""Critic. You are a helpful assistant highly skilled in evaluating the quality of a given image code by providing a score from 1 (bad) - 10 (good) at each of the following dimensions: 
 - Content similarity (content): Does the image present the content in the prompt? If any keyword is missing, the score must less than 5. 
 - Clarity (clarity): Does the image clearly communicate the prompt? Is it easy to understand for the human viewers?
 - Balance and proportion (proportion): Are the elements arranged in a visually pleasing and harmonious way?
@@ -199,7 +199,11 @@ def review_image(image_file_path: str, prompt: str):
     # message = "In Houston at 2pm, Sunny sky with few clouds. Current Temperature at 37 degree"
     # message = "sunny sky with few clouds, Temperature low at 37 degree and high up to 72 degree"
     # message = "overcast sky with dense clouds, some foggy, Temperaturelow at 37 degree and high up to 72 degree"
-    message = "In Houston at 8pm, Heavy rain with very low visibility. Temperature in the 40s"
+    # message = "In Houston at 8pm, Heavy rain with very low visibility. Temperature in the 40s"
+    # message = "A city scenery of Houston at 6:30pm, featuring a sunny sky with few clouds, strong wind, a lot of vehicles on the road but no people walking"
+    # message = "A realistic image of an attractive and professional TV anchor woman, behind a desk, in national TV news station"
+    # message = "A realistic profile image of an attractive young girl for dating"
+    message = "A realistic image of a sexy girl wearing bikini standing with the background of weather forecast"
 
     # text_to_image_generation(message)
 
@@ -209,5 +213,26 @@ def review_image(image_file_path: str, prompt: str):
     Path(OUTPUT_FOLDER).mkdir(parents=True, exist_ok=True)
 
     # # Start the conversation
-    user_proxy.initiate_chat(
+    chat_result = user_proxy.initiate_chat(
         manager, message=message)
+    
+    print("CHAT_RESULT: ", chat_result)
+
+    # get the last generated image, which is the best image 
+    last_image_file_path = None
+    for history in chat_result.chat_history:
+        history_keys = history.keys() 
+        if "content" in history_keys and "name" in history_keys and "role" in history_keys: 
+            if history["name"] == "create_image" and history["role"] == "function":
+                last_image_file_path = history["content"]
+        # if "name" in history_keys and "role" in history_keys and "function_call" in history_keys: 
+        #     if history["name"] in ["graphic_designer", ""] and history["role"] == "assistant":
+        #         func_call = history["function_call"]
+        #         if "arguments" in func_call.keys():
+        #             func_call_arguments = func_call["arguments"]
+        #             if isinstance(func_call_arguments, str): 
+        #                 func_call_arguments = json.loads(func_call["arguments"])
+        #                 if "image_file_path" in func_call_arguments.keys(): 
+        #                     last_image_file_path = func_call_arguments["image_file_path"]
+
+    print("Best Image: ", last_image_file_path)