api key inserted

app.py

@@ -1,8 +1,17 @@
 import streamlit as st
 import pandas as pd
+import google.generativeai as genai  # Import Generative AI library
+import os
 from pymongo import MongoClient
 from db import insert_data_if_empty, get_mongo_client  # Import functions from db.py
-import google.generativeai as genai  # Import Generative AI library
+
+# 🔑 Fetch API key from Hugging Face Secrets
+GEMINI_API_KEY = os.getenv("gemini_api")
+
+if GEMINI_API_KEY:
+    genai.configure(api_key=GEMINI_API_KEY)
+else:
+    st.error("⚠️ Google API key is missing! Set it in Hugging Face Secrets.")
 
 #### **1. Ensure Data is Inserted Before Display**
 insert_data_if_empty()
@@ -30,19 +39,17 @@ if st.button("Show Complete Data"):
 #### **4. GenAI Chatbot Interface**
 st.subheader("🤖 AI Chatbot")
 
-# Configure Generative AI model (replace `your-api-key` with actual key)
-genai.configure(api_key="your-api-key")  # Set up your API key
-
-# Initialize model
-model = genai.GenerativeModel("gemini-pro")
-
 # User input for chatbot
 user_prompt = st.text_input("Ask AI something:")
 
 if st.button("Get AI Response"):
     if user_prompt:
-        response = model.generate_content(user_prompt)
-        st.write("### AI Response:")
-        st.write(response.text)
+        try:
+            model = genai.GenerativeModel("gemini-pro")
+            response = model.generate_content(user_prompt)
+            st.write("### AI Response:")
+            st.write(response.text)
+        except Exception as e:
+            st.error(f"❌ Error: {e}")
     else:
         st.warning("⚠️ Please enter a question.")

backup.py

@@ -2,6 +2,7 @@ import streamlit as st
 import pandas as pd
 from pymongo import MongoClient
 from db import insert_data_if_empty, get_mongo_client  # Import functions from db.py
+import google.generativeai as genai  # Import Generative AI library
 
 #### **1. Ensure Data is Inserted Before Display**
 insert_data_if_empty()
@@ -10,7 +11,7 @@ insert_data_if_empty()
 collection = get_mongo_client()
 
 #### **3. Streamlit App to Display Data**
-st.title("📊 MongoDB Data Viewer")
+st.title("📊 MongoDB Data Viewer with AI Chatbot")
 
 # Show first 5 rows from MongoDB
 st.subheader("First 5 Rows from Database")
@@ -25,3 +26,23 @@ else:
 if st.button("Show Complete Data"):
     all_data = list(collection.find({}, {"_id": 0}))
     st.write(pd.DataFrame(all_data))
+
+#### **4. GenAI Chatbot Interface**
+st.subheader("🤖 AI Chatbot")
+
+# Configure Generative AI model (replace `your-api-key` with actual key)
+genai.configure(api_key="secrets")  # Set up your API key
+
+# Initialize model
+model = genai.GenerativeModel("gemini-pro")
+
+# User input for chatbot
+user_prompt = st.text_input("Ask AI something:")
+
+if st.button("Get AI Response"):
+    if user_prompt:
+        response = model.generate_content(user_prompt)
+        st.write("### AI Response:")
+        st.write(response.text)
+    else:
+        st.warning("⚠️ Please enter a question.")

requirements.txt

@@ -11,3 +11,11 @@ pydantic==1.10.13
 typing-extensions==4.5.0
 requests==2.31.0
 protobuf==4.25.1
+generativeai==0.0.1
+google-ai-generativelanguage==0.6.15
+google-api-core==2.24.1
+google-api-python-client==2.163.0
+google-auth==2.38.0
+google-auth-httplib2==0.2.0
+google-generativeai==0.8.4
+googleapis-common-protos==1.69.1