Enhanced UI

app.py

@@ -82,18 +82,42 @@ def main():
         
         if file_type == 'image':
             results, image = process_image_in_memory(file)
-            st.image(image, caption="Uploaded Image", use_container_width=True, width=100)
-                    
-            display_detailed_analysis(results)
+            st.image(image, caption="Uploaded Image", width=500)
+            
+            if st.button("Analyze Image"):
+                with st.spinner("Analyzing..."):
+                    display_detailed_analysis(results)
                     
         elif file_type == 'video':
-            st.video(file)
+            col1, col2, col3 = st.columns([1, 2, 1])
+            with col2:
+                # Set a smaller width for the video container
+                st.markdown('<div style="width: 300px; margin: auto;">', unsafe_allow_html=True)
+                st.video(file, format="video/mp4", start_time=0)
+                st.markdown('</div>', unsafe_allow_html=True)
             
             if st.button("Analyze Video"):
                 with st.spinner("Analyzing video frames..."):
                     results = process_video_in_memory(file)
                     
-                    display_detailed_analysis(results)
+                    # Display results
+                    st.subheader("Video Analysis Results")
+                    col1, col2 = st.columns(2)
+                    
+                    with col1:
+                        st.metric("Final Prediction", results["Final Video Prediction"])
+                        st.metric("Confidence Score", f"{results['Confidence Score']*100:.2f}%")
+                    
+                    with col2:
+                        st.metric("Fake Frames", results["Fake Frames"])
+                        st.metric("Real Frames", results["Real Frames"])
+
+                    # Detailed frame analysis
+                    st.write("📊 Frame Analysis:")
+                    st.write(f"- **Total Frames Analyzed**: {results['Total Frames Analyzed']}")
+                    st.write(f"- **Fake Frames**: {results['Fake Frames']}")
+                    st.write(f"- **Real Frames**: {results['Real Frames']}")
+                    st.write(f"- **Fake Content Percentage**: {results['Fake Percentage']}%")
 
 if __name__ == "__main__":
     main()