Update app.py

app.py

@@ -299,7 +299,7 @@ def get_rag_system():
     return LangChainRAGSystem()
 
 def main():
-    """Main Streamlit application with enhanced UI."""
+    """Main Streamlit application."""
     st.set_page_config(
         page_title="EduBot for iCodeGuru",
         page_icon="🎓",
@@ -307,316 +307,89 @@ def main():
         initial_sidebar_state="expanded"
     )
     
-    # Custom CSS for better styling
-    st.markdown("""
-    <style>
-    .main-header {
-        background: linear-gradient(90deg, #667eea 0%, #764ba2 100%);
-        padding: 2rem;
-        border-radius: 10px;
-        margin-bottom: 2rem;
-        text-align: center;
-    }
-    .main-header h1 {
-        color: white;
-        margin: 0;
-        font-size: 2.5rem;
-    }
-    .main-header p {
-        color: #f0f0f0;
-        margin: 0.5rem 0 0 0;
-        font-size: 1.1rem;
-    }
-    .metric-card {
-        background: #f8f9fa;
-        padding: 1rem;
-        border-radius: 10px;
-        border-left: 4px solid #667eea;
-        margin: 0.5rem 0;
-    }
-    .status-online {
-        color: #28a745;
-        font-weight: bold;
-    }
-    .status-offline {
-        color: #dc3545;
-        font-weight: bold;
-    }
-    .chat-input {
-        position: fixed;
-        bottom: 0;
-        background: white;
-        padding: 1rem;
-        border-top: 1px solid #e0e0e0;
-    }
-    .source-card {
-        background: #f8f9fa;
-        border: 1px solid #e9ecef;
-        border-radius: 8px;
-        padding: 0.8rem;
-        margin: 0.3rem 0;
-    }
-    .quick-action-btn {
-        background: linear-gradient(45deg, #667eea, #764ba2);
-        color: white;
-        border: none;
-        padding: 0.5rem 1rem;
-        border-radius: 20px;
-        margin: 0.2rem;
-    }
-    </style>
-    """, unsafe_allow_html=True)
-    
-    # Enhanced Header
-    st.markdown("""
-    <div class="main-header">
-        <h1>🎓 EduBot for iCodeGuru</h1>
-        <p>Powered by LangChain | Ask questions based on iCodeGuru's knowledge base</p>
-    </div>
-    """, unsafe_allow_html=True)
+    # Header
+    st.title("🎓 EduBot for @icodeguru0")
+    st.markdown("**Powered by LangChain** | Ask anything based on pre-loaded iCodeGuru knowledge.")
     
     # Initialize RAG system
     rag_system = get_rag_system()
     
-    # Enhanced Sidebar
+    # Sidebar for admin functions
     with st.sidebar:
-        st.markdown("### ⚙️ Control Panel")
-        
-        # Status indicator
-        if rag_system.vectorstore:
-            try:
-                doc_count = rag_system.vectorstore._collection.count()
-                if doc_count > 0:
-                    st.markdown('<p class="status-online">🟢 System Online</p>', unsafe_allow_html=True)
-                else:
-                    st.markdown('<p class="status-offline">🔴 No Knowledge Base</p>', unsafe_allow_html=True)
-            except:
-                st.markdown('<p class="status-offline">🟡 System Loading</p>', unsafe_allow_html=True)
-        
-        st.markdown("---")
-        
-        # Admin Actions
-        col1, col2 = st.columns(2)
-        with col1:
-            if st.button("🔄 Refresh KB", type="primary", use_container_width=True):
-                with st.spinner("Loading documents..."):
-                    success = rag_system.ingest_documents()
-                    if success:
-                        st.balloons()
-                        st.success("Knowledge base updated!")
-                        st.rerun()
+        st.header("⚙️ Admin Panel")
         
-        with col2:
-            if st.button("🗑️ Clear Chat", use_container_width=True):
-                rag_system.reset_conversation()
-                st.session_state.messages = []
-                st.rerun()
+        if st.button("🔄 Refresh Knowledge Base", type="primary"):
+            success = rag_system.ingest_documents()
+            if success:
+                st.balloons()
         
-        # Quick Actions
-        st.markdown("### ⚡ Quick Actions")
-        if st.button("📚 Browse Available Content", use_container_width=True):
-            st.session_state.show_topics = True
-        
-        if st.button("💡 Get Sample Question", use_container_width=True):
-            random_questions = [
-                "What tutorials are available in iCodeGuru?",
-                "Show me resources about web development",
-                "What programming languages does iCodeGuru cover?",
-                "Find me beginner-friendly content",
-                "What are the latest tutorials added?"
-            ]
-            import random
-            random_q = random.choice(random_questions)
-            st.session_state.messages.append({"role": "user", "content": f"🎲 {random_q}"})
-            st.rerun()
+        if st.button("🗑️ Clear Conversation"):
+            rag_system.reset_conversation()
         
         st.markdown("---")
