Update app.py

app.py

@@ -40,6 +40,37 @@ def respond(
     response = tokenizer.decode(outputs[0][input_ids.shape[-1]:], skip_special_tokens=True)
     yield response
 
+    # -----------------------------------------------------------
+# 1. Custom CSS for Logo Injection
+# The Gradio app uses the 'a.gradio-image' selector for the header.
+# We inject the logo as a background image on a custom element.
+# -----------------------------------------------------------
+custom_css = """
+/* Hide the default Gradio title text */
+#app > header > div.gradio-logo { visibility: hidden; }
+
+/* Create a new element with a visible background image */
+#custom_logo_container {
+    visibility: visible;
+    position: absolute;
+    top: 10px;
+    left: 10px;
+    width: 150px; /* Adjust size as needed */
+    height: 40px;
+    background-image: url('file=aeon-logo.png'); /* IMPORTANT: Use 'file=' prefix */
+    background-size: contain;
+    background-repeat: no-repeat;
+    background-position: left center;
+    z-index: 100;
+}
+"""
+
+# -----------------------------------------------------------
+# 2. Custom HTML for Logo Container
+# We insert a <div> into the header to hold the logo.
+# -----------------------------------------------------------
+custom_html = '<div id="custom_logo_container"></div>'
+
 # The rest of your Gradio ChatInterface code remains the same
 chatbot = gr.ChatInterface(
     respond,
@@ -55,7 +86,10 @@ chatbot = gr.ChatInterface(
             step=0.05,
             label="Top-p (nucleus sampling)",
         ),
+        
     ],
+    css=custom_css,
+    title=custom_html, # Inject the HTML element into the title area
 )
 
 if __name__ == "__main__":