Update app.py

app.py

@@ -1,12 +1,9 @@
+import os
 import streamlit as st
 from google.generativeai import GenerativeModel
 
-# Configure API key
-api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"
-model_name = "gemini-pro"
-
-# Create model object
-model = GenerativeModel(api_key=api_key, model_name=model_name)
+# Set API key as an environment variable
+os.environ['GOOGLE_API_KEY'] = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"
 
 # Create chatbot interface
 st.title("Gemini API Chatbot")
@@ -25,7 +22,10 @@ if user_input:
     # Display user message with markdown
     st.markdown(f"**You:** {user_input}")
 
-    # Get model response with start_chat method
+    # Create model object
+    model = GenerativeModel(model_name="gemini-pro")
+
+    # Get model response with generate_content method
     with st.spinner("Thinking..."):
         response = model.generate_content(chat_history)