🧠 TubeMind AI
AI-Powered YouTube Learning Assistant
Retrieval-Augmented Generation • Semantic Search • FAISS • Groq • Chrome Extension
🚀 Overview
TubeMind AI is an AI-powered Chrome Extension that transforms YouTube videos into an interactive learning experience.
Using Retrieval-Augmented Generation (RAG), semantic search, FAISS vector search, and Groq LLMs, TubeMind AI enables users to:
- 💬 Chat with YouTube videos
- 📄 Generate AI summaries
- 📝 Create structured study notes
- 💡 Review key takeaways
- 📍 Jump to timestamp citations
- 🧠 Learn faster with AI
✨ Features
🤖 AI Chat
- Multi-turn conversation memory
- Streaming AI responses
- Timestamp citations
- Markdown rendering
- Retry & Copy responses
📚 AI Learning
- Short Summary
- Detailed Summary
- Bullet Summary
- Study Notes
- Key Takeaways
🧠 RAG Pipeline
- Transcript Retrieval
- Semantic Chunking
- Sentence Transformers
- FAISS Vector Search
- Semantic Retrieval
- Prompt Engineering
🖼️ Screenshots
Add your screenshots to the assets/ folder.
🎥 Demo
🛠️ Tech Stack
AI Stack
- Groq
- Llama 3.3 70B
- Sentence Transformers
- FAISS
- Retrieval-Augmented Generation (RAG)
- Prompt Engineering
🏗️ Architecture
flowchart TD
A[YouTube Video] --> B[Transcript Retrieval]
B --> C[Semantic Chunking]
C --> D[Sentence Transformers]
D --> E[Embeddings]
E --> F[FAISS Index]
F --> G[Semantic Search]
G --> H[Prompt Builder]
H --> I[Groq LLM]
I --> J[Streaming Response]
J --> K[Chrome Extension]
📂 Project Structure
TubeMind-AI
│
├── assets
│ ├── banner.png
│ ├── demo.gif
│ ├── chat.png
│ ├── summary.png
│ ├── notes.png
│ └── takeaways.png
│
├── backend
├── extension
├── README.md
└── .gitignore
⚙️ Installation
Clone
git clone https://github.com/Omkar-narsale/TubeMind-AI.git
cd TubeMind-AI
Backend
cd backend
pip install -r requirements.txt
Create a .env
GROQ_API_KEY=YOUR_API_KEY
GROQ_MODEL=llama-3.3-70b-versatile
Run
uvicorn app.main:app --reload
Extension
cd extension
npm install
npm run build
Load the generated extension/dist folder from `chrome://