Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from huggingface_hub import InferenceClient | |
| from sentence_transformers import SentenceTransformer | |
| import torch | |
| # Load knowledge | |
| with open("recipesplease.txt", "r", encoding="utf-8") as file: | |
| knowledge = file.read() | |
| cleaned_chunks = [chunk.strip() for chunk in knowledge.strip().split("\n") if chunk.strip()] | |
| model = SentenceTransformer('all-MiniLM-L6-v2') | |
| chunk_embeddings = model.encode(cleaned_chunks, convert_to_tensor=True) | |
| def get_top_chunks(query): | |
| query_embedding = model.encode(query, convert_to_tensor=True) | |
| query_embedding_normalized = query_embedding / query_embedding.norm() | |
| similarities = torch.matmul(chunk_embeddings, query_embedding_normalized) | |
| top_indices = torch.topk(similarities, k=5).indices.tolist() | |
| return [cleaned_chunks[i] for i in top_indices] | |
| client = InferenceClient("Qwen/Qwen2.5-72B-Instruct") | |
| def respond(message, history, cuisine, dietary_restrictions, allergies, preferred_ingredient): | |
| response = "" | |
| top_chunks = get_top_chunks(message) | |
| top_chunks += get_top_chunks(cuisine) | |
| top_chunks += get_top_chunks(str(dietary_restrictions)) # convert list to str | |
| top_chunks += get_top_chunks(allergies) | |
| top_chunks += get_top_chunks(preferred_ingredient) | |
| print(top_chunks) | |
| messages = [ | |
| { | |
| "role": "system", | |
| "content": f""" | |
| You are BiteBot, a friendly recipe chatbot. Use only the recipes from the following content: {top_chunks}. | |
| The user prefers {cuisine} cuisine, has the following dietary restrictions: {dietary_restrictions}, and is allergic to: {allergies}. | |
| Suggest one suitable recipe that matches these preferences. Say something like: | |
| "Based on your preferences, would you like to try Elizabeth's Sweet Potato Casserole?" | |
| If they say yes: | |
| - First, share the ingredients. | |
| - Then ask if they'd like the instructions. | |
| - If they agree, provide the instructions. | |
| If they say no: | |
| - Ask if they'd like another recommendation. | |
| Always use recipes only from {top_chunks}. | |
| """ | |
| } | |
| ] | |
| if history: | |
| messages.extend(history) | |
| messages.append({"role": "user", "content": message}) | |
| stream = client.chat_completion( | |
| messages, | |
| max_tokens=700, | |
| temperature=1.5,top_p=0.7, | |
| stream=True, | |
| ) | |
| for message in stream: | |
| token = message.choices[0].delta.content | |
| if token is not None: | |
| response += token | |
| yield response | |
| logo="banner.png" | |
| theme = gr.themes.Monochrome( | |
| primary_hue="orange", | |
| secondary_hue="zinc", | |
| neutral_hue=gr.themes.Color(c100="rgba(255, 227.4411088400613, 206.9078947368421, 1)", c200="rgba(255, 229.53334184977007, 218.0921052631579, 1)", c300="rgba(255, 234.91658150229947, 213.6184210526316, 1)", c400="rgba(189.603125, 154.41663986650488, 133.88641721491229, 1)", c50="#f3d1bbff", c500="rgba(170.2125, 139.18781968574348, 118.70082236842106, 1)", c600="rgba(193.32187499999998, 129.35648241888094, 111.07528782894737, 1)", c700="rgba(184.13125000000002, 141.9707339039346, 106.60230263157897, 1)", c800="rgba(156.06796875, 104.12209005333418, 69.81988075657894, 1)", c900="rgba(156.39999999999998, 117.22008175779253, 80.2578947368421, 1)", c950="rgba(158.43203125, 125.1788770279765, 97.28282620614036, 1)"), | |
| text_size="sm", | |
| spacing_size="md", | |
| radius_size="sm", | |
| ).set( | |
| body_background_fill='*primary_50', | |
| body_background_fill_dark='*primary_50' | |
| ) | |
| with gr.Blocks(theme=theme) as chatbot: | |
| gr.Image( | |
| value="Henrietta.png", | |
| show_label=False, | |
| show_share_button = False, | |
| show_download_button = False) | |
| gr.Markdown("### 👋 Welcome to BiteBot!\nTell me your preferred **cuisine**, any **dietary restrictions**, and **allergies**, and I’ll help you figure out what to cook. You can ask questions like:\n- _“What should I make tonight?”_\n- _“I'm feeling like eating something spicy.”_\n- _“Give me a recipe extra cheesy”_") | |
| cuisine=gr.Textbox(label="cuisine") | |
| dietary_restrictions=gr.Dropdown(["Gluten-Free","Dairy-Free","Vegan","Vegetarian","Keto","Kosher","No Soy","No Seafood","No Pork","No Beef"], label="dietary restrictions", multiselect=True,info="you can select multiple!") | |
| allergies=gr.Textbox(label="allergies") | |
| preferred_ingredients=gr.Textbox(label="preferred ingredients") | |
| gr.ChatInterface( | |
| fn=respond, | |
| type="messages", additional_inputs=[cuisine,dietary_restrictions,allergies,preferred_ingredients] | |
| ) | |
| chatbot.launch() | |