+        st.subheader("📊 System Info")
         
-        # Enhanced System Info
-        st.markdown("### 📊 System Statistics")
-        
+        # Show vectorstore stats
         if rag_system.vectorstore:
             try:
                 doc_count = rag_system.vectorstore._collection.count()
-                st.markdown(f"""
-                <div class="metric-card">
-                    <strong>📄 Documents:</strong><br>
-                    <span style="font-size: 1.5rem; color: #667eea;">{doc_count}</span>
-                </div>
-                """, unsafe_allow_html=True)
+                st.metric("Documents in KB", doc_count)
             except:
-                st.markdown("""
-                <div class="metric-card">
-                    <strong>📄 Documents:</strong><br>
-                    <span style="color: #dc3545;">N/A</span>
-                </div>
-                """, unsafe_allow_html=True)
-        
-        # Conversation count
-        chat_count = len([msg for msg in st.session_state.get("messages", []) if msg["role"] == "user"])
-        st.markdown(f"""
-        <div class="metric-card">
-            <strong>💬 Questions Asked:</strong><br>
-            <span style="font-size: 1.5rem; color: #28a745;">{chat_count}</span>
-        </div>
-        """, unsafe_allow_html=True)
-        
-        st.markdown("---")
-        
-        # Tech Stack
-        st.markdown("### 🛠️ Technology Stack")
-        tech_stack = {
-            "🧠": "ChromaDB",
-            "⚡": "Groq LLM", 
-            "🔗": "LangChain",
-            "🎨": "Streamlit",
-            "🤗": "HuggingFace"
-        }
-        
-        for icon, tech in tech_stack.items():
-            st.markdown(f"{icon} **{tech}**")
+                st.metric("Documents in KB", "N/A")
         
         st.markdown("---")
-        st.markdown("*Built with ❤️ for iCodeGuru community*")
-    
-    # Main Content Area
-    # Initialize session state
-    if "messages" not in st.session_state:
-        st.session_state.messages = []
-    if "show_topics" not in st.session_state:
-        st.session_state.show_topics = False
+        st.caption("🧠 **ChromaDB** for vector storage")
+        st.caption("⚡ **Groq LLM** for answers")
+        st.caption("🔗 **LangChain** for orchestration")
     
-    # Welcome message for new users
-    if len(st.session_state.messages) == 0:
-        st.markdown("""
-        ### 👋 Welcome to iCodeGuru EduBot!
-        
-        I'm your knowledge assistant for iCodeGuru content. I can help you find:
-        - **Tutorials and courses** from iCodeGuru
-        - **Specific topics** and learning materials  
-        - **Resources and references** from the knowledge base
-        - **Answers** based on iCodeGuru's educational content
-        
-        🚀 **Ask me anything about iCodeGuru's content!**
-        """)
-        
-        # Sample questions
-        st.markdown("### 💡 Try these sample questions:")
-        sample_questions = [
-            "What tutorials are available?",
-            "Show me Python resources",
-            "Find web development content",
-            "What beginner courses exist?"
-        ]
-        
-        cols = st.columns(2)
-        for i, question in enumerate(sample_questions):
-            with cols[i % 2]:
-                if st.button(f"❓ {question}", key=f"sample_{i}", use_container_width=True):
-                    st.session_state.messages.append({"role": "user", "content": question})
-                    st.rerun()
-    
-    # Show topics if requested
-    if st.session_state.get("show_topics", False):
-        with st.expander("📚 Available iCodeGuru Content", expanded=True):
-            st.markdown("""
-            Based on iCodeGuru's knowledge base, I can help you find:
-            - **Video Tutorials** and course materials
-            - **Programming Languages** covered by iCodeGuru
-            - **Project Examples** and case studies  
-            - **Learning Paths** and structured content
-            - **Tips and Tricks** from iCodeGuru experts
-            - **Resource Links** and references
-            
-            *Ask me to find specific content from iCodeGuru's library!*
-            """)
-            if st.button("Close Content Browser"):
-                st.session_state.show_topics = False
-                st.rerun()
-    
-    # Chat History with Enhanced Styling
-    if st.session_state.messages:
-        st.markdown("### 💬 Conversation History")
-        
-        for i, message in enumerate(st.session_state.messages):
-            with st.chat_message(message["role"]):
-                # Enhanced message display
-                if message["role"] == "user":
-                    st.markdown(f"**You:** {message['content']}")
-                else:
-                    st.markdown(message["content"])
-                    
-                    # Enhanced sources display
-                    if "sources" in message and message["sources"]:
-                        with st.expander(f"📚 Sources ({len(message['sources'])} found)", expanded=False):
-                            for j, source in enumerate(message["sources"], 1):
-                                st.markdown(f"""
-                                <div class="source-card">
-                                    <strong>Source {j}:</strong><br>
-                                    {source}
-                                </div>
-                                """, unsafe_allow_html=True)
-    
-    # Enhanced User Input Area
+    # Main chat interface
     st.markdown("---")
     
