Update utils/database.py

utils/database.py

@@ -16,99 +16,8 @@ from langchain_core.runnables import RunnablePassthrough
 import os
 import streamlit as st
 
-def create_connection(db_file):
-    try:
-        conn = sqlite3.connect(db_file)
-        return conn
-    except Error as e:
-        st.error(f"Error: {e}")
-    return None
-
-
-def create_tables(conn):
-    try:
-        sql_create_documents_table = """
-        CREATE TABLE IF NOT EXISTS documents (
-            id INTEGER PRIMARY KEY AUTOINCREMENT,
-            name TEXT NOT NULL,
-            content TEXT NOT NULL,
-            upload_date TIMESTAMP DEFAULT CURRENT_TIMESTAMP
-        );
-        """
-
-        sql_create_queries_table = """
-        CREATE TABLE IF NOT EXISTS queries (
-            id INTEGER PRIMARY KEY AUTOINCREMENT,
-            query TEXT NOT NULL,
-            response TEXT NOT NULL,
-            document_id INTEGER,
-            query_date TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
-            FOREIGN KEY (document_id) REFERENCES documents (id)
-        );
-        """
-
-        sql_create_annotations_table = """
-        CREATE TABLE IF NOT EXISTS annotations (
-            id INTEGER PRIMARY KEY AUTOINCREMENT,
-            document_id INTEGER NOT NULL,
-            annotation TEXT NOT NULL,
-            page_number INTEGER,
-            annotation_date TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
-            FOREIGN KEY (document_id) REFERENCES documents (id)
-        );
-        """
-
-        c = conn.cursor()
-        c.execute(sql_create_documents_table)
-        c.execute(sql_create_queries_table)
-        c.execute(sql_create_annotations_table)
-    except Error as e:
-        st.error(f"Error: {e}")
-
-
-def get_documents(conn):
-    """Retrieve documents from database"""
-    try:
-        cursor = conn.cursor()
-        cursor.execute(
-            "SELECT id, name, upload_date FROM documents ORDER BY upload_date DESC"
-        )
-        return cursor.fetchall()
-    except Exception as e:
-        st.error(f"Error retrieving documents: {e}")
-        return []
-
-
-def insert_document(conn, doc_name, doc_content):
-    """Insert a document into database"""
-    try:
-        cursor = conn.cursor()
-        cursor.execute("SELECT id FROM documents WHERE name = ?", (doc_name,))
-        if not cursor.fetchone():
-            conn.execute(
-                "INSERT INTO documents (name, content) VALUES (?, ?)",
-                (doc_name, doc_content),
-            )
-            return True
-    except Exception as e:
-        st.error(f"Error inserting document: {e}")
-    return False
-
-
-def format_chat_history(messages: list[BaseMessage]) -> list[dict]:
-    """Convert chat history to the format expected by langchain"""
-    formatted = []
-    for msg in messages:
-        if isinstance(msg, HumanMessage):
-            formatted.append({"role": "user", "content": msg.content})
-        elif isinstance(msg, AIMessage):
-            formatted.append({"role": "assistant", "content": msg.content})
-        elif isinstance(msg, SystemMessage):
-            formatted.append({"role": "system", "content": msg.content})
-    return formatted
-
 def initialize_qa_system(vector_store):
-
+    """Initialize QA system with proper chat handling"""
     try:
         llm = ChatOpenAI(
             temperature=0.5,
@@ -170,7 +79,7 @@ def initialize_qa_system(vector_store):
         return None
         
 def initialize_faiss(embeddings, documents, document_names):
-
+    """Initialize FAISS vector store"""
     try:
         from langchain.vectorstores import FAISS
 
@@ -187,7 +96,7 @@ def initialize_faiss(embeddings, documents, document_names):
 
 @st.cache_resource
 def get_embeddings_model():
-    
+    """Get the embeddings model"""
     try:
         from langchain.embeddings import HuggingFaceEmbeddings
 
@@ -197,13 +106,3 @@ def get_embeddings_model():
     except Exception as e:
         st.error(f"Error loading embeddings model: {e}")
         return None
-# Defining the file paths explicitly again before writing the updated content
-components_chat_path = 'components/chat.py'
-utils_database_path = 'utils/database.py'
-
-# Writing the updated contents back to the original files
-with open(components_chat_path, 'w') as chat_file:
-    chat_file.write(updated_components_chat_content)
-
-with open(utils_database_path, 'w') as database_file:
-    database_file.write(updated_utils_database_content)