As others had pointed out, you can add prerequisite libraries in the requirements.txt file and it this case, in addition to your other dependencies, you can add google-generativeai to the file.
For further information, we actually have an FAQ about this topic – please check it out: FAQ: ModuleNotFoundError.
Happy Streamlit-ing! 
These cookies are necessary for the website to function and cannot be switched off. They are usually only set in response to actions made by you which amount to a request for services, such as setting your privacy preferences, logging in or filling in forms.
These cookies allow us to count visits and traffic sources so we can measure and improve the performance of our site. They help us understand how visitors move around the site and which pages are most frequently visited.
These cookies are used to record your choices and settings, maintain your preferences over time and recognize you when you return to our website. These cookies help us to personalize our content for you and remember your preferences.
These cookies may be deployed to our site by our advertising partners to build a profile of your interest and provide you with content that is relevant to you, including showing you relevant ads on other websites.