Update app.py

app.py

@@ -1,4 +1,7 @@
-import os, torch
+# ---------- 0. Imports & constants ----------
+import os
+import torch
+import gradio as gr
 from pathlib import Path
 from huggingface_hub import login
 
@@ -10,7 +13,6 @@ from llama_index.llms.huggingface import HuggingFaceLLM
 from llama_index.embeddings.langchain import LangchainEmbedding
 from langchain_huggingface import HuggingFaceEmbeddings
 
-# ---------- Constants ----------
 SYSTEM_PROMPT = """
 You are a friendly café assistant for Café Eleven. Your job is to:
 1. Greet the customer warmly
@@ -18,6 +20,7 @@ You are a friendly café assistant for Café Eleven. Your job is to:
 3. Answer questions about ingredients, preparation, etc.
 4. Process special requests (allergies, modifications)
 5. Provide a friendly farewell
+6. Provide calories if they ask.
 Always be polite and helpful!
 """
 
@@ -77,12 +80,15 @@ def chat_with_cafe_eleven(message: str) -> str:
     response = engine.chat(message).response
     return response
 
-# ---------- Example usage ----------
+# ---------- 4. Gradio UI ----------
+iface = gr.Interface(
+    fn=chat_with_cafe_eleven,
+    inputs=gr.Textbox(lines=2, placeholder="Ask about menu items, orders, etc..."),
+    outputs="text",
+    title="Café Eleven Assistant",
+    description="A friendly café assistant to help you with orders and questions!"
+)
+
+# ---------- 5. Launch App ----------
 if __name__ == "__main__":
-    while True:
-        user_message = input("You: ")
-        bot_response = chat_with_cafe_eleven(user_message)
-        print("Café Eleven:", bot_response)
-        
-        if user_message.lower().strip() in {"quit", "exit", "done"}:
-            break
+    iface.launch(server_name="0.0.0.0", server_port=7860)