Update app.py

app.py

@@ -1,42 +1,5 @@
-import streamlit as st
-import os
-from utils.database import initialize_vector_db, add_to_collection, get_vector_db
-from utils.rag_utils import process_pdf, get_groq_response
+# ... (previous imports remain the same)
 
-# Configuration
-os.environ["TOKENIZERS_PARALLELISM"] = "false"
-PERSISTENT_DB_DIR = "chroma_db_storage"
-PDF_STORAGE_DIR = "data/pdf_files"
-
-# Initialize directories
-os.makedirs(PDF_STORAGE_DIR, exist_ok=True)
-os.makedirs(PERSISTENT_DB_DIR, exist_ok=True)
-
-# Initialize session state
-if 'vector_db' not in st.session_state:
-    st.session_state.vector_db = initialize_vector_db(PERSISTENT_DB_DIR)
-
-# PDF Processing
-def handle_pdf_upload():
-    st.subheader("📄 Document Management")
-    uploaded_file = st.file_uploader("Upload PDF", type="pdf")
-    
-    if uploaded_file:
-        # Save PDF permanently
-        pdf_path = os.path.join(PDF_STORAGE_DIR, uploaded_file.name)
-        with open(pdf_path, "wb") as f:
-            f.write(uploaded_file.getbuffer())
-        
-        # Process and store in vector DB
-        with st.spinner(f"Processing {uploaded_file.name}..."):
-            try:
-                chunks = process_pdf(pdf_path)
-                add_to_collection(chunks)
-                st.success(f"✅ Added {len(chunks)} chunks from {uploaded_file.name}")
-            except Exception as e:
-                st.error(f"Error: {str(e)}")
-
-# Chat Interface
 def handle_chat():
     st.subheader("💬 Document Chat")
     
@@ -59,21 +22,11 @@ def handle_chat():
                     response = get_groq_response(
                         prompt, 
                         get_vector_db(),
-                        model_name="mixtral-8x7b-32768"
+                        model_name="llama3-70b-8192"  # Updated model
                     )
                     st.markdown(response)
-                    st.session_state.messages.append({"role": "assistant", "content": response})
+                    st.session_state.messages.append(
+                        {"role": "assistant", "content": response}
+                    )
                 except Exception as e:
-                    st.error(f"Error: {str(e)}")
-
-# Main App
-def main():
-    st.title("📚 Smart Document Assistant")
-    tab1, tab2 = st.tabs(["Manage Documents", "Chat"])
-    with tab1:
-        handle_pdf_upload()
-    with tab2:
-        handle_chat()
-
-if __name__ == "__main__":
-    main()
+                    st.error(f"Please check the model name in rag_utils.py. Error: {str(e)}")