Gracefully disable YouTube in Cloud Mode

app.py

@@ -411,6 +411,16 @@ if "expanded_topics" not in st.session_state: st.session_state.expanded_topics =
 if "show_analytics" not in st.session_state: st.session_state.show_analytics = False
 if "topic_scores" not in st.session_state: st.session_state.topic_scores = {}  # Track quiz performance by topic_id
 
+if "app_config" not in st.session_state:
+    try:
+        resp = requests.get(f"{API_URL}/config", timeout=5)
+        if resp.status_code == 200:
+            st.session_state.app_config = resp.json()
+        else:
+            st.session_state.app_config = {"youtube_enabled": True}
+    except Exception:
+        st.session_state.app_config = {"youtube_enabled": True}
+
 # Helper function to add auth headers to all API requests
 def get_headers():
     """Get auth headers for API requests — uses Firebase token if available."""
@@ -1138,57 +1148,75 @@ if not st.session_state.focus_mode:
 
 
             # URL Input
-            with st.expander("+ Add URL / YouTube"):
-                url_input = st.text_input("URL", placeholder="https://youtube.com/... or any webpage", label_visibility="collapsed")
-                if st.button("Process URL", use_container_width=True):
-                    if not url_input:
-                        st.warning("Please enter a URL")
-                    else:
-                        import re as re_mod
-                        is_youtube = "youtube.com" in url_input or "youtu.be" in url_input
-                        
-                        if is_youtube:
-                            # Extract video ID
-                            vid_match = re_mod.search(r'(?:v=|youtu\.be/|shorts/|embed/)([a-zA-Z0-9_-]{11})', url_input)
-                            if not vid_match:
-                                st.error("❌ Invalid YouTube URL format. Supported: youtube.com/watch?v=ID, youtu.be/ID, or youtube.com/shorts/ID")
+            youtube_enabled = st.session_state.get("app_config", {}).get("youtube_enabled", True)
+
+            if youtube_enabled:
+                with st.expander("+ Add URL / YouTube"):
+                    url_input = st.text_input("URL", placeholder="https://youtube.com/... or any webpage", label_visibility="collapsed")
+                    if st.button("Process URL", use_container_width=True):
+                        if not url_input:
+                            st.warning("Please enter a URL")
+                        else:
+                            import re as re_mod
+                            is_youtube = "youtube.com" in url_input or "youtu.be" in url_input
+                            
+                            if is_youtube:
+                                # Extract video ID
+                                vid_match = re_mod.search(r'(?:v=|youtu\.be/|shorts/|embed/)([a-zA-Z0-9_-]{11})', url_input)
+                                if not vid_match:
+                                    st.error("❌ Invalid YouTube URL format. Supported: youtube.com/watch?v=ID, youtu.be/ID, or youtube.com/shorts/ID")
+                                else:
+                                    video_id = vid_match.group(1)
+                                    with st.spinner("⏳ Fetching transcript via Invidious..."):
+                                        try:
+                                            resp = requests.post(f"{API_URL}/ingest_youtube", json={"video_id": video_id}, headers=get_headers(), timeout=120)
+                                            if resp.status_code == 200:
+                                                st.success("✅ YouTube transcript processed successfully!")
+                                                time.sleep(1)
+                                                st.rerun()
+                                            else:
+                                                error_detail = resp.json().get('detail', resp.text)
+                                                if "No captions available" in str(error_detail):
+                                                    st.error("❌ No captions found. Try a video with CC enabled.")
+                                                elif "Could not reach any transcript" in str(error_detail):
+                                                    st.error("⚠️ Transcript service unavailable. Try again later.")
+                                                else:
+                                                    st.error(f"Failed: {error_detail}")
+                                        except requests.Timeout:
+                                            st.error("⏱️ Request timed out. Please try again.")
+                                        except Exception as e:
+                                            st.error(f"Error: {str(e)}")
                             else:
-                                video_id = vid_match.group(1)
-                                with st.spinner("⏳ Fetching transcript via Invidious..."):
+                                # Non-YouTube URL: use server-side ingestion
+                                with st.spinner("Fetching content..."):
                                     try:
-                                        resp = requests.post(f"{API_URL}/ingest_youtube", json={"video_id": video_id}, headers=get_headers(), timeout=120)
+                                        resp = requests.post(f"{API_URL}/ingest_url", json={"url": url_input}, headers=get_headers(), timeout=120)
                                         if resp.status_code == 200:
-                                            st.success("✅ YouTube transcript processed successfully!")
+                                            st.success(f"✅ {resp.json().get('message', 'Content added!')}")
                                             time.sleep(1)
                                             st.rerun()
                                         else:
                                             error_detail = resp.json().get('detail', resp.text)
