Upload 2 files

langchain_helper.py

@@ -0,0 +1,23 @@
+import getpass
+import os
+from langchain_core.messages import HumanMessage, SystemMessage
+from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain_core.prompts import PromptTemplate
+
+if "GOOGLE_API_KEY" not in os.environ:
+    os.environ["GOOGLE_API_KEY"] = getpass.getpass(
+        "Provide your Google API Key")
+
+try:
+    llm = ChatGoogleGenerativeAI(model="gemini-pro")
+    
+
+except Exception as e:
+    print(f"Error creating model: {e}")
+
+
+def generate_name(cuisine):
+    result = llm.invoke("Restaurant name generate a name for a {cuisine} restaurant. give in a response['restaurant_name'] and response['menu_items']. type format ")
+    print(result.content)
+
+    return result.content

main.py

@@ -0,0 +1,18 @@
+import streamlit as st
+from langchain_helper import generate_name
+import json
+
+st.title('Restaurant Name Generator with GEMINI')
+
+cuisine = st.sidebar.selectbox("Pick a Cuisine", ("Indian", "Italian", "Chinese", "Mexican", "American"))
+
+if cuisine:
+    response = generate_name(cuisine)
+    response_dict = json.loads(response)
+    
+
+    st.header(response_dict['restaurant_name'])
+
+    st.write("Menu Items:")
+    for item in response_dict['menu_items']:
+        st.write(f"- {item}")