Upload 5 files

app.py

@@ -0,0 +1,117 @@
+import streamlit as st
+from signup import signup
+from login import login
+from query import run_query_app
+from streamlit_option_menu import option_menu
+
+# this the fucntion to show the greater value
+
+
+def myFUnc():
+    print('moazzam riaz')
+    i = 1
+    if i == 0:
+        print('the number is geater than the value')
+
+    else:
+        print('the number is not grate')
+
+# this the recursion function to calculate the factorial
+
+
+def thiFunc(n):
+
+    input('enter to calculate factorial', n)
+    if n == 0 or n == 1:
+        return 1
+
+    else:
+        return n * thiFunc(n - 1)
+
+# this the function to including inheritaNCE and method overloading polymorphism
+
+
+class Animal:
+
+    def sound():
+        print('aniaml sound')
+
+
+class Dog(Animal):
+
+    def sound():
+        print('boww  bpoww')
+
+
+# this class is with a constructor which will be called automaticlly when the object is created
+class Cat(Animal):
+
+    def sound():
+        print('meoww meoww')
+
+    def __init__(self, a, b):
+
+        self.a = a,
+        self.b = b,
+
+
+'''
+the concecpts of oop
+    inheritance--this concept is about the classes like we have class animanls now it will have sub classes like cat or dog this is inheritance(single level, multi level, hybrid)
+    polymorphism--two methods one is method overloading which means that the function name and parameters are similar other one is method overloading in which the function name is similar but parameters are different
+    encapsulation--the hiddin of data which can not be accessed from outside the class three types private, public and protected
+    classes--the blueprint containing all the infomartion of the particular rela world thing contain functuion or methods
+    abstraction--the function which do not have any return value or dont contain any info in the function all the implementstion is done from the parent or subclass
+    recursion--the recursion is all about that we created a function and calling it inside the function
+    arrays--the arrays are that contain lists or any info like numbers 1 -100 this can of three types 1d, 2d and 3d everyone ahve it's pwn functinality
+    objects--the objects are the instance of classes to call the fnctions
+    these concept are from oop which can be used from any proraming lang
+    where as python is a dynamic language
+    so there is many the langugae= uncluding pythobn etc
+'''
+# my name is moazzam riaz the student of university of lahore
+
+
+def main():
+
+    st.title("Document Query System")
+
+    # Step 1: Initialize session state variables
+    if 'username' not in st.session_state:
+        st.session_state['username'] = None
+    if 'login_successful' not in st.session_state:
+        st.session_state['login_successful'] = False
+
+    # Step 2: Check if user is logged in
+    if st.session_state['username'] is None:
+        # User is not logged in, display login and signup options
+        selection = option_menu(
+            menu_title="Main Menu",
+            options=["Login", "Signup"],
+            icons=["person", "person"],
+            menu_icon="cast",
+            default_index=1
+        )
+
+        if selection == "Login":
+            st.session_state['username'] = login()
+            if st.session_state['username']:
+                st.session_state['login_successful'] = True
+        elif selection == "Signup":
+            signup()
+
+    # Step 3: Check if user is logged in successfully
+    if 'login_successful' in st.session_state and st.session_state['login_successful']:
+        # User is logged in, display welcome message and query page
+        if 'username' in st.session_state and st.session_state['username']:
+            st.subheader(f"Welcome, {st.session_state['username']}!")
+            run_query_app(st.session_state['username'])
+
+
+if st.sidebar.button("Logout"):
+    st.session_state['username'] = None
+    st.session_state['login_successful'] = False
+    st.empty()  # Clear the contents of the page
+
+if __name__ == '__main__':
+    main()

login.py

@@ -0,0 +1,46 @@
+import streamlit as st
+import hashlib
+import sqlite3
+from query import run_query_app
+
+def create_user_table():
+    conn = sqlite3.connect('user_credentials.db')
+    c = conn.cursor()
+    c.execute('''
+        CREATE TABLE IF NOT EXISTS users (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            email TEXT UNIQUE,
+            username TEXT,
+            password TEXT
+        )
+    ''')
+    conn.commit()
+    conn.close()
+
+
+
+def login():
+    st.subheader("Login")
+
+    email = st.text_input("Email")
+    password = st.text_input("Password", type="password")
+    submit_button = st.button("Login")
+
+    if submit_button:
+        hashed_password = hashlib.sha256(password.encode()).hexdigest()
+        conn = sqlite3.connect('user_credentials.db')
+        c = conn.cursor()
+        c.execute("SELECT * FROM users WHERE email = ? AND password = ?", (email, hashed_password))
+        user = c.fetchone()
+        conn.close()
+
+        if user:
+            st.success("Login successful!")
+            return user[1]  # Return the username
+        else:
+            st.error("Invalid email or password.")
+            return None
+
+
+
+

