Redesign Video Brain: Step 1=Enter URL, Step 2=Auto-analyze, Step 3=Ask follow-up questions

streamlit_app.py

@@ -32,6 +32,10 @@ if "youtube_url" not in st.session_state:
     st.session_state.youtube_url = ""
 if "video_loaded" not in st.session_state:
     st.session_state.video_loaded = False
+if "video_summary" not in st.session_state:
+    st.session_state.video_summary = None  # Stores the initial video analysis
+if "video_transcript" not in st.session_state:
+    st.session_state.video_transcript = ""
 if "product_ideas" not in st.session_state:
     st.session_state.product_ideas = []
 
@@ -594,59 +598,84 @@ else:
 
 
 # =====================================
-# UNIFIED SEARCH BOX (All elements inside)
+# UNIFIED SEARCH BOX (Hide for Video Brain - it has dedicated UI)
 # =====================================
-st.markdown('<div class="search-wrapper">', unsafe_allow_html=True)
-
-# Use a form so Enter key submits
-with st.form(key="search_form", clear_on_submit=False):
-    col1, col2, col3, col4 = st.columns([2, 8, 1, 1])
-    
-    with col1:
-        # Mode selector dropdown
-        mode_list = list(MODES.keys())
-        current_idx = mode_list.index(st.session_state.mode)
-        selected = st.selectbox(
-            "mode",
-            mode_list,
-            index=current_idx,
-            format_func=lambda x: f"{MODES[x]['icon']} {x}",
-            label_visibility="collapsed",
-            key="mode_select"
-        )
-    
-    with col2:
-        # Search input - Enter key will submit the form
-        query = st.text_input(
-            "search",
-            placeholder="Ask anything... (Press Enter to search)",
-            label_visibility="collapsed",
-            key="query_input"
-        )
-    
-    with col3:
-        # Placeholder for alignment
-        st.write("")
-    
-    with col4:
-        # Submit button
-        submit = st.form_submit_button("→", help="Search")
+if st.session_state.mode != "Video Brain":
+    st.markdown('<div class="search-wrapper">', unsafe_allow_html=True)
 
-# Handle mode change outside form
-if selected != st.session_state.mode:
-    st.session_state.mode = selected
-    st.rerun()
+    # Use a form so Enter key submits
+    with st.form(key="search_form", clear_on_submit=False):
+        col1, col2, col3, col4 = st.columns([2, 8, 1, 1])
+        
+        with col1:
+            # Mode selector dropdown
+            mode_list = list(MODES.keys())
+            current_idx = mode_list.index(st.session_state.mode)
+            selected = st.selectbox(
+                "mode",
+                mode_list,
+                index=current_idx,
+                format_func=lambda x: f"{MODES[x]['icon']} {x}",
+                label_visibility="collapsed",
+                key="mode_select"
+            )
+        
+        with col2:
+            # Search input - Enter key will submit the form
+            query = st.text_input(
+                "search",
+                placeholder="Ask anything... (Press Enter to search)",
+                label_visibility="collapsed",
+                key="query_input"
+            )
+        
+        with col3:
+            # Placeholder for alignment
+            st.write("")
+        
+        with col4:
+            # Submit button
+            submit = st.form_submit_button("→", help="Search")
+
+    # Handle mode change outside form
+    if selected != st.session_state.mode:
+        st.session_state.mode = selected
+        st.rerun()
+
+    # File upload button (outside form)
+    col_attach, col_spacer_attach = st.columns([1, 11])
+    with col_attach:
+        if st.button("📎 Attach files", key="attach_btn"):
+            st.session_state.show_upload = not st.session_state.show_upload
 
-# File upload button (outside form)
-col_attach, col_spacer_attach = st.columns([1, 11])
-with col_attach:
-    if st.button("📎 Attach files", key="attach_btn"):
-        st.session_state.show_upload = not st.session_state.show_upload
+    st.markdown('</div>', unsafe_allow_html=True)
 
