Update app.py

app.py

@@ -4,52 +4,154 @@ import tempfile
 import os
 import time
 
+# Set page configuration
+st.set_page_config(
+    page_title="AI Dance Partner",
+    page_icon="💃",
+    layout="wide",
+    initial_sidebar_state="expanded"
+)
+
+# Custom CSS for better styling
+def local_css():
+    st.markdown("""
+        <style>
+        .main {
+            padding: 2rem;
+        }
+        .stButton>button {
+            background-color: #FF4B4B;
+            color: white;
+            border-radius: 20px;
+            padding: 0.5rem 2rem;
+            font-weight: bold;
+        }
+        .stButton>button:hover {
+            background-color: #FF6B6B;
+            border-color: #FF4B4B;
+        }
+        .upload-text {
+            font-size: 1.2rem;
+            color: #666;
+            margin-bottom: 1rem;
+        }
+        .title-container {
+            background: linear-gradient(90deg, #FF4B4B, #FF8C8C);
+            padding: 2rem;
+            border-radius: 10px;
+            margin-bottom: 2rem;
+            color: white;
+            text-align: center;
+        }
+        .info-box {
+            background-color: #f0f2f6;
+            padding: 1rem;
+            border-radius: 10px;
+            margin-bottom: 1rem;
+        }
+        </style>
+    """, unsafe_allow_html=True)
+
 def main():
-    st.title("AI Dance Partner")
-    st.write("Upload your dance video and get an AI dance partner!")
-    
-    # Video upload
-    uploaded_file = st.file_uploader("Choose a video file", type=['mp4', 'avi', 'mov'])
+    local_css()
     
+    # Title section with gradient background
+    st.markdown("""
+        <div class="title-container">
+            <h1>🕺 AI Dance Partner 💃</h1>
+            <p style="font-size: 1.2rem;">Transform your solo dance into a dynamic duet!</p>
+        </div>
+    """, unsafe_allow_html=True)
+
+    # Create two columns for layout
+    col1, col2 = st.columns([2, 1])
+
+    with col1:
+        st.markdown('<p class="upload-text">Upload your dance video and watch the magic happen!</p>', unsafe_allow_html=True)
+        uploaded_file = st.file_uploader("", type=['mp4', 'avi', 'mov'])
+
+    with col2:
+        st.markdown('<div class="info-box">', unsafe_allow_html=True)
+        st.markdown("### How it works")
+        st.markdown("""
+        1. Upload your solo dance video
+        2. Choose your preferred dance style
+        3. Watch as AI creates your perfect dance partner!
+        """)
+        st.markdown('</div>', unsafe_allow_html=True)
+
     if uploaded_file is not None:
         # Create a temporary file
         with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as tfile:
             tfile.write(uploaded_file.read())
             temp_input_path = tfile.name
         
-        # Dance style selection
-        style = st.selectbox(
-            "Choose dance partner style",
-            ["Sync Partner", "Creative Partner"]
+        # Style selection with custom design
+        st.markdown("### Choose Your Dance Partner Style")
+        style = st.select_slider(
+            "",
+            options=["Sync Partner", "Creative Partner"],
+            value="Sync Partner"
         )
         
-        if st.button("Generate Dance Partner"):
+        # Add description based on selected style
+        if style == "Sync Partner":
+            st.info("💫 Sync Partner will mirror your movements in perfect harmony.")
+        else:
+            st.info("🎨 Creative Partner will add its own artistic flair to your dance.")
+        
+        if st.button("Generate Dance Partner 🎬"):
             try:
-                with st.spinner("Processing your dance video..."):
-                    # Process video
-                    dance_partner = AIDancePartner()
-                    output_path = dance_partner.process_video(temp_input_path, mode=style)
-                    
-                    # Display result
-                    st.video(output_path)
-                    
-                    # Download button
+                # Create a progress bar
+                progress_bar = st.progress(0)
+                status_text = st.empty()
+                
+                # Processing steps
+                steps = ["Analyzing dance moves...", "Generating partner movements...", "Creating final video..."]
+                for i, step in enumerate(steps):
+                    status_text.text(step)
+                    progress_bar.progress((i + 1) * 33)
+                    time.sleep(1)
+                
+                # Process video
+                dance_partner = AIDancePartner()
+                output_path = dance_partner.process_video(temp_input_path, mode=style)
+                
+                # Update progress
+                progress_bar.progress(100)
+                status_text.text("Done! 🎉")
+                
+                # Display result in a nice container
+                st.markdown("### 🎥 Your Dance Duet")
+                st.video(output_path)
+                
+                # Download button with custom styling
+                col1, col2, col3 = st.columns([1, 2, 1])
+                with col2:
                     with open(output_path, 'rb') as file:
                         st.download_button(
-                            label="Download video",
+                            label="Download Your Dance Duet 📥",
                             data=file,
                             file_name="ai_dance_partner.mp4",
                             mime="video/mp4"
                         )
-                    
-                    # Cleanup temporary files
-                    os.unlink(temp_input_path)
-                    os.unlink(output_path)
-                    
+                
+                # Cleanup temporary files
+                os.unlink(temp_input_path)
+                os.unlink(output_path)
+                
             except Exception as e:
-                st.error(f"An error occurred: {str(e)}")
+                st.error(f"Oops! Something went wrong: {str(e)}")
                 if os.path.exists(temp_input_path):
                     os.unlink(temp_input_path)
 
+    # Add footer
+    st.markdown("""
+        ---
+        <div style="text-align: center; color: #666;">
+            Made with ❤️ by AI Dance Partner Team
+        </div>
+    """, unsafe_allow_html=True)
+
 if __name__ == "__main__":
     main()