Update app.py

app.py

@@ -7,8 +7,6 @@ from huggingface_hub.inference._generated.types.chat_completion import ChatCompl
 MODEL = "nomiChroma3.1"
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-from huggingface_hub.inference._generated.types.chat_completion import ChatCompletionStreamOutput
-
 def respond(
     message,
     history: list[tuple[str, str]],
@@ -71,7 +69,20 @@ def respond(
         else:
             yield f"An error occurred: {e}"
 
-# Gradio interface setup
+# Custom CSS for light blue background
+custom_css = """
+    .gradio-container {
+        background-color: #e6f3ff !important;
+    }
+    .chat-window {
+        background-color: #f0f8ff !important;
+    }
+    .message-wrap {
+        background-color: #ffffff !important;
+    }
+"""
+
+# Gradio interface setup with custom theme
 demo = gr.ChatInterface(
     respond,
     additional_inputs=[
@@ -91,15 +102,21 @@ demo = gr.ChatInterface(
     ],
     title="Maritime Legal Compliance",
     description="This chatbot uses Fine-tuned LLAMA-3.1 model personalised specifically to provide assistance with Indian maritime legal queries.",
-    theme="soft",
+    theme=gr.themes.Soft(
+        primary_hue="blue",
+        secondary_hue="blue",
+        neutral_hue="blue",
+    ).set_background(color="#e6f3ff"),
     examples=[
         ["What are the key regulations governing ports in India?"],
         ["Explain the concept of cabotage in Indian maritime law."],
         ["What are the legal requirements for registering a vessel in India?"],
     ],
+    css=custom_css,
     cache_examples=False,
 )
 
 # Launch the Gradio app
 if __name__ == "__main__":
     demo.launch()
+