LOVEBHATTI commited on
Commit
64aab44
1 Parent(s): 514745b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -16
app.py CHANGED
@@ -1,27 +1,31 @@
1
  import streamlit as st
2
  import google.generativeai as genAi
3
  from PIL import Image
4
-
5
- genAi.configure(api_key="AIzaSyBZQmot8_8bDxuGiffJ06woJCzH140Erc4str")
6
 
7
- def get_response(inputt,image):
8
- model=genAi.GenerativeModel('gemini-pro-vision')
9
- if inputt!="":
10
- response=model.generate_content([inputt,image])
 
 
 
11
  else:
12
- # response=model.generate_content([image])
13
- response=model.generate_content(image)
14
  return response.text
15
 
 
16
  st.header("PROJECT")
17
- input=st.text_input("Input Prompt",key='input')
18
- upload_file=st.file_uploader("Choose yout file",type=['jpg','jpeg','png'])
19
- image=''
 
20
  if upload_file is not None:
21
- image=Image.open(upload_file)
22
- st.image(image,caption="uploaded_image",use_column_width=True)
 
 
 
23
 
24
- submit=st.button("ASK")
25
  if submit:
26
- response=get_response(input,image)
27
- st.write(response)
 
1
  import streamlit as st
2
  import google.generativeai as genAi
3
  from PIL import Image
 
 
4
 
5
+ # Configure API key
6
+ genAi.configure(api_key="AIzaSyBZQmot8_8bDxuGiffJ06woJCzH140Erc4")
7
+
8
+ def get_response(inputt, image):
9
+ model = genAi.GenerativeModel('gemini-pro-vision')
10
+ if inputt != "":
11
+ response = model.generate_content([inputt, image])
12
  else:
13
+ response = model.generate_content(image)
 
14
  return response.text
15
 
16
+ # Streamlit app header and user inputs
17
  st.header("PROJECT")
18
+ inputt = st.text_input("Input Prompt", key='inputt')
19
+ upload_file = st.file_uploader("Choose your file", type=['jpg', 'jpeg', 'png'])
20
+ image = ''
21
+
22
  if upload_file is not None:
23
+ image = Image.open(upload_file)
24
+ st.image(image, caption="Uploaded Image", use_column_width=True)
25
+
26
+ # Button to trigger content generation
27
+ submit = st.button("ASK")
28
 
 
29
  if submit:
30
+ response = get_response(inputt, image)
31
+ st.write(response)