| import streamlit as st | |
| import os | |
| from PIL import Image | |
| import google.generativeai as genai | |
| secret_key = os.getenv("SECRET_KEY") | |
| genai.configure(api_key=secret_key) | |
| def get_gemini_response(image): | |
| model = genai.GenerativeModel('gemini-pro-vision') | |
| input='''You are a prompt generator.You will get an image. | |
| Your work is to write a prompt such that an image generator model would create most identical picture | |
| as the image given to you''' | |
| response = model.generate_content([input,image]) | |
| return response.text | |
| st.set_page_config(page_title="Prompt generation from image") | |
| st.header("Application") | |
| uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"]) | |
| image="" | |
| if uploaded_file is not None: | |
| image = Image.open(uploaded_file) | |
| st.image(image, caption="Uploaded Image.", use_column_width=True) | |
| submit=st.button("Generate Prompt") | |
| if submit: | |
| response=get_gemini_response(image) | |
| st.subheader("The Generated Prompt is") | |
| st.write(response) |