Update app.py

app.py

@@ -1,14 +1,14 @@
-import streamlit as st
+import streamlit as st 
 import os
 import pathlib
-import textwrap
+import textwrap 
 from PIL import Image
-import json
 
 import google.generativeai as genai
 
 genai.configure(api_key='AIzaSyCeNgXfZx0kJ736XFVtxXxev_RdscB0i5s')
 
+## Function to load OpenAI model and get respones
 def get_gemini_response(input,image,prompt):
     model = genai.GenerativeModel('gemini-pro-vision')
     response = model.generate_content([input,image[0],prompt])
@@ -48,14 +48,7 @@ input_prompt ="""
                """
 
 if submit:
-    try:
-        image_data = input_image_setup(uploaded_file)
-        response = get_gemini_response(input_prompt,image_data,input)
-        try:
-            json_response = json.loads(response)
-            st.subheader("Output :")
-            st.write(json.dumps(json_response, indent=4))
-        except json.JSONDecodeError as e:
-            st.error(f"Invalid JSON response: {e}")
-    except FileNotFoundError as e:
-        st.error(str(e))
+    image_data = input_image_setup(uploaded_file)
+    response = get_gemini_response(input_prompt,image_data,input)
+    st.subheader("Output :")
+    st.write(response)