modification

app.py

@@ -30,7 +30,7 @@ chat_history = st.session_state.get("chat_history", [])
 # Display previous messages
 for message in chat_history:
     role, text = message
-    st.markdown(f"**{role}:** {text}")
+    st.markdown(f"**{role.title()}:** {text}")
 
 # Get user input from text box
 user_input = st.text_input("You")
@@ -38,23 +38,26 @@ user_input = st.text_input("You")
 # Check if user input is not empty
 if user_input:
   # Add user message to chat history
-  chat_history.append(("user", user_input))
+  chat_history.append({"role": "user", "content": user_input})
+
+  # Display user message with markdown
+  st.markdown(f"**You:** {user_input}")
 
   # Create model object
   model = genai.GenerativeModel(model_name="gemini-pro")
 
-  # Prepare chat history for generate_content method
-  genai_chat_history = [{"role": role, "content": text} for role, text in chat_history]
-
   # Get model response with generate_content method
   with st.spinner("Thinking..."):
-      response = model.generate_content(genai_chat_history)
+      response = model.generate_content(chat_history)
 
   # Get response text from response object
   response_text = response.text
 
   # Add response message to chat history
-  chat_history.append(("assistant", response_text))
+  chat_history.append({"role": "assistant", "content": response_text})
+
+  # Display response message with markdown
+  st.markdown(f"**Gemini Bot:** {response_text}")
 
   # Update session state with chat history
   st.session_state["chat_history"] = chat_history
@@ -68,7 +71,7 @@ if st.button("Display History"):
     c.execute("SELECT * FROM history")
     rows = c.fetchall()
     for row in rows:
-        st.markdown(f"**{row[0]}:** {row[1]}")
+        st.markdown(f"**{row[0].title()}:** {row[1]}")
 
 # Add a button to clear chat history from database
 if st.button("Clear History"):
@@ -77,7 +80,7 @@ if st.button("Clear History"):
 
 # Save chat history to database
 for message in chat_history:
-    c.execute("INSERT INTO history VALUES (?, ?)", message)
+    c.execute("INSERT INTO history VALUES (?, ?)", (message["role"], message["content"]))
 conn.commit()
 
 # Close the connection