query.py

@@ -0,0 +1,107 @@
+import streamlit as st
+import os
+import openai
+import PyPDF2
+from langchain.embeddings import OpenAIEmbeddings
+from langchain.vectorstores import Chroma
+from langchain import OpenAI
+from langchain import VectorDBQA
+from langchain.document_loaders import UnstructuredFileLoader, UnstructuredPDFLoader
+from langchain.text_splitter import CharacterTextSplitter
+import nltk
+from streamlit_chat import message
+
+nltk.download("punkt")
+
+
+def run_query_app(username):
+    openai_api_key = st.sidebar.text_input("OpenAI API Key", key="openai_api_key_input", type="password")
+
+    uploaded_file = st.file_uploader("Upload a file", type=['txt', 'pdf'], key="file_uploader")
+    if uploaded_file:
+        # Save the uploaded file
+        file_path = os.path.join('./uploaded_files', uploaded_file.name)
+        with open(file_path, "wb") as f:
+            f.write(uploaded_file.read())
+
+        # Initialize OpenAIEmbeddings
+
+        os.environ['OPENAI_API_KEY'] = openai_api_key
+
+        # Initialize OpenAIEmbeddings
+        embeddings = OpenAIEmbeddings(openai_api_key=os.environ['OPENAI_API_KEY'])
+
+        # Load the file as document
+        _, ext = os.path.splitext(file_path)
+        if ext == '.txt':
+            loader = UnstructuredFileLoader(file_path)
+        elif ext == '.pdf':
+            loader = UnstructuredPDFLoader(file_path)
+        else:
+            st.write("Unsupported file format.")
+            return
+
+        documents = loader.load()
+
+        # Split the documents into texts
+        text_splitter = CharacterTextSplitter(chunk_size=800, chunk_overlap=0)
+        texts = text_splitter.split_documents(documents)
+
+        # Create Chroma vectorstore from documents
+        doc_search = Chroma.from_documents(texts, embeddings)
+
+        # Initialize VectorDBQA
+        chain = VectorDBQA.from_chain_type(llm=OpenAI(), chain_type="stuff", vectorstore=doc_search)
+
+        if 'messages' not in st.session_state:
+            st.session_state['messages'] = []
+
+        if 'past' not in st.session_state:
+            st.session_state['past'] = []
+
+        if 'generated' not in st.session_state:
+            st.session_state['generated'] = []
+
+        def update_chat(messages, sender, text):
+            message = {'sender': sender, 'text': text}
+            messages.append(message)
+            return messages
+
+        def get_response(chain, messages):
+            input_text = [m['text'] for m in messages if m['sender'] == 'user']
+            result = chain.run(input_text[-1])
+            return result
+
+        def get_text():
+            input_text = st.text_input("You: ", key="input")
+            return input_text
+
+        query = get_text()
+        user_input = query
+
+        if st.button("Run Query"):
+            with st.spinner("Generating..."):
+                messages = st.session_state.get('messages', [])
+                messages = update_chat(messages, "user", query)
+                response = get_response(chain, messages)
+                messages = update_chat(messages, "assistant", response)
+                st.session_state['messages'] = messages
+                st.session_state['past'].append(query)
+                st.session_state['generated'].append(response)
+        if uploaded_file is not None:
+            message(f"You are chatting with {uploaded_file.name}. Ask anything about it?")
+        if st.session_state['generated']:
+
+            for i in range(len(st.session_state['generated']) - 1, -1, -1):
+                message(st.session_state['past'][i], is_user=True, key=str(i) + '_user')
+                message(st.session_state['generated'][i], key=str(i))
+
+            with st.expander("Show Messages"):
+                for i, msg in enumerate(st.session_state['messages']):
+                    if msg['sender'] == 'user':
+                        message("User", msg['text'], key=f"user_{i}")
+                    else:
+                        message("Assistant", msg['text'], key=f"assistant_{i}")
+
+    if __name__ == '__main__':
+        run_query_app()

signup.py

@@ -0,0 +1,22 @@
+import streamlit as st
+import hashlib
+import sqlite3
+
+def signup():
+    st.subheader("Signup")
+    email = st.text_input("Email")
+    username = st.text_input("Username")
+    password = st.text_input("Password", type="password")
+
+    if len(password) < 8:
+        st.warning("Password must be at least 8 characters long.")
+        return
+
+    if st.button("Signup"):
+        hashed_password = hashlib.sha256(password.encode()).hexdigest()
+        conn = sqlite3.connect('user_credentials.db')
+        c = conn.cursor()
+        c.execute("INSERT INTO users (email, username, password) VALUES (?, ?, ?)", (email, username, hashed_password))
+        conn.commit()
+        conn.close()
+        st.success("Signup successful! Please login.")