stats

app/streamlit_app.py

@@ -8,6 +8,7 @@ import tempfile
 import streamlit as st
 from dotenv import load_dotenv
 import base64
+import threading
 from pathlib import Path
 import shutil
 import cv2
@@ -27,6 +28,7 @@ from models.swing_analyzer import segment_swing, analyze_trajectory
 from models.llm_analyzer import generate_swing_analysis, create_llm_prompt, prepare_data_for_llm, check_llm_services, parse_and_format_analysis, display_formatted_analysis
 from utils.visualizer import create_annotated_video
 from utils.comparison import create_key_frame_comparison, extract_key_swing_frames
+from utils.stats_publisher import record_swing_event
 
 # Import RAG functionality
 print("=== RAG Import Debug Information ===")
@@ -763,6 +765,28 @@ def main():
                 'prompt': prompt
             }
 
+            # Fire-and-forget stats publish to Hugging Face Space branch
+            try:
+                llm_status = {
+                    'ollama_available': llm_services.get('ollama', {}).get('available', False),
+                    'openai_available': llm_services.get('openai', {}).get('available', False),
+                }
+                stats_payload = {
+                    'event_type': 'swing_analyzed',
+                    'source': 'youtube' if input_option == 'YouTube URL' else 'upload',
+                    'video_name': os.path.basename(video_path) if video_path else None,
+                    'sample_rate': sample_rate,
+                    'frame_count': len(frames) if frames is not None else None,
+                    'llm_status': llm_status,
+                    'pro_comparison_enabled': False,
+                    'app_version': os.getenv('APP_VERSION', 'unknown'),
+                    'space_id': os.getenv('SPACE_ID'),
+                }
+
+                threading.Thread(target=record_swing_event, args=(stats_payload,), daemon=True).start()
+            except Exception as stats_err:
+                print(f"Stats publish error: {stats_err}")
+
             # Keep the original video file for potential annotation
             # Video will be cleaned up when user uploads a new video or session ends
 
@@ -772,7 +796,7 @@ def main():
 
             with options_col1:
                 st.info(
-                    "**Option 1: Generate Improvement Recommendations**\n\nGet AI-powered analysis of your swing with specific tips for improvement."
+                    "**Option 1: Generate Recommendations**\n\nGet AI-powered analysis of your swing with specific tips for improvement."
                 )
                 
             with options_col2:

requirements.txt

@@ -9,7 +9,7 @@ torchvision==0.17.0
 openai>=1.0.0
 python-dotenv==1.0.0
 tqdm==4.66.1
-streamlit==1.29.0
+streamlit>=1.32.0
 pandas==2.1.4
 sentence-transformers>=2.2.0
 faiss-cpu>=1.7.0
@@ -17,3 +17,4 @@ scikit-learn==1.3.2
 plotly==5.17.0
 tiktoken==0.5.2
 transformers>=4.20.0 
+huggingface_hub>=0.23.0