Update app.py

app.py

@@ -54,23 +54,37 @@ def analyze_colors(image):
         return None
 
 
-
 # Function to detect emotions from colors (simplified emotion-color mapping)
 def color_emotion_analysis(dominant_colors):
     try:
         emotions = []
+        stress_levels = []
+        
         for color in dominant_colors:
-            # Simple logic: darker tones could indicate sadness
-            if np.mean(color) < 85:
+            brightness = np.mean(color)
+
+            # Simple logic for emotion and stress based on brightness
+            if brightness < 85:
                 emotions.append("Sadness")
-            elif np.mean(color) > 170:
+                stress_levels.append("High Stress")
+            elif 85 <= brightness < 120:
+                emotions.append("Neutral")
+                stress_levels.append("Moderate Stress")
+            elif 120 <= brightness < 170:
+                emotions.append("Okay")
+                stress_levels.append("Low Stress")
+            elif 170 <= brightness < 200:
                 emotions.append("Happiness")
+                stress_levels.append("Very Low Stress")
             else:
-                emotions.append("Neutral")
-        return emotions
+                emotions.append("Very Happy")
+                stress_levels.append("No Stress")
+        
+        return emotions, stress_levels
+
     except Exception as e:
         print(f"Error in color_emotion_analysis: {e}")
-        return ["Error analyzing emotions"]
+        return ["Error analyzing emotions"], ["Error analyzing stress levels"]
 
 # Function to analyze patterns and shapes using OpenCV
 def analyze_patterns(image):
@@ -83,12 +97,12 @@ def analyze_patterns(image):
         num_edges = np.sum(edges > 0)
 
         if num_edges > 10000:  # Arbitrary threshold for "chaos"
-            return "Chaotic patterns - possibly distress"
+            return "Chaotic patterns - possibly distress", "High Stress"
         else:
-            return "Orderly patterns - possibly calm"
+            return "Orderly patterns - possibly calm", "Low Stress"
     except Exception as e:
         print(f"Error in analyze_patterns: {e}")
-        return "Error analyzing patterns"
+        return "Error analyzing patterns", "Error analyzing stress levels"
 
 # Main function to process image and analyze emotional expression
 def analyze_emotion_from_image(image):
@@ -102,12 +116,19 @@ def analyze_emotion_from_image(image):
         if dominant_colors is None:
             return "Error analyzing colors"
         
-        color_emotions = color_emotion_analysis(dominant_colors)
+        color_emotions, color_stress_levels = color_emotion_analysis(dominant_colors)
 
         # Analyze patterns
-        pattern_analysis = analyze_patterns(image)
+        pattern_analysis, pattern_stress_level = analyze_patterns(image)
+
+        # Combine color and pattern stress levels
+        overall_stress_level = pattern_stress_level if "High Stress" in color_stress_levels else "Moderate Stress"
+
+        return (f"Color-based emotions: {color_emotions}\n"
+                f"Color-based stress levels: {color_stress_levels}\n"
+                f"Pattern analysis: {pattern_analysis}\n"
+                f"Overall stress level: {overall_stress_level}")
 
-        return f"Color-based emotions: {color_emotions}\nPattern analysis: {pattern_analysis}"
     except Exception as e:
         return f"Error processing image: {str(e)}"