Update app_pages/video_analyzer.py

app_pages/video_analyzer.py

@@ -10,104 +10,142 @@ from utils.dataframe import analysis_to_csv
 from database import insert_video_analysis, get_all_video_analyses
 
 def analyzer_page():
+    st.sidebar.header("Analyzer")
     selected_tab = st.sidebar.radio("Select Mode", ["Video Analyser", "History"], index=0)
 
     if selected_tab == "Video Analyser":
         st.subheader(" Video Analyser")
-        uploaded_video = st.file_uploader("Upload Video",
-                                          type=['mp4','mov','avi','mkv'],
-                                          help="Upload a video for analysis")
+        uploaded_video = st.file_uploader(
+            "Upload Video",
+            type=['mp4', 'mov', 'avi', 'mkv'],
+            help="Upload a video for analysis"
+        )
         analyse_button = st.button("Run Analysis", use_container_width=True)
 
         if uploaded_video and analyse_button:
-            print("Analysing")
-            with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(uploaded_video.name)[1]) as tmp:
+            # Save to a temp file
+            suffix = os.path.splitext(uploaded_video.name)[1]
+            with tempfile.NamedTemporaryFile(delete=False, suffix=suffix) as tmp:
                 tmp.write(uploaded_video.read())
                 video_path = tmp.name
+
             with st.spinner("Analyzing video..."):
