Update app.py

app.py

@@ -128,16 +128,16 @@ def conversation(qa_chain, message, history):
         return qa_chain, history, "", 0, "", 0, "", 0
 
 def main():
-    st.title("PDF Chatbot")
+    st.sidebar.title("PDF Chatbot")
 
-    st.markdown("### Step 1 - Upload PDF documents and Initialize RAG pipeline")
-    uploaded_files = st.file_uploader("Upload PDF documents", type="pdf", accept_multiple_files=True)
+    st.sidebar.markdown("### Step 1 - Upload PDF documents and Initialize RAG pipeline")
+    uploaded_files = st.sidebar.file_uploader("Upload PDF documents", type="pdf", accept_multiple_files=True)
 
     if uploaded_files:
-        if st.button("Create vector database"):
+        if st.sidebar.button("Create vector database"):
             with st.spinner("Creating vector database..."):
                 vector_db, db_message = initialize_database(uploaded_files)
-                st.success(db_message)
+                st.sidebar.success(db_message)
                 st.session_state['vector_db'] = vector_db
 
     if 'vector_db' not in st.session_state:
@@ -146,16 +146,17 @@ def main():
     if 'qa_chain' not in st.session_state:
         st.session_state['qa_chain'] = None
 
-    st.markdown("### Select Large Language Model (LLM) and input parameters")
-    llm_option = st.radio("Available LLMs", list_llm_simple)
+    st.sidebar.markdown("### Select Large Language Model (LLM) and input parameters")
+    llm_option = st.sidebar.radio("Available LLMs", list_llm_simple)
 
-    if st.button("Initialize Question Answering Chatbot"):
+    if st.sidebar.button("Initialize Question Answering Chatbot"):
         with st.spinner("Initializing QA chatbot..."):
             qa_chain, llm_message = initialize_LLM(list_llm_simple.index(llm_option), st.session_state['vector_db'])
             st.session_state['qa_chain'] = qa_chain
-            st.success(llm_message)
+            st.sidebar.success(llm_message)
+
+    st.title("Chat with your Document")
 
-    st.markdown("### Step 2 - Chat with your Document")
     if st.session_state['qa_chain']:
         history = []
         message = st.text_input("Ask a question")