testing a more compact demo

app.py

@@ -125,47 +125,48 @@ def render_experiment(query):
         f"{CLASSIFIER_TAG}-Output"
     ]
 
-    ################################### SHOW DESCRIPTION OF CLASS
-    with st.expander("Show Class Description"):
-        st.write(f"**Name**: {prediction_label}")
-        st.write("**Class Definition**:")
-        st.markdown("`" + class_def + "`")
-        st.image(
-            Image.open(f"demonstrations/{predicted_wnid}.jpeg"),
-            caption=f"Class Explanation",
-            use_column_width=True,
-        )
-
     ################################### SHOW QUERY and PREDICTION
-    with st.expander("Show Query"):
-        col1, col2 = st.columns(2)
-        with col1:
-            st.image(load_query(current_query), caption=f"Query ID: {query_id}")
-        with col2:
-            default_value = 0
-            if query_id in session_state.user_feedback.keys():
-                if session_state.user_feedback[query_id] == "Correct":
-                    default_value = 1
-                elif session_state.user_feedback[query_id] == "Wrong":
-                    default_value = 2
-
-            session_state.user_feedback[query_id] = st.radio(
-                "What do you think about model's prediction?",
-                ("-", "Correct", "Wrong"),
-                key=query_id,
-                index=default_value,
-            )
-            st.write(f"**Model Prediction**: {prediction_label}")
-            st.write(f"**Model Confidence**: {prediction_confidence}")
 
-        ################################### SHOW Model Explanation
-        if selected_xai_tool is not None:
+    col1, col2 = st.columns(2)
+    with col1:
+        st.image(load_query(current_query), caption=f"Query ID: {query_id}")
+    with col2:
+        ################################### SHOW DESCRIPTION OF CLASS
+        with st.expander("Show Class Description"):
+            st.write(f"**Name**: {prediction_label}")
+            st.write("**Class Definition**:")
+            st.markdown("`" + class_def + "`")
             st.image(
-                selected_xai_tool(current_query),
-                caption=f"Explaination",
+                Image.open(f"demonstrations/{predicted_wnid}.jpeg"),
+                caption=f"Class Explanation",
                 use_column_width=True,
             )
 
+        
+        default_value = 0
+        if query_id in session_state.user_feedback.keys():
+            if session_state.user_feedback[query_id] == "Correct":
+                default_value = 1
+            elif session_state.user_feedback[query_id] == "Wrong":
+                default_value = 2
+
+        session_state.user_feedback[query_id] = st.radio(
+            "What do you think about model's prediction?",
+            ("-", "Correct", "Wrong"),
+            key=query_id,
+            index=default_value,
+        )
+        st.write(f"**Model Prediction**: {prediction_label}")
+        st.write(f"**Model Confidence**: {prediction_confidence}")
+
+    ################################### SHOW Model Explanation
+    if selected_xai_tool is not None:
+        st.image(
+            selected_xai_tool(current_query),
+            caption=f"Explaination",
+            use_column_width=True,
+        )
+
     ################################### SHOW DEBUG INFO
 
     if st.button("Debug: Show Everything"):