Merge pull request #3 from gcapuzzi/test

Test

Author: gcapuzzi

.gitignore

@@ -33,4 +33,3 @@ dataset*/
 responses.txt
 secrets.toml
 users.db
-homepage.py

config.yaml.example

@@ -0,0 +1,19 @@
+llm_provider: "" # Options: "openai" or "mistral"
+model_name: "" # for OpenAI: "gpt-3.5-turbo" or "gpt-4", or "gpt-4-turbo", for Mistral: "mistral-medium" or "mistral-large-latest" or "mistral-small-2506"
+embedding_model: "" # "text-embedding-ada-002" or "text-embedding-3-small" for OpenAI or "mistral-embed" for Mistral
+use_query_rewriting: false # enable or disable query rewriting
+dataset_private_path: "./dataset/private"
+dataset_public_path: "./dataset/public"
+web_urls: "web_urls"
+yt_video_links: "yt_video_links"
+text_files: "text_files"
+pdf_files: "pdf_files"
+rtdocs_files: "rtdocs_files"
+html_files: "html_files"
+persist_directory: "faiss_index"
+host: "127.0.0.1"
+port: 8080
+system_prompt: ""
+query_rewriting_prompt: ""
+logo_pth: ""
+company_name: ""

homepage.py

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+
+from utils import load_yaml_file
+
+# Read config data
+config_data = load_yaml_file("config.yaml")
+
+def gethomepage():
+    markdown_string = f"""
+<div style="text-align: center;">
+  <img src={config_data["logo_pth"]} alt="AIFAQ logo" style="height: 110px">
+  <h1>AI Agent powered by <em>{config_data["company_name"]}</em></h1>
+</div>
+
+## Try for free
+
+Here the [Official GitHub Repository](https://github.com/hyperledger-labs/aifaq)
+
+## Links and Resources
+
+AIFAQ Inc. [website](https://aifaqpro.wordpress.com/)
+
+## {config_data["company_name"]}
+
+This chatbot is a {config_data["company_name"]} conversational AI tool. Please, register or login:
+"""
+    return markdown_string

ingest_process.py

@@ -157,7 +157,7 @@ def get_vectordb(owner: str, access: str, datasetdir: str):
   if config_data["llm_provider"] == "mistral":
     embeddings = MistralAIEmbeddings(model=config_data["embedding_model"], mistral_api_key=mistral_api_key)
   else:  # default to OpenAI
-    embeddings = OpenAIEmbeddings(openai_api_key=openai_api_key)
+    embeddings = OpenAIEmbeddings(openai_api_key=openai_api_key, model=config_data["embedding_model"])
 
   if documents:
     # Create the vector store 

main.py

@@ -35,7 +35,8 @@ def get_ragchain(filter):
         )
     else:  # default to OpenAI
         embeddings = OpenAIEmbeddings(
-            openai_api_key=openai_api_key
+            openai_api_key=openai_api_key,
+            model=config_data["embedding_model"]
         )
         model = ChatOpenAI(
             openai_api_key=openai_api_key,

menu.py

@@ -1,4 +1,8 @@
 import streamlit as st
+from utils import load_yaml_file
+
+# Read config data
+config_data = load_yaml_file("config.yaml")
 
 def handle_logout():
     """Handle logout action and state cleanup"""
@@ -14,7 +18,8 @@ def authenticated_menu():
     handle_logout()
 
     # Show a navigation menu for authenticated users
-    st.sidebar.page_link("pages/chatbot.py", label="AIFAQ ChatBot")
+    chatbot_label = config_data["company_name"] + " ChatBot"
+    st.sidebar.page_link("pages/chatbot.py", label=chatbot_label)
     if st.session_state.user_type in ["admin"]:
         st.sidebar.page_link("pages/config_page_public.py", label="Config Public Page")
         st.sidebar.page_link("pages/config_page_private.py", label="Config Private Page")

pages/chatbot.py

@@ -11,12 +11,14 @@
 # Redirect to app.py if not logged in
 menu_with_redirect()
 
-st.markdown("# AIFAQ")
-
 config_path = "./config.yaml"
 logo_path = "https://github.com/hyperledger-labs/aifaq/blob/mvt-streamlit/images/logo.png?raw=true"
 config_data = load_yaml_file(config_path)
 
+title = config_data["company_name"]
+title = "# " + title if title else "# AIFAQ"
+st.markdown(title)
+
 # filter public document in case of guest user
 filter = None
 if st.session_state.user_type in ['guest']: