Update app.py

app.py

@@ -27,6 +27,30 @@ class DogBehaviorAnalyzer:
             'high_activity': {'threshold': 0.05, 'description': 'Your dog is very energetic!'}
         }
         
+        self.suggestions = {
+            'tail_wagging': [
+                "This is a great time for positive reinforcement training!",
+                "Consider engaging in some fun play activities",
+                "Your dog is in a social mood - perfect for bonding exercises"
+            ],
+            'movement': [
+                "A good opportunity for some basic training exercises",
+                "Consider introducing some puzzle toys",
+                "This energy level is perfect for a short training session"
+            ],
+            'stationary': [
+                "Perfect time for gentle petting and calming interactions",
+                "Consider offering a chew toy for mental stimulation",
+                "Good moment for quiet bonding or rest"
+            ],
+            'high_activity': [
+                "Your dog might benefit from structured exercise",
+                "Consider redirecting energy into agility training",
+                "A good play session with toys would be beneficial",
+                "Make sure fresh water is available"
+            ]
+        }
+        
         # Motion detection parameters
         self.history = []
         self.max_history = 10
@@ -115,6 +139,38 @@ class DogBehaviorAnalyzer:
         
         return detected_behaviors
 
+    def get_suggestions(self, detected_behaviors, behavior_counts):
+        """Generate suggestions based on detected behaviors and their frequency"""
+        active_suggestions = []
+        
+        # Get total frames analyzed
+        total_frames = sum(behavior_counts.values())
+        if total_frames == 0:
+            return []
+            
+        # Calculate behavior percentages
+        behavior_percentages = {
+            behavior: count / total_frames * 100 
+            for behavior, count in behavior_counts.items()
+        }
+        
+        # Generate relevant suggestions based on current behaviors and their frequencies
+        for behavior, _ in detected_behaviors:
+            if behavior in self.suggestions:
+                # Select suggestions based on how frequently the behavior occurs
+                if behavior_percentages[behavior] > 30:  # If behavior occurs more than 30% of the time
+                    suggestions = self.suggestions[behavior]
+                    active_suggestions.extend(suggestions[:2])  # Add top 2 suggestions
+                    
+        # Add general suggestions based on overall behavior patterns
+        if behavior_percentages.get('high_activity', 0) > 50:
+            active_suggestions.append("Consider incorporating more calming activities in your routine")
+        elif behavior_percentages.get('stationary', 0) > 70:
+            active_suggestions.append("Your dog might benefit from more physical activity")
+            
+        # Remove duplicates and return
+        return list(set(active_suggestions))
+
 def main():
     st.title("🐕 Dog Behavior Analyzer")
     st.write("Upload a video of your dog for behavior analysis!")
@@ -188,6 +244,13 @@ def main():
                     analysis_text += f"  Count: {count} | Confidence: {avg_conf:.2f}\n"
                     analysis_text += f"  {analyzer.behaviors[behavior]['description']}\n\n"
             
+            # Get and display suggestions
+            suggestions = analyzer.get_suggestions(detected_behaviors, behavior_counts)
+            if suggestions:
+                analysis_text += "\nSuggestions:\n\n"
+                for suggestion in suggestions:
+                    analysis_text += f"💡 {suggestion}\n"
+            
             analysis_placeholder.text_area(
                 "Analysis Results",
                 analysis_text,
@@ -241,4 +304,4 @@ def main():
             st.write("Upload a video to see behavior analysis!")
 
 if __name__ == "__main__":
-    main()
+    main()