Panchamurthy_bhargavi / dish_making
bhargavipanchamurthy's picture
dish_making
0a5fa01 verified
raw
history blame contribute delete
No virus
958 Bytes
from langchain_google_genai import ChatGoogleGenerativeAI
from langchain_core.prompts import ChatPromptTemplate
import streamlit as st
import os
openai_api_k = os.getenv("api_key")
model = ChatGoogleGenerativeAI(temperature=0, model="gemini-1.5-pro",google_api_key=openai_api_k,max_tokens=160)
prompt = ChatPromptTemplate.from_messages([
("system", "You are a good chef. You are a helpful assistant that suggests how to cook a dish."),
("human", "Give a proper way with mentioning ingredients\n{input}")
])
st.title("Dish Maker")
st.write("Enter the dish you want to make:")
# User input
dish = st.text_area("Dish")
if st.button("Generate process"):
chain = prompt | model
response = chain.invoke({"input": dish})
if response:
st.write("Generated Cooking Process:")
st.write(response.content)
else:
st.write("No response from the model. Please try again.")