Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 import os
-import sys
 from utils.database import initialize_vector_db, add_to_collection
 from utils.rag_utils import process_pdf, get_groq_response
 from typing import Optional
@@ -8,62 +7,38 @@ from typing import Optional
 # Configure environment
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
-# ---- Authentication Setup with Fallback ----
+# ---- Authentication Setup ----
 try:
     from utils.auth import authenticate, generate_token, verify_token
     AUTH_ENABLED = True
 except ImportError as e:
-    st.warning(f"Authentication disabled due to: {str(e)}")
+    st.warning(f"Authentication disabled: {str(e)}")
     AUTH_ENABLED = False
     
-    # Fallback dummy auth functions
+    # Fallback auth functions
     def authenticate(*args, **kwargs): return True
     def generate_token(*args, **kwargs): return "dummy_token"
     def verify_token(*args, **kwargs): return True
 
-# ---- Session State Initialization ----
+# ---- Initialize Vector DB ----
 if 'vector_db' not in st.session_state:
-    with st.spinner("🔄 Initializing AI system..."):
+    with st.spinner("Initializing AI system..."):
         st.session_state.vector_db = initialize_vector_db()
 
-if 'logged_in' not in st.session_state:
-    st.session_state.logged_in = False if AUTH_ENABLED else True
-
-# ---- UI Components ----
-def login_section():
-    st.sidebar.title("🔐 Admin Login")
-    username = st.sidebar.text_input("Username")
-    password = st.sidebar.text_input("Password", type="password")
-    
-    if st.sidebar.button("Login"):
-        if authenticate(username, password):
-            st.session_state.token = generate_token(username)
-            st.session_state.logged_in = True
-            st.sidebar.success("Login successful!")
-            st.rerun()
-        else:
-            st.sidebar.error("Invalid credentials")
-
-def pdf_uploader():
-    st.subheader("📤 Upload PDF Documents")
-    uploaded_file = st.file_uploader(
-        "Choose a PDF file", 
-        type="pdf",
-        label_visibility="collapsed"
-    )
-    
+# ---- PDF Processing Section ----
+def handle_pdf_upload():
+    uploaded_file = st.file_uploader("Upload PDF", type="pdf")
     if uploaded_file:
-        with st.spinner("🔍 Extracting knowledge from PDF..."):
+        with st.spinner("Processing PDF..."):
             try:
                 chunks = process_pdf(uploaded_file)
                 if add_to_collection(chunks):
-                    st.success(f"✅ Processed {len(chunks)} text chunks!")
+                    st.success(f"Processed {len(chunks)} text chunks!")
             except Exception as e:
-                st.error(f"PDF processing failed: {str(e)}")
+                st.error(f"PDF processing error: {str(e)}")
 
-def chat_interface():
-    st.subheader("💬 Chat with Your Documents")
-    
+# ---- Chat Interface ----
+def handle_chat():
     if "messages" not in st.session_state:
         st.session_state.messages = []
     
@@ -77,34 +52,49 @@ def chat_interface():
             st.markdown(prompt)
         
         with st.chat_message("assistant"):
-            with st.spinner("🤔 Thinking..."):
+            with st.spinner("Generating response..."):
                 try:
-                    response = get_groq_response(prompt, st.session_state.vector_db)
+                    # Updated to use current Groq model
+                    response = get_groq_response(
+                        prompt, 
+                        st.session_state.vector_db,
+                        model_name="mixtral-8x7b-32768"  # Updated model name
+                    )
                     st.markdown(response)
                     st.session_state.messages.append({"role": "assistant", "content": response})
                 except Exception as e:
-                    st.error(f"Error generating response: {str(e)}")
+                    st.error(f"Chat error: {str(e)}")
 
-# ---- Main App Flow ----
+# ---- Main App ----
 def main_app():
     st.title("⚡ Groq-Powered RAG Chatbot")
-    st.caption("Upload PDFs and ask questions with lightning-fast responses")
     
-    tab1, tab2 = st.tabs(["📄 Documents", "💬 Chat"])
+    tab1, tab2 = st.tabs(["Documents", "Chat"])
     
     with tab1:
-        pdf_uploader()
+        handle_pdf_upload()
     
     with tab2:
-        chat_interface()
+        handle_chat()
     
-    if st.sidebar.button("🔄 Reset Chat"):
+    if st.sidebar.button("Reset Chat"):
         st.session_state.messages = []
         st.rerun()
 
 # ---- Authentication Flow ----
-if AUTH_ENABLED and not st.session_state.logged_in:
-    login_section()
+if AUTH_ENABLED and not st.session_state.get("logged_in", False):
+    st.sidebar.title("Login")
+    username = st.sidebar.text_input("Username")
+    password = st.sidebar.text_input("Password", type="password")
+    
+    if st.sidebar.button("Login"):
+        if authenticate(username, password):
+            st.session_state.logged_in = True
+            st.session_state.token = generate_token(username)
+            st.sidebar.success("Logged in!")
+            st.rerun()
+        else:
+            st.sidebar.error("Invalid credentials")
 else:
     if AUTH_ENABLED and not verify_token(st.session_state.get("token", "")):
         st.error("Session expired")
@@ -112,15 +102,14 @@ else:
         st.rerun()
     else:
         main_app()
-        if AUTH_ENABLED and st.sidebar.button("🚪 Logout"):
+        if AUTH_ENABLED and st.sidebar.button("Logout"):
             st.session_state.clear()
             st.rerun()
 
 # ---- Footer ----
 st.sidebar.markdown("---")
 st.sidebar.markdown("""
-**Technical Stack:**
-- Groq Cloud (Mixtral 8x7b)
+**Using:**
+- Groq Cloud (Mixtral)
 - FAISS Vector Store
-- Streamlit UI
 """)