Update app.py

app.py

@@ -73,11 +73,14 @@ def analyze_image(image):
                     "text": """Your task is to determine if the image is surprising or not surprising.    
                     if the image is surprising, determine which element, figure or object in the image is making the image surprising and write it only in one sentence with no more then 6 words, otherwise, write 'NA'.    
                     Also rate how surprising the image is on a scale of 1-5, where 1 is not surprising at all and 5 is highly surprising.
+                    Additionally, write one sentence about what would be expected in this scene, and one sentence about what is unexpected.
                     Provide the response as a JSON with the following structure:    
                     {
                         "label": "[surprising OR not surprising]",
                         "element": "[element]",
-                        "rating": [1-5]
+                        "rating": [1-5],
+                        "expected": "[one sentence about what would be expected]",
+                        "unexpected": "[one sentence about what is unexpected]"
                     }"""
                 },
                 {
@@ -250,13 +253,13 @@ def process_image_detection(image, target_label, surprise_rating):
                ax.add_patch(rect)
 
                # Only add the probability score
-               plt.text(
-                   box[0], box[1] - base_fontsize,
-                   f'{max_score:.2f}',
-                   color='red',
-                   fontsize=base_fontsize,
-                   fontweight='bold'
-               )
+               #plt.text(
+               #    box[0], box[1] - base_fontsize,
+               #    f'{max_score:.2f}',
+               #    color='red',
+               #    fontsize=base_fontsize,
+               #    fontweight='bold'
+               #)
 
        plt.axis('off')
 
@@ -354,8 +357,10 @@ def process_and_analyze(image):
                 analysis_text = (
                     f"Label: {response_data['label']}\n"
                     f"Element: {response_data['element']}\n"
-                    f"Rating: {response_data['rating']}/5"
-                )
+                    f"Rating: {response_data['rating']}/5\n"
+                    f"Expected: {response_data['expected']}\n"
+                    f"Unexpected: {response_data['unexpected']}"
+            )
                 return result_image, analysis_text
             except Exception as detection_error:
                 print(f"Detection error details: {str(detection_error)}")  # Debug print