Update app.py

app.py

@@ -37,7 +37,6 @@ class ChatBot:
             json=data,
         )
         response_json = response.json()
-        response_text = response_json.get("contents", [{}])[0].get("parts", [{}])[0].get("text", "")
         return response_text
 
     def get_history(self) -> list:
@@ -69,7 +68,8 @@ if uploaded_file is not None:
         user_input = st.text_input("Ask a question about the image:")
         
         if user_input:
-            response = chatbot.generate_response(user_input)
+            json_response = chatbot.generate_response(user_input)
+            response = json_response["candidates"][0]["content"]["parts"][0]["text"]
             st.session_state.conversation.append({"user": user_input, "bot": response})
         
         if st.session_state.conversation: