Update app.py

app.py

@@ -450,6 +450,29 @@ def main():
             st.error("Please fill in at least the topic and key message fields.")
             return
 
+        # Initialize style configuration
+        style_config = ContentStyle(
+            font_size=font_size,
+            font_color=font_color,
+            background_effect=background_effect,
+            transition_effect=transition_effect,
+            text_animation=text_animation,
+            theme=theme,
+            layout=layout
+        )
+
+        # Collect preferences before trying to use them
+        preferences = {
+            'topic': topic,
+            'style': style,
+            'tone': tone,
+            'audience': audience,
+            'purpose': purpose,
+            'message': message,
+            'mood': mood,
+            'voice_type': voice_type
+        }
+
         try:
             # Create progress tracker
             update_progress = create_progress_tracker()
@@ -472,163 +495,28 @@ def main():
             # Display results
             update_progress(100, "Complete!")
             
-            # ... [display logic remains the same] ...
+            if video_path and os.path.exists(video_path):
+                try:
+                    video_file = open(video_path, 'rb')
+                    video_bytes = video_file.read()
+                    st.video(video_bytes)
+                    video_file.close()
+                    
+                    # Download button
+                    st.download_button(
+                        label="Download Video",
+                        data=video_bytes,
+                        file_name=f"content_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp4",
+                        mime="video/mp4"
+                    )
+                finally:
+                    if 'video_file' in locals():
+                        video_file.close()
+                    if os.path.exists(video_path):
+                        os.remove(video_path)
             
         except Exception as e:
             st.error(f"An error occurred: {str(e)}")
-        finally:
-            # Cleanup
-            if 'video_path' in locals() and video_path and os.path.exists(video_path):
-                try:
-                    os.remove(video_path)
-                except:
-                    pass
-
-        # Initialize style configuration
-        style_config = ContentStyle(
-            font_size=font_size,
-            font_color=font_color,
-            background_effect=background_effect,
-            transition_effect=transition_effect,
-            text_animation=text_animation,
-            theme=theme,
-            layout=layout
-        )
-
-        # Collect preferences
-        preferences = {
-            'topic': topic,
-            'style': style,
-            'tone': tone,
-            'audience': audience,
-            'purpose': purpose,
-            'message': message,
-            'mood': mood,
-            'voice_type': voice_type
-        }
-
-        # Show progress
-        with st.spinner("🎨 Generating your content..."):
-            try:
-                progress_bar = st.progress(0)
-                status_text = st.empty()  # For showing status updates
-                
-                # Initialize AI engine
-                status_text.text("Initializing AI engine...")
-                ai_engine = AIContentEngine()
-                progress_bar.progress(10)
-                
-                # Generate content with better error handling
-                status_text.text("Generating content...")
-                try:
-                    content_package = ai_engine.generate_content_package(preferences)
-                    progress_bar.progress(30)
-                except Exception as e:
-                    st.error(f"Error in content generation: {str(e)}")
-                    return
-                
-                # Preview generated content with validation
-                if all(key in content_package for key in ['main_content', 'quote', 'tips', 'call_to_action', 'hashtags']):
-                    status_text.text("Preparing content preview...")
-                    st.subheader("📝 Generated Content Preview")
-                    with st.expander("View Content Details", expanded=True):
-                        st.markdown("### Main Content")
-                        st.write(content_package['main_content'])
-                        
-                        st.markdown("### Quote")
-                        st.write(content_package['quote'])
-                        
-                        st.markdown("### Tips")
-                        st.write(content_package['tips'])
-                        
-                        st.markdown("### Call to Action")
-                        st.write(content_package['call_to_action'])
-                        
-                        st.markdown("### Hashtags")
-                        st.write(content_package['hashtags'])
-                    
-                    progress_bar.progress(50)
-                else:
-                    st.error("Content generation incomplete. Please try again.")
-                    return
-                
-                # Initialize video generator with progress tracking
-                status_text.text("Preparing video generator...")
-                video_gen = VideoGenerator()
-                progress_bar.progress(60)
-                
-                # Generate video with enhanced error handling
-                status_text.text("Creating video...")
-                try:
-                    video_path = video_gen.create_video(content_package, preferences, style_config)
-                    progress_bar.progress(90)
-                except Exception as e:
-                    st.error(f"Error in video generation: {str(e)}")
-                    return
-                
-                if video_path and os.path.exists(video_path):
-                    progress_bar.progress(100)
-                    status_text.text("Completed!")
-                    
-                    try:
-                        # Display video with error handling
-                        st.subheader("🎥 Generated Video")
-                        video_file = open(video_path, 'rb')
-                        video_bytes = video_file.read()
-                        video_file.close()  # Properly close the file
-                        
-                        if len(video_bytes) > 0:
-                            st.video(video_bytes)
-                            
-                            # Download button
-                            st.download_button(
-                                label="Download Video",
-                                data=video_bytes,
-                                file_name=f"content_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp4",
-                                mime="video/mp4"
-                            )
-                            
-                            # Analytics and metrics with validation
-                            st.subheader("📊 Content Analytics")
-                            col1, col2, col3 = st.columns(3)
-                            
-                            quality_score = ai_engine.analyze_content_quality(content_package['main_content'])
-                            
-                            with col1:
-                                st.metric("Content Quality Score", 
-                                         f"{quality_score:.2f}")
-                            with col2:
-                                video_duration = VideoFileClip(video_path).duration
-                                st.metric("Video Duration", f"{video_duration:.1f}s")
-                            with col3:
-                                engagement_score = "High" if quality_score > 0.7 else "Medium"
-                                st.metric("Estimated Engagement", engagement_score)
-                        else:
-                            st.error("Generated video file is empty. Please try again.")
-                    
-                    except Exception as e:
-                        st.error(f"Error displaying video: {str(e)}")
-                    
-                    finally:
-                        # Cleanup temporary files
-                        try:
-                            if 'video_file' in locals():
-                                video_file.close()
-                            if os.path.exists(video_path):
-                                os.remove(video_path)
-                        except Exception as e:
-                            st.warning(f"Warning: Could not clean up temporary files: {str(e)}")
-                else:
-                    st.error("Video generation failed. Please try again.")
-                    
-            except Exception as e:
-                st.error(f"An unexpected error occurred: {str(e)}")
-                if 'progress_bar' in locals():
-                    progress_bar.progress(0)
             
-            finally:
-                if 'status_text' in locals():
-                    status_text.empty()
-
 if __name__ == "__main__":
     main()