Update app.py

app.py

@@ -3,7 +3,7 @@ import google.generativeai as genai
 import streamlit as st
 
 # Configure the google-generativeai library by providing your API key
-genai.configure(api_key="AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM")
+genai.configure(api_key="YOUR_API_KEY")
 
 # Set up the model
 generation_config = {
@@ -32,20 +32,58 @@ safety_settings = [
   }
 ]
 
-model = genai.GenerativeModel(model_name="gemini-pro",
+# Initialize the chat history and the model instance using session state
+if "chat_history" not in st.session_state:
+  st.session_state.chat_history = []
+if "model" not in st.session_state:
+  st.session_state.model = genai.GenerativeModel(model_name="gemini-pro",
                               generation_config=generation_config,
                               safety_settings=safety_settings)
 
-# Create a streamlit interface with a title, a description, an input box, and an output box
+# Create a streamlit interface with a title, a description, and a chat container
 st.title("Gemini API") # the title of the interface
 st.markdown("A simple interface for Gemini API that explains code snippets in natural language.") # the description of the interface
-user_input = st.text_area("User Input", height=200) # the input box for the user input
-if st.button("Generate"): # the button to trigger the generation
-  # Start a chat session with the model
-  convo = model.start_chat(history=[])
-  # Send the user input to the model
-  convo.send_message(user_input)
-  # Get the response from the model
-  response = convo.last.text
-  # Display the response in the output box
-  st.text_area("Response", response, height=200) # the output box for the response
+chat_container = st.empty() # the chat container to display chat messages
+
+# Display chat messages from history on app rerun
+for message in st.session_state.chat_history:
+  role = message["role"]
+  content = message["content"]
+  with chat_container.chat_message(role):
+    st.markdown(content)
+
+# Accept user input and attachment using chat input widget
+user_input, attachment = st.chat_input("Enter a message or a code snippet")
+
+if user_input or attachment:
+  # Add user message to chat history
+  if user_input:
+    st.session_state.chat_history.append({"role": "user", "content": user_input})
+  # Display user message in chat container
+  with chat_container.chat_message("user"):
+    st.markdown(user_input)
+  if attachment:
+    # You can process the attachment here or save it as needed
+    # For example, you can save it to a temporary folder and store the file path in the chat history
+    attachment_path = f"attachments/{attachment.name}"
+    with open(attachment_path, "wb") as f:
+      f.write(attachment.read())
+    # Add attachment message to chat history
+    st.session_state.chat_history.append({"role": "user", "content": f"Attachment: {attachment.name}"})
+    # Display attachment message in chat container
+    with chat_container.chat_message("user"):
+      st.markdown(f"Attachment: {attachment.name}")
+
+  # Display assistant response in chat container with a spinner
+  with chat_container.chat_message("assistant"):
+    with st.spinner("Generating a response..."):
+      # Start a chat session with the model
+      convo = st.session_state.model.start_chat(history=[])
+      # Send the user input to the model
+      convo.send_message(user_input)
+      # Get the response from the model
+      response = convo.last.text
+      # Display the response in the chat container
+      st.markdown(response)
+      # Add assistant message to chat history
+      st.session_state.chat_history.append({"role": "assistant", "content": response})