-                                            if "No captions available" in str(error_detail):
-                                                st.error("❌ No captions found. Try a video with CC enabled.")
-                                            elif "Could not reach any transcript" in str(error_detail):
-                                                st.error("⚠️ Transcript service unavailable. Try again later.")
-                                            else:
-                                                st.error(f"Failed: {error_detail}")
+                                            st.error(f"Failed: {error_detail}")
                                     except requests.Timeout:
                                         st.error("⏱️ Request timed out. Please try again.")
                                     except Exception as e:
                                         st.error(f"Error: {str(e)}")
-                        else:
-                            # Non-YouTube URL: use server-side ingestion
-                            with st.spinner("Fetching content..."):
-                                try:
-                                    resp = requests.post(f"{API_URL}/ingest_url", json={"url": url_input}, headers=get_headers(), timeout=120)
-                                    if resp.status_code == 200:
-                                        st.success(f"✅ {resp.json().get('message', 'Content added!')}")
-                                        time.sleep(1)
-                                        st.rerun()
-                                    else:
-                                        error_detail = resp.json().get('detail', resp.text)
-                                        st.error(f"Failed: {error_detail}")
-                                except requests.Timeout:
-                                    st.error("⏱️ Request timed out. Please try again.")
-                                except Exception as e:
-                                    st.error(f"Error: {str(e)}")
+            else:
+                with st.expander("▶️ Add YouTube Video"):
+                    st.info(
+                        "⚠️ **YouTube is only available in local mode.**\n\n"
+                        "HuggingFace Spaces blocks outbound network requests "
+                        "which prevents YouTube transcript fetching.\n\n"
+                        "**Alternatives:**\n"
+                        "- 📄 Upload a PDF of your notes instead\n"
+                        "- 💻 Run FocusFlow locally to use YouTube\n"
+                        "- 📋 Paste text directly (coming soon)"
+                    )
+                    st.markdown(
+                        "[▶️ How to run locally](https://github.com/thesivarohith/focusflow#local-setup)",
+                        unsafe_allow_html=False
+                    )
 
 # --- FOCUS MODE UI ---
 if st.session_state.focus_mode:

backend/config.py

@@ -13,6 +13,11 @@ class LLMProvider(Enum):
 # Read from environment variable, default to Ollama (local)
 USE_PROVIDER = os.getenv("LLM_PROVIDER", "ollama").lower()
 
+# Set DEPLOYMENT_MODE=cloud in HuggingFace Spaces secrets
+# Leave unset or set to "local" for local development
+DEPLOYMENT_MODE = os.environ.get("DEPLOYMENT_MODE", "local").lower()
+IS_CLOUD = DEPLOYMENT_MODE == "cloud"
+
 # Configuration for both providers
 CONFIG = {
     "llm_provider": LLMProvider.OLLAMA if USE_PROVIDER == "ollama" else LLMProvider.HUGGINGFACE,

backend/main.py

@@ -20,6 +20,15 @@ def health_check():
     """Health check endpoint"""
     return {"status": "healthy"}
 
+@app.get("/config")
+async def get_config():
+    from backend.config import IS_CLOUD, DEPLOYMENT_MODE
+    return {
+        "is_cloud": IS_CLOUD,
+        "deployment_mode": DEPLOYMENT_MODE,
+        "youtube_enabled": not IS_CLOUD
+    }
+
 # Dependency
 def get_db():
     db = SessionLocal()

backend/rag_engine.py

@@ -26,6 +26,13 @@ INVIDIOUS_INSTANCES = [
 
 def get_youtube_transcript(video_id: str) -> str:
     """Fetch YouTube transcripts via the Invidious API to bypass network blocks."""
+    from backend.config import IS_CLOUD
+    if IS_CLOUD:
+        raise ValueError(
+            "YouTube is not available in cloud mode. "
+            "Please upload a PDF instead."
+        )
+
     # Step 1: Try each Invidious instance until one works
     captions_data = None
     last_error = None

push_all.sh

@@ -0,0 +1,23 @@
+#!/bin/bash
+# A script to easily commit and push to both GitHub and HuggingFace
+
+if [ -z "$1" ]; then
+    echo "Usage: ./push_all.sh \"Commit message\""
+    exit 1
+fi
+
+COMMIT_MSG=$1
+
+echo "Staging changes..."
+git add .
+
+echo "Committing with message: $COMMIT_MSG"
+git commit -m "$COMMIT_MSG"
+
+echo "Pushing to HuggingFace (origin)..."
+git push origin main
+
+echo "Pushing to GitHub (github)..."
+git push github main
+
+echo "All done!"