From b5edaecefc914740a225a54b6090d53db2f5e3bd Mon Sep 17 00:00:00 2001 From: sajjadrahman56 Date: Tue, 26 Dec 2023 03:31:53 +0600 Subject: [PATCH] images and text updated --- app.py | 5 +++-- text_image.py | 51 +++++++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 54 insertions(+), 2 deletions(-) create mode 100644 text_image.py diff --git a/app.py b/app.py index 3671151..08ac07a 100644 --- a/app.py +++ b/app.py @@ -13,9 +13,10 @@ def get_response(input_text): return response.text st.set_page_config(page_title="Q & A", page_icon=":gem:") -st.header("Gemini LLM application") -input = st.text_input("Enter your question here", key="input") +st.header("I am your FRIEND Gemini") +st.subheader("Ask me anything and I will try to answer it for you as best as I can" ) +input = st.text_input("Enter your question here", key="input") submit = st.button("Submit") diff --git a/text_image.py b/text_image.py new file mode 100644 index 0000000..4f43566 --- /dev/null +++ b/text_image.py @@ -0,0 +1,51 @@ +from dotenv import load_dotenv +load_dotenv() + +import streamlit as st +import os +import pathlib +import textwrap +from PIL import Image + + +import google.generativeai as genai + + +os.getenv("GOOGLE_API_KEY") +genai.configure(api_key=os.getenv("GOOGLE_API_KEY")) + +## Function to load OpenAI model and get respones + +def get_gemini_response(input,image): + model = genai.GenerativeModel('gemini-pro-vision') + if input!="": + response = model.generate_content([input,image]) + else: + response = model.generate_content(image) + return response.text + +##initialize our streamlit app + +st.set_page_config(page_title="Gemini Image Demo") + +st.header("You and Me ! Here we go !! ") + +input=st.text_input("Input Prompt: ",key="input") +uploaded_file = st.file_uploader("select 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("Tell me about the image") + +## If ask button is clicked + +if submit: + + response=get_gemini_response(input,image) + st.subheader("The Response is") + st.write(response) \ No newline at end of file