update app.py

Fixed chatbox issue and added module for user unput

app.py

@@ -1,12 +1,11 @@
 import streamlit as st
 import textwrap
 import google.generativeai as genai
-from IPython.display import display
 from IPython.display import Markdown
 import PIL.Image
+#from PIL import Image
+
 
-def load_img(file_pth):
-    return PIL.Image.open(file_pth)
 
 def to_markdown(text):
     text = text.replace('•', '  *')
@@ -31,31 +30,36 @@ def chat(inpt_frm_vsn):
 
     while True:
         response = chat.send_message(res, generation_config=genai.types.GenerationConfig(
-            max_output_tokens=200
+            max_output_tokens=100
         ))
         to_markdown(response.text)
 
         st.text('Gemini: ' + response.text)
-        res = st.text_input('User: ')
+        res = st.chat_input('Please write your message here...')
+        if res:
+            st.write(f"User: {res}")
 
 def main():
     st.title("Gemini")
 
-    img_path = 'assets/apple.jpg'
-    img = load_img(img_path)
-    st.image(img, caption='Uploaded Image.', use_column_width=True)
 
     GOOGLE_API_KEY = 'AIzaSyDMDS5TghIYsEV4nD6FsC8PDCtqGvkuUls'
 
     genai.configure(api_key=GOOGLE_API_KEY)
+    uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
+
+    if uploaded_file is not None:
+        img = PIL.Image.open(uploaded_file)
+        st.image(img, caption='Uploaded Image', use_column_width=True)
+
+        vision_result = vision(img)
+        st.text(vision_result)
 
-    vision_result = vision(img)
-    st.text(vision_result)
+        check_point = st.text_input("If you want to continue chat write 'Nutrition': ")
 
-    check_point = st.text_input("If you want to continue chat write 'nutrition': ")
+        if check_point.lower() == 'nutrition':
+            chat(vision_result)
 
-    if check_point.lower() == 'nutrition':
-        chat(vision_result)
 
 if __name__ == "__main__":
     main()