Update app.py

app.py

@@ -1,51 +1,22 @@
 import streamlit as st
-import os
-import logging
-from gully_drs_core.ball_detection import process_video
+import tempfile
+from gully_drs_core import ball_detection, replay_utils
 
-# Set up logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
+st.title("🏏 GullyDRS – LBW Review System")
 
-# Streamlit configuration
-st.set_page_config(page_title="GullyDRS", layout="wide")
-st.title("GullyDRS - AI-Powered Decision Review System")
+video_file = st.file_uploader("Upload a cricket match video", type=["mp4", "avi"])
+if video_file:
+    tfile = tempfile.NamedTemporaryFile(delete=False)
+    tfile.write(video_file.read())
+    st.video(video_file)
 
-# Directories for uploads and outputs
-UPLOAD_DIR = "uploads"
-OUTPUT_DIR = "outputs"
-os.makedirs(UPLOAD_DIR, exist_ok=True)
-os.makedirs(OUTPUT_DIR, exist_ok=True)
-
-def main():
-    st.header("Upload Match Video for DRS Analysis")
-    uploaded_file = st.file_uploader("Choose a video file (MP4, AVI)", type=["mp4", "avi"])
-    
-    if uploaded_file is not None:
-        try:
-            # Save uploaded video
-            video_path = os.path.join(UPLOAD_DIR, uploaded_file.name)
-            with open(video_path, "wb") as f:
-                f.write(uploaded_file.getbuffer())
-            st.success(f"Video '{uploaded_file.name}' uploaded successfully!")
-            
-            # Process video
-            with st.spinner("Processing video for DRS analysis..."):
-                result = process_video(video_path)
-                if result["status"] == "success":
-                    # Display replay video (live preview)
-                    st.subheader("Replay Video")
-                    st.video(result["replay_path"])
-                    st.write(f"**LBW Decision**: {result['decision']}")
-                    st.write(f"**Ball Speed**: {result['speed_kmh']:.2f} km/h")
-                    # External test frame (local file access for testing)
-                    st.subheader("External Replay (Test Frame)")
-                    st.markdown(f'<video width="640" height="360" controls><source src="file://{result["replay_path"]}" type="video/mp4"></video>', unsafe_allow_html=True)
-                else:
-                    st.error(f"Error: {result['error']}")
-        except Exception as e:
-            logger.error(f"Error processing video: {str(e)}")
-            st.error(f"Failed to process video: {str(e)}")
-
-if __name__ == "__main__":
-    main()
+    if st.button("Analyze Video"):
+        with st.spinner("Processing..."):
+            result = ball_detection.analyze_video(tfile.name)
+            replay_path = replay_utils.generate_replay(
+                result["frames"],
+                result["trajectory"],
+                fps=result["fps"]
+            )
+        st.success("Replay generated!")
+        st.video(replay_path)