-st.markdown('</div>', unsafe_allow_html=True)
+    # Mode description
+    st.markdown(f'<div class="mode-desc">{MODES[st.session_state.mode]["icon"]} {st.session_state.mode}: {MODES[st.session_state.mode]["desc"]}</div>', unsafe_allow_html=True)
 
-# Mode description
-st.markdown(f'<div class="mode-desc">{MODES[st.session_state.mode]["icon"]} {st.session_state.mode}: {MODES[st.session_state.mode]["desc"]}</div>', unsafe_allow_html=True)
+else:
+    # Video Brain mode - just show mode selector
+    mode_list = list(MODES.keys())
+    current_idx = mode_list.index(st.session_state.mode)
+    selected = st.selectbox(
+        "Switch Mode",
+        mode_list,
+        index=current_idx,
+        format_func=lambda x: f"{MODES[x]['icon']} {x}",
+        key="mode_select_video"
+    )
+    if selected != st.session_state.mode:
+        st.session_state.mode = selected
+        # Reset video state when switching away
+        st.session_state.video_summary = None
+        st.session_state.video_loaded = False
+        st.session_state.youtube_url = ""
+        st.session_state.video_transcript = ""
+        st.rerun()
+    
+    # Initialize these for Video Brain mode
+    query = ""
+    submit = False
 
 # =====================================
 # SPECIAL UI FOR PRODUCT MVP MODE
@@ -666,61 +695,165 @@ if st.session_state.mode == "Product MVP" and not st.session_state.current_resul
 # =====================================
 # SPECIAL UI FOR VIDEO BRAIN MODE
 # =====================================
-if "video_transcript" not in st.session_state:
-    st.session_state.video_transcript = ""
-
-if st.session_state.mode == "Video Brain" and not st.session_state.current_result:
+if st.session_state.mode == "Video Brain":
+    # Header
     st.markdown("""
     <div style="text-align: center; padding: 20px; margin: 20px auto; max-width: 700px; 
                 background: linear-gradient(135deg, #3B82F6 0%, #8B5CF6 100%); 
                 border-radius: 16px; color: white;">
         <h3 style="margin: 0; font-size: 24px;">🎥 Video Brain – Understand Any YouTube Video</h3>
-        <p style="margin: 10px 0 0; opacity: 0.9;">📋 Paste the video transcript for accurate analysis</p>
+        <p style="margin: 10px 0 0; opacity: 0.9;">Enter a YouTube URL to analyze the video</p>
     </div>
     """, unsafe_allow_html=True)
     
