Update app.py

app.py

@@ -1,57 +1,45 @@
-import os
 import streamlit as st
-import google.generativeai as genai
+from google.cloud.generativeai import GenerativeModel, Part, Content
 
+# Configure API key
+api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"
+model_name = "gemini-pro"
 
-# Title for the application
-st.title("Chat with Gemini Bot")
+# Create model object
+model = GenerativeModel(api_key=api_key, model_name=model_name)
 
-# Set your Google API Key as an environment variable
-os.environ['GOOGLE_API_KEY'] = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"
-genai.configure(api_key=os.environ['GOOGLE_API_KEY'])
+# Create chatbot interface
+st.title("Gemini API Chatbot")
 
-# Create the model
-model = genai.GenerativeModel('gemini-pro')
+# Get chat history from session state
+chat_history = st.session_state.get("chat_history", [])
 
-# Initialize chat history
-if 'chat_history' not in st.session_state:
-    st.session_state.chat_history = []
-
-# Display chat history
-for message in st.session_state.chat_history:
-    if message['role'] == 'user':
-        st.markdown(f"**You:** {message['content']}")
-    else:
-        st.markdown(f"**Gemini Bot:** {message['content']}")
-
-# User input
+# Get user input from text box
 user_input = st.text_input("You")
 
-# If user input is not empty
+# Check if user input is not empty
 if user_input:
+    # Create user message object
+    user_message = Content(parts=[Part(text=user_input)], role="user")
+
     # Add user message to chat history
-    st.session_state.chat_history.append({
-        'role': 'user',
-        'content': user_input
-    })
+    chat_history.append(user_message)
 
-    # Display user message
+    # Display user message with markdown
     st.markdown(f"**You:** {user_input}")
 
-    # Create a message for the Gemini model
-    message = [{'role': 'user', 'content': msg['content']} for msg in st.session_state.chat_history]
+    # Get model response with start_chat method
+    with st.spinner("Thinking..."):
+        response = model.generate_content(chat_history)
 
-    # Generate response from the model
-    response = model.generate_content(message)
+    # Get response text from response object
+    response_text = response.contents[-1].parts[0].text
 
-    # Extract bot's message from the response
-    bot_message = response['choices'][0]['message']['content']
+    # Add response message to chat history
+    chat_history.append(Content(parts=[Part(text=response_text)], role="assistant"))
 
-    # Add bot message to chat history
-    st.session_state.chat_history.append({
-        'role': 'assistant',
-        'content': bot_message
-    })
+    # Display response message with markdown
+    st.markdown(f"**Gemini Bot:** {response_text}")
 
-    # Display bot message
-    st.markdown(f"**Gemini Bot:** {bot_message}")
+    # Update session state with chat history
+    st.session_state["chat_history"] = chat_history