Update app.py

app.py

@@ -1,23 +1,23 @@
 import streamlit as st
-from langchain_openai import ChatOpenAI
+from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.chains import ConversationChain
 from langchain.memory import ConversationBufferMemory
 
-# Sidebar for OpenAI key
+# Sidebar for Google API key
 st.sidebar.title("⚙️ Settings")
-api_key = st.sidebar.text_input("Enter your OpenAI API Key", type="password")
+api_key = st.sidebar.text_input("Enter your Google API Key", type="password")
 
 st.title("🤖 AI Conversational Data Science Tutor")
 
 if api_key:
-    # Initialize model
-    llm = ChatOpenAI(
-        model="gpt-4o-mini",
+    # Initialize Gemini model
+    llm = ChatGoogleGenerativeAI(
+        model="gemini-1.5-pro",
+        google_api_key=api_key,
         temperature=0.5,
-        openai_api_key=api_key
     )
 
-    # Memory for conversation awareness
+    # Memory for conversation
     if "memory" not in st.session_state:
         st.session_state.memory = ConversationBufferMemory(return_messages=True)
 
@@ -28,15 +28,17 @@ if api_key:
         verbose=False
     )
 
-    # User input
+    # Chat input
     user_input = st.chat_input("Ask a Data Science question...")
     if user_input:
         with st.chat_message("user"):
             st.write(user_input)
 
         with st.chat_message("assistant"):
-            response = conversation.predict(input=f"You are a Data Science Tutor. Only answer Data Science questions. User asked: {user_input}")
+            response = conversation.predict(
+                input=f"You are a Data Science Tutor. Only answer Data Science questions. User asked: {user_input}"
+            )
             st.write(response)
 
 else:
-    st.warning("🔑 Please enter your OpenAI API key in the sidebar to start.")
+    st.warning("🔑 Please enter your Google API key in the sidebar to start.")