Update app.py

app.py

@@ -51,8 +51,11 @@ if submit:
     try:
         image_data = input_image_setup(uploaded_file)
         response = get_gemini_response(input_prompt,image_data,input)
-        json_response = json.loads(response)
-        st.subheader("Output :")
-        st.write(json.dumps(json_response, indent=4))
+        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))