Update app.py

app.py

@@ -58,42 +58,119 @@ def get_top_chunks(query): # store a function that gets the most relevant_info a
 client = InferenceClient("google/gemma-3-27b-it")
 
 def respond(message, history):
+    # Get relevant information from knowledge base
+    relevant_chunks = get_top_chunks(message)
+    context = "\n".join(relevant_chunks)
+
+    # Build the system message including context
+    system_message = (
+        "You are a helpful chatbot named Scooby — kind of like the cartoon character but not too much. "
+        "You specialize in pets and their diets and ONLY answer questions about pets.\n\n"
+        "Relevant Info:\n" + context
+    )
+
+    # Construct chat messages
+    messages = [{"role": "system", "content": system_message}]
+    if history:
+        messages.extend(history)
+    messages.append({"role": "user", "content": message})
+
+    # Streaming response
+    response = ""
+    for chunk in client.chat_completion(messages, max_tokens=2500, stream=True):
+        token = chunk.choices[0].delta.content or ""
+        response += token
+        yield response
+
+
+#def respond(message, history):
 #use created variable for relevant info\
 
-    system_message = "You are a helpful chatbot named scooby, kinda like the cartoon character but not too much like it, who knows alot about pets and their diets and loves helping pet owners, and can only answer questions about pets"
-    messages = [{"role":  "system", "content": system_message}]
+ #   system_message = "You are a helpful chatbot named scooby, kinda like the cartoon character but not too much like it, who knows alot about pets and their diets and loves helping pet owners, and can only answer questions about pets"
+  #  messages = [{"role":  "system", "content": system_message}]
 
     
 #change HOW the bot responds
       
-    if history:
-        messages.extend(history)
+   # if history:
+    #    messages.extend(history)
         
-    messages.append({"role": "user", "content": message})
-    response = ""
-    for messages in client.chat_completion(messages,max_tokens = 2500, stream = True):
-        token = messages.choices[0].delta.content
-        response += token
-        yield response
+    #messages.append({"role": "user", "content": message})
+    #response = ""
+    #for messages in client.chat_completion(messages,max_tokens = 2500, stream = True):
+     #   token = messages.choices[0].delta.content
+      #  response += token
+       # yield response
 
 #theme = gr.themes.Origin(primary_hue="orange",secondary_hue="indigo", neutral_hue="teal")
 
     
 #chatbot = gr.ChatInterface(respond, type = "messages", theme)
 
-with gr.Blocks(theme=gr.themes.Origin(primary_hue="orange", secondary_hue="indigo", neutral_hue="teal")) as demo:
-    gr.Markdown("## 🐶 Scooby the Pet Expert")
-
+#with gr.Blocks(theme=gr.themes.Origin(primary_hue="orange", secondary_hue="indigo", neutral_hue="teal")) as demo:
+ #   gr.Markdown("## 🐶 Scooby the Pet Expert")
+
+  #  with gr.Row():
+   #     text_input = gr.Textbox(label="Ask Scooby a question about your pet")
+    #    image_input = gr.Image(type="pil", label="Upload an image of your pet (optional)")
+     #   gr.Image(value="logo.jpg", width=200, show_label=False, show_share_button = False, show_download_button = False, container=False)
+
+   # output = gr.Textbox(label="Scooby's Answer")
+
+    #submit_btn = gr.Button("Ask Scooby")
+
+    #submit_btn.click(fn=respond, inputs=[text_input, image_input], outputs=output)
+
+welcome_message = """
+# Welcome to ScoobyAI!
+## ScoobyAI is perfect for pet owners! We know it can be overwhelming to take care of a pet, so our chatbot Scooby is perfect for you!
+"""
+
+topics = """
+### Scooby is happy to help you with these topics!
+- Feeding Schedules
+- Pet health issues
+- Food Recipes for Pets
+"""
+created_by_messages = """
+### Created by Anavi, Ananya, Nithyla
+"""
+theme = gr.themes.Monochrome(
+    primary_hue="teal",
+    secondary_hue="orange",
+).set(
+    background_fill_primary='*primary_200',
+    background_fill_primary_dark='*primary_200',
+    background_fill_secondary='*secondary_300',
+    background_fill_secondary_dark='*secondary_300',
+    border_color_accent='*secondary_200',
+    border_color_accent_dark='*secondary_600',
+    border_color_accent_subdued='*secondary_200',
+    border_color_primary='*secondary_300',
+    block_border_color='*secondary_200',
+    button_primary_background_fill='*secondary_300',
+    button_primary_background_fill_dark='*secondary_300'
+)
+
+
+
+
+with gr.Blocks(theme=theme) as demo:
+    gr.Image("logo.jpg", container=False, show_share_button=False, show_download_button=False, label="output", show_label=True, elem_id="output_image")
+    gr.Markdown(welcome_message)
+    
     with gr.Row():
-        text_input = gr.Textbox(label="Ask Scooby a question about your pet")
-        image_input = gr.Image(type="pil", label="Upload an image of your pet (optional)")
-        gr.Image(value="logo.jpg", width=200, show_label=False, show_share_button = False, show_download_button = False, container=False)
+        with gr.Column():
+            gr.Markdown(topics)
+        with gr.Column():
+            question = gr.Textbox(label="Your question", placeholder="What do you want to ask Scooby?")
+            answer = gr.Textbox(label="Scooby's Response", placeholder="Scooby will respond here...", interactive=False, lines=17)
+            submit_button = gr.Button("Submit")
+            submit_button.click(fn=respond, inputs=question, outputs=answer, stream=True)
+
 
-    output = gr.Textbox(label="Scooby's Answer")
 
-    submit_btn = gr.Button("Ask Scooby")
 
-    submit_btn.click(fn=respond, inputs=[text_input, image_input], outputs=output)
 
 demo.launch(debug=True)