Update app.py

app.py

@@ -122,18 +122,21 @@ if prompt := st.chat_input("Hey?"):
                     processed_file = upload_and_process_file(temp_file.name)
                     content.append(processed_file)
 
+        
         # Send user entry to Gemini and read the response
-        response = model.generate_content(content)
+        response = model.generate_content(content, stream=True)
+        response.resolve()
 
-        # Add the response to the chat history
-        st.session_state.chat.history.extend([
-            genai.types.Part(text=prompt, role="user"),
-            genai.types.Part(text=response.text, role="model"),
-        ])
-        # Display last
+        # Display the response
         with st.chat_message("assistant"):
             st.markdown(response.text)
 
+        # Update the chat history
+        st.session_state.chat.history.extend([
+            genai.types.Content(parts=[genai.types.Part(text=prompt)], role="user"),
+            genai.types.Content(parts=[genai.types.Part(text=response.text)], role="model")
+        ])
+
     except Exception as e:
         st.error(f"An error occurred: {e}")
 
@@ -143,5 +146,4 @@ if prompt := st.chat_input("Hey?"):
             try:
                 os.unlink(temp_file)
             except Exception as e:
-                print(
-                    f"Error deleting temporary file {temp_file}: {e}")
+                print(f"Error deleting temporary file {temp_file}: {e}")