-    # Input with suggestions
-    col1, col2 = st.columns([4, 1])
-    
-    with col1:
-        user_input = st.text_input(
-            "💬 Ask about iCodeGuru content:",
-            placeholder="e.g., Show me Python tutorials from iCodeGuru",
-            key="user_input"
-        )
-    
-    with col2:
-        send_button = st.button("Send 🚀", type="primary", use_container_width=True)
+    # Initialize session state for chat history
+    if "messages" not in st.session_state:
+        st.session_state.messages = []
     
-    # Alternative chat input (modern style)
-    if prompt := st.chat_input("💬 Ask about iCodeGuru's knowledge base..."):
-        user_input = prompt
-        send_button = True
+    # Display chat history
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
+            if "sources" in message and message["sources"]:
+                with st.expander("📚 Sources"):
+                    for i, source in enumerate(message["sources"], 1):
+                        st.markdown(f"**Source {i}:** {source}")
     
-    # Process user input
-    if (send_button and user_input) or prompt:
-        question = user_input if send_button else prompt
-        
-        # Add user message
-        st.session_state.messages.append({"role": "user", "content": question})
+    # User input
+    if prompt := st.chat_input("💬 Ask your question about iCodeGuru..."):
+        # Add user message to chat history
+        st.session_state.messages.append({"role": "user", "content": prompt})
         
         # Display user message
         with st.chat_message("user"):
-            st.markdown(f"**You:** {question}")
+            st.markdown(prompt)
         
-        # Get and display assistant response
+        # Get assistant response
         with st.chat_message("assistant"):
-            with st.spinner("🤔 Let me think about that..."):
-                response = rag_system.get_answer(question)
-                answer = response.get("answer", "I couldn't find that information in iCodeGuru's knowledge base.")
+            with st.spinner("🤔 Thinking..."):
+                response = rag_system.get_answer(prompt)
+                answer = response.get("answer", "No answer available.")
                 source_docs = response.get("source_documents", [])
                 
                 st.markdown(answer)
                 
-                # Enhanced sources display
+                # Show sources if available
                 if source_docs:
                     sources = []
-                    for doc in source_docs[:4]:  # Show top 4 sources
-                        source_file = doc.metadata.get('source_file', 'Unknown')
-                        content_preview = doc.page_content[:150] + "..." if len(doc.page_content) > 150 else doc.page_content
-                        sources.append(f"**{source_file}**\n{content_preview}")
+                    for doc in source_docs[:3]:  # Show top 3 sources
+                        source = doc.metadata.get('source_file', 'Unknown source')
+                        content_preview = doc.page_content[:100] + "..." if len(doc.page_content) > 100 else doc.page_content
+                        sources.append(f"{source}: {content_preview}")
                     
                     if sources:
-                        with st.expander(f"📚 Sources ({len(sources)} found)", expanded=False):
+                        with st.expander("📚 Sources"):
                             for i, source in enumerate(sources, 1):
-                                st.markdown(f"""
-                                <div class="source-card">
-                                    <strong>📄 Source {i}:</strong><br>
-                                    {source}
-                                </div>
-                                """, unsafe_allow_html=True)
+                                st.markdown(f"**Source {i}:** {source}")
                         
-                        # Add to session state
+                        # Add to session state with sources
                         st.session_state.messages.append({
                             "role": "assistant", 
                             "content": answer,
@@ -626,19 +399,6 @@ def main():
                         st.session_state.messages.append({"role": "assistant", "content": answer})
                 else:
                     st.session_state.messages.append({"role": "assistant", "content": answer})
-        
-        # Clear input and rerun
-        if send_button:
-            st.rerun()
-    
-    # Footer
-    st.markdown("---")
-    st.markdown("""
-    <div style="text-align: center; color: #666; padding: 1rem;">
-        <p>🎓 <strong>EduBot for iCodeGuru</strong> | Your gateway to iCodeGuru's knowledge</p>
-        <p>Made with ❤️ using Streamlit • LangChain • ChromaDB • Groq</p>
-    </div>
-    """, unsafe_allow_html=True)
 
 if __name__ == "__main__":
     main()