Update app.py

app.py

@@ -6,6 +6,8 @@ from langchain_community.utilities import SQLDatabase
 from langchain_community.agent_toolkits import SQLDatabaseToolkit
 from langchain.agents import create_agent
 import markdown as md
+import base64
+
 # from google.colab import userdata
 
 # --- GLOBALS ---
@@ -26,11 +28,19 @@ PROVIDER_MODELS = {
     "perplexity": ["sonar-small-chat", "sonar-medium-chat", "sonar-large-chat"]
 }
 
+def encode_image(image_path):
+    with open(image_path, "rb") as image_file:
+        return base64.b64encode(image_file.read()).decode('utf-8')
+
+# Encode the images
+github_logo_encoded = encode_image("Images/github-logo.png")
+linkedin_logo_encoded = encode_image("Images/linkedin-logo.png")
+website_logo_encoded = encode_image("Images/ai-logo.png")
 
 def create_chatbot_interface():
     with gr.Blocks(theme=gr.themes.Ocean(font=[gr.themes.GoogleFont("Noto Sans")]),
                    css='footer {visibility: hidden}') as demo:
-        
+        gr.Markdown("# DB Assistant 📊👨🏻‍🏭")
         with gr.Accordion("📔 Description:", open=False):
                 gr.Markdown(md.description)
             
@@ -58,6 +68,9 @@ def create_chatbot_interface():
                 chatbot = gr.Chatbot()
                 msg = gr.Textbox(label="Enter your question")
                 clear = gr.Button("Clear")
+                
+            gr.HTML(md.footer.format(github_logo_encoded, linkedin_logo_encoded, website_logo_encoded))
+
 
         # --- FUNCTIONS ---