Update app.py

app.py

@@ -16,6 +16,7 @@ from langchain_community.llms import HuggingFaceEndpoint
 from langchain_openai import ChatOpenAI
 import torch
 import fitz
+from google.colab import userdata
 from dotenv import load_dotenv, set_key
 
 load_dotenv(dotenv_path=".env")
@@ -198,7 +199,7 @@ def setup_gradio_interface():
                           db_btn = gr.Button("Create vector database")
                       with gr.Row():
                               db_progress = gr.Textbox(value="Not initialized", show_label=False) # label="Vector database status",
-                      gr.Markdown("<style>body { font-size: 16px; }</style><b>Select Large Language Model (LLM) and input parameters</b>")
+                      gr.Markdown("<style>body { font-size: 16px; }</style><b>Step 2 - Select Large Language Model (LLM) and input parameters</b>")
                       with gr.Row():
                           llm_btn = gr.Radio(list_llm_simple, label="Available LLMs", value = list_llm_simple[0], type="index") # info="Select LLM", show_label=False
                       with gr.Row():
@@ -215,7 +216,7 @@ def setup_gradio_interface():
                               llm_progress = gr.Textbox(value="Not initialized", show_label=False) # label="Chatbot status",
 
                   with gr.Column():
-                      gr.Markdown("<b>Step 2 - Chat with your Document</b>")
+                      gr.Markdown("<b>Step 3 - Chat with your Document</b>")
                       chatbot = gr.Chatbot(height=505)
                       with gr.Accordion("Relevent context from the source document", open=False):
                           with gr.Row():