Update app.py

app.py

@@ -34,7 +34,11 @@ def main():
             if st.button("Submit"):
                 if question:
                     response = query_engine.query(question)
-                    #st.markdown(f"**Response:** {response}")
+                    # Displaying the response
+                    if response.startswith('http'):
+                        st.image(response, use_column_width=True)
+                    else:
+                        st.markdown(f"**Response:** {response}")
 
 if __name__ == "__main__":
     main()