Update app.py

app.py

@@ -6,6 +6,20 @@ from huggingface_hub import InferenceClient
 from sentence_transformers import SentenceTransformer
 import torch 
 
+theme = gr.themes.Soft(
+    primary_hue="rose",
+    secondary_hue="zinc",
+    neutral_hue="pink"
+    )
+
+custom_css = """
+:root { /* This applies to the light mode */
+    --background-fill-primary: *primary_100 !important; /* Light pink */
+}
+.dark { /* This applies to the dark mode */
+    --background-fill-primary: #FFB6C1 !important; /* Hot pink */
+}
+"""
 
 #STEP 2 FROM SEMANTIC SEARCH 
 # Open the water_cycle.txt file in read mode with UTF-8 encoding
@@ -102,7 +116,7 @@ client=InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 def respond(message, history):
     top_results= get_top_chunks(message, chunk_embeddings, cleaned_chunks)
     str_top_results= '\n'.join(top_results)
-    messages = [{'role':'system', 'content': f'You are a friend chatbot. Complete all your sentences with a period and do not cut your sentences off even if it is under the word limit. Only give one piece of information at a time, unless prompted for more. Base your response on the provided context:\n{str_top_results}'}]
+    messages = [{'role':'system', 'content': f'You are a friend chatbot. Complete all your sentences, and do not cut yourself off. The word limit is 100 words. Start off by only giving a career, and then if prompted by the user provide more information like salary, college course,etc. Base your response on the provided context:\n{str_top_results}'}]
     if history:
         messages.extend(history)
 
@@ -110,33 +124,19 @@ def respond(message, history):
 
     response = client.chat_completion(
         messages,
-        max_tokens = 150,
+        max_tokens = 1000,
         temperature = 0.2
     )
     return response['choices'][0]['message']['content'].strip()
 
 demo = gr.ChatInterface(respond, type="messages", autofocus=False)
 
-theme = gr.themes.Soft(
-    primary_hue="fuchsia",
-    secondary_hue="amber",
-    neutral_hue="teal"
-)
-
-# Your exact hex colors applied via CSS
-custom_css = """
-:root {
-  --color-primary-500: #CC338B;
-  --color-secondary-500: #EAA221;
-  --color-neutral-500: #43B3AE;
-}
+with gr.Blocks (theme = theme) as chatbot:
+    gr.ChatInterface(respond, type = "messages",
+    title = "Hi, I'm PathPilot! ",
+    textbox= gr.Textbox(placeholder="Share Your Age and Ask Me Anything!"),
+    description = "This tool is here to listen and provide information on STEM Careers, and all discussions will be kept confidential. ❤️‍🩹",
 
-.dark {
-  --color-primary-500: #CC338B;
-  --color-secondary-500: #EAA221;
-  --color-neutral-500: #43B3AE;
-}
-"""
 
 
 demo.launch()