Update app.py

app.py

@@ -18,14 +18,16 @@ st.set_page_config(page_title="Enhanced Gemini Q&A App", layout="wide")
 if "chat_history" not in st.session_state:
     st.session_state.chat_history = []  # Empty list for chat history
 
-# Function to get response from Gemini API
 # Function to get response from Gemini API
 def get_gemini_response(question):
     # Reformat the session chat history to match the API requirements
-    formatted_history = [
-        {"parts": [{"text": chat["user"]}], "role": "user"} if "user" in chat else {"parts": [{"text": chat["bot"]}], "role": "bot"}
-        for chat in st.session_state.chat_history
-    ]
+    formatted_history = []
+    for chat in st.session_state.chat_history:
+        if "user" in chat:
+            formatted_history.append({"parts": [{"text": chat["user"]}], "role": "user"})
+        else:
+            formatted_history.append({"parts": [{"text": chat["bot"]}], "role": "bot"})
+
     chat = model.start_chat(history=formatted_history)  # Use the correctly formatted history
     try:
         response = chat.send_message(question, stream=True)
@@ -37,7 +39,6 @@ def get_gemini_response(question):
     except Exception as e:
         return f"Error: {str(e)}"
 
-
 # UI Layout
 st.title("🤖 Gemini AI - Interactive Q&A")
 st.write("Ask me anything and I'll try to answer!")