Update app.py

app.py

@@ -4,6 +4,7 @@ import numpy as np
 
 import gradio as gr
 from sentence_transformers import SentenceTransformer
+from gradio.themes import Soft
 
 # here we define our basic setup. we decided to store all nutrition text files inside a folder called data
 # this lets us update content later just by dropping new .txt files without changing any code
@@ -155,8 +156,8 @@ def build_rag_answer(question: str) -> str:
     return full_text
 
 
-# this is the function that the chat interface calls every time the user sends a new message
-# we ignore the chat history on purpose because our design focuses on single-turn rag questions
+# this is the core chat function. gradio will give us the message and the current chat history
+# in our design, we ignore the history for retrieval because we focus on single-turn questions
 def nutri_chat(message: str, history: list) -> str:
     if not message or not message.strip():
         return "please type a question about healthy eating to chat with nutribud."
@@ -167,21 +168,46 @@ def nutri_chat(message: str, history: list) -> str:
     return build_rag_answer(message)
 
 
-# here we use gradio's chatinterface so the layout feels like a normal chat app (similar to chatgpt)
-# we picked the soft theme to get a calm, slightly green look without needing custom css
-demo = gr.ChatInterface(
-    fn=nutri_chat,
-    title="🌿 nutribud — friendly nutrition rag chatbot",
-    description=(
-        "ask nutribud general questions about healthy eating and it will answer using trusted public health documents.\n"
-        "nutribud does not give medical advice or personalized meal plans.\n\n"
-        "example questions:\n"
-        "• how can i eat more vegetables every day?\n"
-        "• what are some healthier drink options than soda?\n"
-        "• what does a balanced plate usually look like?"
-    ),
-    theme="soft",
-)
+# here we build a chat-style ui using blocks and a chatbot component
+# this looks more like a normal chat app (similar to chatgpt) instead of a plain form
+green_theme = Soft(primary_hue="green", neutral_hue="green")
+
+with gr.Blocks(theme=green_theme) as demo:
+    gr.Markdown(
+        """
+# 🌿 nutribud
+
+chat with nutribud about general healthy eating questions.  
+nutribud is based on public health nutrition documents and does **not** give medical advice.
+
+try questions like:
+- how can i eat more vegetables every day?
+- what are some healthier drink options than soda?
+- what does a balanced plate usually look like?
+"""
+    )
+
+    chatbot = gr.Chatbot(label="nutribud chat")
+    with gr.Row():
+        msg = gr.Textbox(
+            show_label=False,
+            placeholder="type your nutrition question here and press enter...",
+            scale=4,
+        )
+        clear = gr.Button("clear chat", scale=1)
+
+    # this function connects our rag logic to the visual chat history
+    # we append the user message and nutribud’s reply as a pair so they show up as chat bubbles
+    def respond(message, history):
+        if history is None:
+            history = []
+        reply = nutri_chat(message, history)
+        history = history + [(message, reply)]
+        return "", history
+
+    msg.submit(respond, [msg, chatbot], [msg, chatbot])
+    clear.click(lambda: ([], ""), None, [chatbot, msg])
+
 
 if __name__ == "__main__":
     demo.launch()