-    st.markdown("""
-    **How to use Video Brain:**
-    1. Open your YouTube video
-    2. Click **⋮** (three dots) → **Show transcript**
-    3. Copy the transcript text and paste below
-    4. Ask questions about the video!
-    """)
-    
-    # YouTube URL input (optional, for reference)
-    youtube_url = st.text_input(
-        "YouTube URL (optional)",
-        value=st.session_state.youtube_url,
-        placeholder="Paste YouTube URL here for reference",
-        key="youtube_url_input"
-    )
-    
-    if youtube_url and ("youtube.com" in youtube_url or "youtu.be" in youtube_url):
-        if youtube_url != st.session_state.youtube_url:
-            st.session_state.youtube_url = youtube_url
-            st.session_state.video_loaded = True
-    
-    # TRANSCRIPT INPUT - Main feature
-    st.markdown("### 📋 Paste Video Transcript")
-    transcript_input = st.text_area(
-        "Video Transcript",
-        value=st.session_state.video_transcript,
-        placeholder="Paste the YouTube transcript here...\n\nTo get it: Open YouTube video → Click ⋮ → Show transcript → Select All → Copy",
-        height=200,
-        key="transcript_input"
-    )
-    
-    if transcript_input and transcript_input != st.session_state.video_transcript:
-        st.session_state.video_transcript = transcript_input
-        st.session_state.video_loaded = True
+    # STEP 1: If no video loaded yet, show URL input
+    if not st.session_state.video_summary:
+        st.markdown("### Step 1: Enter YouTube URL")
+        
+        video_url_col, load_btn_col = st.columns([4, 1])
+        with video_url_col:
+            youtube_url_input = st.text_input(
+                "YouTube URL",
+                value=st.session_state.youtube_url,
+                placeholder="https://www.youtube.com/watch?v=...",
+                key="video_url_input",
+                label_visibility="collapsed"
+            )
+        
+        with load_btn_col:
+            load_video = st.button("🎬 Load", key="load_video_btn", use_container_width=True)
+        
+        # Also allow transcript paste as alternative
+        with st.expander("📋 Or paste transcript manually (if URL doesn't work)"):
+            transcript_input = st.text_area(
+                "Paste transcript here",
+                value=st.session_state.video_transcript,
+                placeholder="Open YouTube → Click ⋮ → Show transcript → Copy all text",
+                height=150,
+                key="transcript_paste_input"
+            )
+            if transcript_input != st.session_state.video_transcript:
+                st.session_state.video_transcript = transcript_input
+        
+        # Handle Load button click
+        if load_video and youtube_url_input:
+            if "youtube.com" in youtube_url_input or "youtu.be" in youtube_url_input:
+                st.session_state.youtube_url = youtube_url_input
+                
+                # Call API to analyze video
+                with st.spinner("🔄 Analyzing video... This may take a moment..."):
+                    payload = {
+                        "message": "Summarize this video completely. Give me: 1) What the video is about, 2) Key points and takeaways, 3) Important details mentioned, 4) Main conclusions",
+                        "workspace_id": WORKSPACE,
+                        "mode": "video_brain",
+                        "youtube_url": youtube_url_input,
+                        "transcript": st.session_state.video_transcript
+                    }
+                    try:
+                        response = requests.post(f"{API_URL}/api/video_brain", json=payload, timeout=180)
+                        if response.ok:
+                            result = response.json()
+                            st.session_state.video_summary = result
+                            st.session_state.video_loaded = True
+                            st.rerun()
+                        else:
+                            st.error(f"Error loading video: {response.text}")
+                    except Exception as e:
+                        st.error(f"Error: {str(e)}")
+            else:
+                st.warning("⚠️ Please enter a valid YouTube URL")
+        
+        # Also handle Enter key in URL input
+        if youtube_url_input and youtube_url_input != st.session_state.youtube_url:
+            if "youtube.com" in youtube_url_input or "youtu.be" in youtube_url_input:
+                st.session_state.youtube_url = youtube_url_input
     
-    if st.session_state.video_transcript:
-        word_count = len(st.session_state.video_transcript.split())
-        st.success(f"✅ Transcript loaded! ({word_count} words) - Now ask questions below.")
-    elif st.session_state.video_loaded and st.session_state.youtube_url:
-        st.info("💡 For best results, also paste the transcript above")
+    # STEP 2: Video is loaded - show summary and question input
     else:
