color change red h

app.py

@@ -23,11 +23,12 @@ if photo:
         genus, species = txt.split(' ')
         venomousity = openai.Completion.create(engine="text-davinci-003", prompt=f"Is {genus} {species} is venomous? if venomous just say venomous otherwise say nonvenomous", max_tokens=50)
         desc = openai.Completion.create(engine="text-davinci-003", prompt=f"Give a short description of {genus} {species}", max_tokens=500)
+        snake_type = venomousity.choices[0].text
         with col1:
-            if venomousity.choices[0].text == 'Venomous':
-                st.markdown(f'<span style="color: red;">{venomousity.choices[0].text}</span>', unsafe_allow_html=True)
-            else:
-                st.markdown(f'<span style="color: green;">{venomousity.choices[0].text}</span>', unsafe_allow_html=True)
+            if snake_type == 'Venomous':
+                st.markdown(f'<span style="color: red;">{snake_type}</span>', unsafe_allow_html=True)
+            elif snake_type == 'Nonvenomous':
+                st.markdown(f'<span style="color: green;">{snake_type}</span>', unsafe_allow_html=True)
             st.markdown(desc.choices[0].text)
     except:
         col1.write(f"Error: There might be some network issue")