Spaces:
Running
Running
File size: 686 Bytes
f0a1f4f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
import streamlit as st
import google.generativeai as genai
def CREATEAD():
st.title("Create a script/idea")
prompt = st.text_input("Write about product/Advertisement Idea you already have.")
if prompt:
# Or use `os.getenv('GOOGLE_API_KEY')` to fetch an environment variable.
GOOGLE_API_KEY = st.secrets["GOOGLE_API_KEY"]
genai.configure(api_key=GOOGLE_API_KEY)
model = genai.GenerativeModel('gemini-1.5-flash')
response = model.generate_content(f"Write a funny short advertisement script for this {prompt}", stream=True)
response.resolve()
st.write(response.text)
if __name__ == "__main__":
CREATEAD()
|