Customize graphical UI:

- Include avatars
- Include greetings message
- Improve readability

app.py

@@ -1,4 +1,6 @@
-# https://python.langchain.com/docs/tutorials/rag/
+# AI assistant with a RAG system to query information from the CAMELS cosmological simulations using Langchain
+# Author: Pablo Villanueva Domingo
+
 import gradio as gr
 from langchain import hub
 from langchain_chroma import Chroma
@@ -50,6 +52,8 @@ urls = get_subpages(base_url)
 loader = WebBaseLoader(urls)
 docs = loader.load()
 
+print("Pages loaded:",len(docs))
+
 # Join content pages for processing
 def format_docs(docs):
     return "\n\n".join(doc.page_content for doc in docs)
@@ -91,6 +95,7 @@ embeddings = HuggingFaceInstructEmbeddings(model_name=embed_model)
 # RAG chain
 rag_chain = RAG(llm, docs, embeddings)
 
+# Function to handle prompt and query the RAG chain
 def handle_prompt(message, history):
     try:
         # Stream output
@@ -101,7 +106,9 @@ def handle_prompt(message, history):
     except:
         raise gr.Error("Requests rate limit exceeded")
 
-greetingsmessage = "Hi, I'm the CAMELS DocBot, I'm here to assist you with any question related to the CAMELS simulations documentation"
+# Predefined messages and examples
+description = "AI powered assistant which answers any question related to the [CAMELS simulations](https://www.camel-simulations.org/)."
+greetingsmessage = "Hi, I'm the CAMELS DocBot, I'm here to assist you with any question related to the CAMELS simulations."
 example_questions = [
                     "How can I read a halo file?",
                     "Which simulation suites are included in CAMELS?",
@@ -109,7 +116,19 @@ example_questions = [
                     "Write a complete snippet of code getting the power spectrum of a simulation"
                      ]
 
+# Define customized Gradio chatbot
+chatbot = gr.Chatbot([{"role":"assistant", "content":greetingsmessage}],
+                     type="messages",
+                     avatar_images=["ims/userpic.png","ims/camelslogo.jpg"],
+                     height="60vh")
+
 # Define Gradio interface
-demo = gr.ChatInterface(handle_prompt, type="messages", title="CAMELS DocBot", examples=example_questions, theme=gr.themes.Soft(), description=greetingsmessage)#, chatbot=chatbot)
+demo = gr.ChatInterface(handle_prompt,
+                        type="messages",
+                        title="CAMELS DocBot",
+                        examples=example_questions,
+                        theme=gr.themes.Soft(),
+                        description=description,
+                        chatbot=chatbot)
 
 demo.launch()