Create app.py

app.py

@@ -0,0 +1,72 @@
+import streamlit as st
+import os
+from utils.auth import authenticate, generate_token, verify_token
+from utils.database import initialize_vector_db, add_to_collection
+from utils.rag_utils import process_pdf, get_rag_response
+from huggingface_hub import hf_hub_download
+import time
+
+# Configuration
+st.set_page_config(page_title="RAG Chatbot", layout="wide")
+os.environ["TOKENIZERS_PARALLELISM"] = "false"
+
+# Initialize FAISS vector DB
+if 'vector_db' not in st.session_state:
+    with st.spinner("Loading AI model..."):
+        st.session_state.vector_db = initialize_vector_db()
+
+# Authentication
+def login_section():
+    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.token = generate_token(username)
+            st.session_state.logged_in = True
+            st.sidebar.success("Logged in successfully!")
+            time.sleep(1)
+            st.rerun()
+        else:
+            st.sidebar.error("Invalid credentials")
+
+# Main App
+def main_app():
+    st.title("📚 RAG Chatbot with PDF Processing")
+    
+    col1, col2 = st.columns([1, 2])
+    
+    with col1:
+        st.subheader("Upload PDF")
+        uploaded_file = st.file_uploader("Choose file", type="pdf", label_visibility="collapsed")
+        if uploaded_file:
+            with st.spinner("Processing PDF..."):
+                chunks = process_pdf(uploaded_file)
+                if add_to_collection(chunks):
+                    st.success(f"Processed {len(chunks)} text chunks!")
+    
+    with col2:
+        st.subheader("Chat Interface")
+        if prompt := st.chat_input("Ask about your documents"):
+            with st.spinner("Generating answer..."):
+                response = get_rag_response(prompt, st.session_state.vector_db)
+                st.chat_message("user").write(prompt)
+                st.chat_message("assistant").write(response)
+
+# Authentication flow
+if "logged_in" not in st.session_state:
+    st.session_state.logged_in = False
+
+if st.session_state.logged_in:
+    if verify_token(st.session_state.get("token", "")):
+        main_app()
+        if st.sidebar.button("Logout"):
+            st.session_state.logged_in = False
+            st.rerun()
+    else:
+        st.error("Session expired")
+        st.session_state.logged_in = False
+        login_section()
+else:
+    login_section()