Update app.py

app.py

@@ -1,34 +1,35 @@
 import streamlit as st
 import tempfile
+import os
 from gully_drs_core import ball_detection, replay_utils
 
 st.set_page_config(page_title="GullyDRS – LBW Review", layout="centered")
 st.title("🏏 GullyDRS – LBW Decision Review System")
 
-st.markdown(
-    """
-    Upload a cricket match video and let AI analyze whether it's **OUT** or **NOT OUT** 
-    based on ball trajectory, bounce point, and stump zone detection.
-    """
-)
+st.markdown("""
+Upload a cricket match video and let AI analyze whether it's **OUT** or **NOT OUT** 
+based on ball trajectory, bounce point, and stump zone detection.
+""")
 
-# Upload video file
 video_file = st.file_uploader("🎥 Upload your match video", type=["mp4", "avi"])
 
 if video_file:
-    # Store video temporarily
     tfile = tempfile.NamedTemporaryFile(delete=False)
     tfile.write(video_file.read())
-
-    # Preview uploaded video
     st.video(video_file)
 
-    # Trigger analysis
     if st.button("🧠 Analyze Video"):
         with st.spinner("Analyzing with AI..."):
             result = ball_detection.analyze_video(tfile.name)
 
-            # Generate replay with overlays
+            st.write(f"🔍 Trajectory points: {len(result['trajectory'])}")
+            st.write(f"🏁 Final Decision: **{result['decision']}**")
+            st.write(f"🚀 Ball Speed: **{result['speed_kmh']} km/h**")
+            if result["bounce_point"]:
+                st.info(f"📍 Bounce detected at: {result['bounce_point']}")
+            if result["impact_point"]:
+                st.info(f"🎯 Impact point: {result['impact_point']}")
+
             replay_path = replay_utils.generate_replay(
                 frames=result["frames"],
                 ball_path=result["trajectory"],
@@ -36,15 +37,12 @@ if video_file:
                 impact_point=result["impact_point"],
                 decision=result["decision"],
                 stump_zone=result["stump_zone"],
+                speed_kmh=result["speed_kmh"],
                 fps=result["fps"]
             )
 
-        # Show decision and replay
-        st.success(f"🏁 Final Decision: **{result['decision']}**")
-        st.video(replay_path)
-
-        # Show extra info
-        if result["bounce_point"]:
-            st.info(f"📍 Bounce detected at: {result['bounce_point']}")
-        if result["impact_point"]:
-            st.info(f"🎯 Impact point: {result['impact_point']}")
+        if replay_path and os.path.exists(replay_path):
+            st.success("✅ Replay generated successfully!")
+            st.video(replay_path)
+        else:
+            st.error("⚠️ Replay generation failed or produced no video.")