Update main.py

main.py

@@ -1,9 +1,9 @@
-
+import os
 import streamlit as st
 import google.generativeai as genai
 
-# Set your API key
-os.environ["GEMINI_API_KEY"] = "AIzaSyA6SC9JEIneWVcks_AEuacuF-L49HSt-qo"
+# Set your API key (you can also set this via Hugging Face environment settings for security)
+os.environ["GEMINI_API_KEY"] = st.secrets["GEMINI_API_KEY"]
 
 # Configure the Google Generative AI SDK with your API key
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
@@ -24,9 +24,7 @@ model = genai.GenerativeModel(
 )
 
 # Initialize a chat session
-chat_session = model.start_chat(
-    history=[]
-)
+chat_session = model.start_chat(history=[])
 
 # Streamlit app setup
 st.title("Gemini 1.5 Chatbot")
@@ -43,7 +41,3 @@ if user_input:
     # Display the response
     st.write("Gemini 1.5 says:")
     st.write(response.text)
-
-# Start Streamlit app
-# Save the file as `app.py`, and run the Streamlit app using:
-# streamlit run app.py