Spaces:
Running
Running
| import streamlit as st | |
| st.set_page_config( | |
| page_title="Home", | |
| page_icon="🏠", | |
| layout="wide" | |
| ) | |
| from transcript_extractor import test_api_key, initialize_youtube_api | |
| import logging | |
| import os | |
| import sys | |
| # Configure logging to use only stdout | |
| logging.basicConfig( | |
| level=logging.INFO, | |
| format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', | |
| stream=sys.stdout # This ensures logging goes to stdout only | |
| ) | |
| logger = logging.getLogger(__name__) | |
| def main(): | |
| st.title("YouTube Transcript RAG System 🎥") | |
| st.write("Welcome to the YouTube Transcript RAG System!") | |
| # Check API key | |
| if not test_api_key(): | |
| st.error("YouTube API key is invalid or not set. Please check your configuration.") | |
| new_api_key = st.text_input("Enter your YouTube API key:") | |
| if new_api_key: | |
| os.environ['YOUTUBE_API_KEY'] = new_api_key | |
| if test_api_key(): | |
| st.success("API key validated successfully!") | |
| st.experimental_rerun() | |
| else: | |
| st.error("Invalid API key. Please try again.") | |
| return | |
| st.success("System is ready! Please use the sidebar to navigate between different functions.") | |
| # Display system overview | |
| st.header("System Overview") | |
| st.write(""" | |
| This system provides the following functionality: | |
| 1. **Data Ingestion** 📥 | |
| - Process YouTube videos and transcripts | |
| - Support for single videos or entire channels | |
| 2. **Chat Interface** 💬 | |
| - Interactive chat with processed videos | |
| - Multiple query rewriting methods | |
| - Various search strategies | |
| 3. **Ground Truth Generation** 📝 | |
| - Generate and manage ground truth questions | |
| - Export ground truth data | |
| 4. **RAG Evaluation** 📊 | |
| - Evaluate system performance | |
| - View detailed metrics and analytics | |
| """) | |
| if __name__ == "__main__": | |
| main() |