Update Home.py

Home.py

@@ -79,11 +79,14 @@ def load_image():
 def visual_qna():
     st.title("Visual Q&A")
     img = load_image()
-    if query := st.chat_input("Enter your message"):
-        response = visualqna(question=query, image=img)
-    with st.chat_message("assistant"):
-        st.write(response)
-        
+    if img:
+        if query := st.chat_input("Enter your message"):
+            response = visualqna(question=query, image=img)
+            with st.chat_message("assistant"):
+                st.write(response)
+    else:
+        st.warning("Please enter an image URL and click 'Load Image' before asking a question.")
+