-                st.session_state["analysis"] = analyze_video_only(video_path)
-                st.session_state["video_name"] = uploaded_video.name
-                st.session_state["video_path"] = video_path
-                st.session_state["thumbnail"] = get_video_thumbnail_base64(video_path)
-
-        if "analysis" in st.session_state and st.session_state["analysis"]:
-            render_analyzer_results(st.session_state["analysis"])
-
-            col1, col2 = st.columns(2)
-            with col1:
-                analysis = st.session_state["analysis"]
-                frames = []
-                if "storyboard" in analysis:
-                    df_storyboard = pd.DataFrame(analysis["storyboard"])
-                    df_storyboard["section"] = "Storyboard"
-                    frames.append(df_storyboard)
-                if "script" in analysis:
-                    df_script = pd.DataFrame(analysis["script"])
-                    df_script["section"] = "Script"
-                    frames.append(df_script)
-                if "video_analysis" in analysis and "video_metrics" in analysis["video_analysis"]:
-                    df_metrics = pd.DataFrame(analysis["video_analysis"]["video_metrics"])
-                    df_metrics["section"] = "Metrics"
-                    frames.append(df_metrics)
-                if "timestamp_improvements" in analysis:
-                    df_improvements = pd.DataFrame(analysis["timestamp_improvements"])
-                    df_improvements["section"] = "Improvements"
-                    frames.append(df_improvements)
-
-                if frames:
-                    csv_content = pd.concat(frames, ignore_index=True).to_csv(index=False)
-                    st.download_button(
-                        "Download CSV",
-                        data=csv_content,
-                        file_name=f"{st.session_state.get('video_name','analysis')}.csv",
-                        mime="text/csv",
-                        use_container_width=True
+                result = analyze_video_only(video_path)
+
+    
+            st.session_state["video_name"] = uploaded_video.name
+            st.session_state["video_path"] = video_path
+            st.session_state["thumbnail"] = get_video_thumbnail_base64(video_path)
+            st.session_state["analysis"] = result
+
+        analysis = st.session_state.get("analysis")
+
+        if analysis is None:
+            return
+
+     
+        if not analysis:
+            st.error("Analysis returned no data.")
+            return
+
+        if "__error__" in analysis:
+            st.error(f"Analysis failed: {analysis['__error__']}")
+            return
+
+        
+        render_analyzer_results(analysis)
+
+        col1, col2 = st.columns(2)
+
+        # CSV export
+        with col1:
+            frames = []
+            if "storyboard" in analysis:
+                df_storyboard = pd.DataFrame(analysis["storyboard"])
+                df_storyboard["section"] = "Storyboard"
+                frames.append(df_storyboard)
+
+            if "script" in analysis:
+                df_script = pd.DataFrame(analysis["script"])
+                df_script["section"] = "Script"
+                frames.append(df_script)
+
+            if "video_analysis" in analysis and "video_metrics" in analysis["video_analysis"]:
+                df_metrics = pd.DataFrame(analysis["video_analysis"]["video_metrics"])
+                df_metrics["section"] = "Metrics"
+                frames.append(df_metrics)
+
+            if "timestamp_improvements" in analysis:
+                df_improvements = pd.DataFrame(analysis["timestamp_improvements"])
+                df_improvements["section"] = "Improvements"
+                frames.append(df_improvements)
+
+            if frames:
+                csv_content = pd.concat(frames, ignore_index=True).to_csv(index=False)
+                st.download_button(
+                    "Download CSV",
+                    data=csv_content,
+                    file_name=f"{st.session_state.get('video_name', 'analysis')}.csv",
+                    mime="text/csv",
+                    use_container_width=True,
+                )
+            else:
+                st.info("No tabular data available for CSV export.")
+
+        # Save to DB
+        with col2:
+            if st.button("Save to DB", use_container_width=True):
+                try:
+                    insert_video_analysis(
+                        video_name=st.session_state.get("video_name", "unknown"),
+                        response=analysis,
+                        thumbnail=st.session_state.get("thumbnail", "")
                     )
-                else:
-                    st.info("No tabular data available for CSV export.")
-
-            with col2:
-                if st.button("Save to DB", use_container_width=True):
-                    try:
-                        insert_video_analysis(
-                            video_name=st.session_state.get("video_name", "unknown"),
-                            response=st.session_state["analysis"],
-                            thumbnail=st.session_state.get("thumbnail", "")
-                        )
-                        st.success("Analysis saved to database ")
-                    except Exception as e:
-                        st.error(f"Failed to save analysis: {e}")
+                    st.success("Analysis saved to database ")
+                except Exception as e:
+                    st.error(f"Failed to save analysis: {e}")
 
     else:
-        history_items = get_all_video_analyses(limit=20)
-        if history_items:
-            video_titles = [
-                f"{item['video_name']} ({item['created_at'].strftime('%Y-%m-%d %H:%M')})"
-                for item in history_items
-            ]
-            selected = st.sidebar.radio("History Items", video_titles, index=0)
-            idx = video_titles.index(selected)
-            selected_data = history_items[idx]
-
-            st.subheader(f"Analysis for: {selected_data['video_name']}")
-            if selected_data.get("thumbnail"):
-                st.image("data:image/jpeg;base64," + selected_data["thumbnail"], width=150)
-
-            json_response = selected_data.get("response")
-            if json_response:
-                tabs = st.tabs(["Video Analysis"])
-                with tabs[0]:
-                    render_analyzer_results(json_response)
-                    try:
-                        csv_data = analysis_to_csv(json_response)
-                        st.download_button(
-                            "Download CSV",
-                            data=csv_data,
-                            file_name=f"{selected_data['video_name']}_analysis.csv",
-                            mime="text/csv",
-                            use_container_width=True
-                        )
-                    except Exception as e:
-                        st.error(f"CSV export failed: {e}")
-        else:
+        # History tab
+        st.subheader(" History")
+        try:
+            history_items = get_all_video_analyses(limit=20)
+        except Exception as e:
+            st.error(f"Failed to load history: {e}")
+            return
+
+        if not history_items:
             st.info("No saved history available.")
+            return
+
+        video_titles = [
+            f"{item['video_name']} ({item['created_at'].strftime('%Y-%m-%d %H:%M')})"
+            for item in history_items
+        ]
+        selected = st.sidebar.radio("History Items", video_titles, index=0)
+        idx = video_titles.index(selected)
+        selected_data = history_items[idx]
+
+        st.subheader(f"Analysis for: {selected_data['video_name']}")
+        if selected_data.get("thumbnail"):
+            st.image("data:image/jpeg;base64," + selected_data["thumbnail"], width=150)
+
+        json_response = selected_data.get("response")
+        if not json_response:
+            st.info("No analysis payload found for this item.")
+            return
+
+        tabs = st.tabs(["Video Analysis"])
+        with tabs[0]:
+            render_analyzer_results(json_response)
+            try:
+                csv_data = analysis_to_csv(json_response)
+                st.download_button(
+                    "Download CSV",
+                    data=csv_data,
+                    file_name=f"{selected_data['video_name']}_analysis.csv",
+                    mime="text/csv",
+                    use_container_width=True,
+                )
+            except Exception as e:
+                st.error(f"CSV export failed: {e}")