Update app.py

app.py

@@ -7,6 +7,8 @@ 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]],
@@ -69,36 +71,7 @@ def respond(
         else:
             yield f"An error occurred: {e}"
 
-# Custom CSS for light blue background and message tiles
-custom_css = """
-    .gradio-container {
-        background-color: #e6f3ff !important;
-    }
-    .chat-window {
-        background-color: #f0f8ff !important;
-    }
-    .message.user, .message.bot {
-        background-color: #e6f3ff !important;
-        border: 1px solid #cce4ff !important;
-        padding: 15px !important;
-        border-radius: 8px !important;
-    }
-    .message.user {
-        background-color: #e1f0ff !important;
-    }
-    .message.bot {
-        background-color: #e6f3ff !important;
-    }
-    .input-box, .output-box {
-        background-color: #e6f3ff !important;
-        border: 1px solid #cce4ff !important;
-    }
-    textarea {
-        background-color: #e6f3ff !important;
-    }
-"""
-
-# Gradio interface setup with custom theme
+# Gradio interface setup
 demo = gr.ChatInterface(
     respond,
     additional_inputs=[
@@ -118,20 +91,15 @@ 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=gr.themes.Soft(
-        primary_hue="blue",
-        secondary_hue="blue",
-        neutral_hue="blue",
-    ).set_background(color="#e6f3ff"),
+    theme="soft",
     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__":
+if _name_ == "_main_":
     demo.launch()