-        st.info("👆 Paste the video transcript above to get started")
+        # Show video info
+        st.markdown(f"**📺 Loaded Video:** `{st.session_state.youtube_url}`")
+        
+        # Button to load different video
+        if st.button("🔄 Load Different Video", key="reset_video"):
+            st.session_state.video_summary = None
+            st.session_state.video_loaded = False
+            st.session_state.youtube_url = ""
+            st.session_state.video_transcript = ""
+            st.session_state.current_result = None
+            st.rerun()
+        
+        st.markdown("---")
+        
+        # Show the video summary
+        st.markdown("### 📝 Video Summary")
+        summary_data = st.session_state.video_summary
+        st.markdown(summary_data.get("answer", "No summary available"))
+        
+        # Show sources if available
+        if summary_data.get("sources"):
+            with st.expander("📚 Sources"):
+                for src in summary_data["sources"]:
+                    st.markdown(f"- [{src.get('title', 'Source')}]({src.get('url', '#')})")
+        
+        st.markdown("---")
+        
+        # STEP 3: Ask follow-up questions
+        st.markdown("### 💬 Ask Questions About This Video")
+        
+        followup_question = st.text_input(
+            "Your question",
+            placeholder="Ask anything about the video...",
+            key="video_followup_input",
+            label_visibility="collapsed"
+        )
+        
+        ask_btn = st.button("🔍 Ask", key="ask_followup_btn")
+        
+        # Quick question buttons
+        st.markdown("**Quick questions:**")
+        quick_q_cols = st.columns(3)
+        quick_questions = [
+            "What are the main points?",
+            "Explain in simple terms",
+            "What should I remember?"
+        ]
+        
+        selected_quick_q = None
+        for i, qq in enumerate(quick_questions):
+            with quick_q_cols[i]:
+                if st.button(qq, key=f"quick_q_{i}"):
+                    selected_quick_q = qq
+        
+        # Handle question submission
+        question_to_ask = followup_question if ask_btn and followup_question else selected_quick_q
+        
+        if question_to_ask:
+            with st.spinner("🤔 Thinking..."):
+                payload = {
+                    "message": question_to_ask,
+                    "workspace_id": WORKSPACE,
+                    "mode": "video_brain",
+                    "youtube_url": st.session_state.youtube_url,
+                    "transcript": st.session_state.video_transcript
+                }
+                try:
+                    response = requests.post(f"{API_URL}/api/video_brain", json=payload, timeout=180)
+                    if response.ok:
+                        result = response.json()
+                        st.session_state.current_result = {
+                            "query": question_to_ask,
+                            "mode": "Video Brain",
+                            "data": result
+                        }
+                        st.rerun()
+                except Exception as e:
+                    st.error(f"Error: {str(e)}")
+        
+        # Show previous Q&A if exists
+        if st.session_state.current_result and st.session_state.current_result.get("mode") == "Video Brain":
+            st.markdown("---")
+            st.markdown(f"**❓ Q:** {st.session_state.current_result['query']}")
+            st.markdown(f"**💡 A:** {st.session_state.current_result['data'].get('answer', '')}")
 
 # Show file uploader when icon is clicked
 if st.session_state.show_upload:
@@ -747,45 +880,12 @@ if st.session_state.uploaded_files:
 
 
 # =====================================
-# HANDLE SEARCH
+# HANDLE SEARCH (Skip for Video Brain - it has its own UI)
 # =====================================
-if submit and query.strip():
+if submit and query.strip() and st.session_state.mode != "Video Brain":
     extra_data = None
     actual_query = query.strip()
     
-    # For Video Brain mode, include the YouTube URL and transcript
-    if st.session_state.mode == "Video Brain":
-        # Check if query itself contains a YouTube URL
-        query_text = query.strip()
-        url_in_query = None
-        for word in query_text.split():
-            if "youtube.com" in word or "youtu.be" in word:
-                url_in_query = word
-                break
-        
-        if url_in_query:
-            # User pasted URL directly in the query
-            st.session_state.youtube_url = url_in_query
-            st.session_state.video_loaded = True
-            extra_data = {"youtube_url": url_in_query}
-            # If query is JUST the URL, ask a default question
-            if query_text == url_in_query:
-                actual_query = "Summarize this video and give me the key takeaways"
-        elif st.session_state.video_loaded and st.session_state.youtube_url:
-            extra_data = {"youtube_url": st.session_state.youtube_url}
-        elif st.session_state.get("video_transcript"):
-            # User has transcript but no URL - still allow it
-            extra_data = {"youtube_url": "user_provided_transcript"}
-        else:
-            st.warning("⚠️ Please paste a YouTube URL or transcript above!")
-            st.stop()
-        
-        # Include pasted transcript if available
-        if st.session_state.get("video_transcript"):
-            if extra_data is None:
-                extra_data = {}
-            extra_data["transcript"] = st.session_state.video_transcript
-    
     # For Product MVP mode, save to ideas history
     if st.session_state.mode == "Product MVP":
         st.session_state.product_ideas.append({