Update app.py

app.py

@@ -166,7 +166,7 @@ def run_inference():
 
         if st.session_state['kbvqa']:
             display_model_settings()
-            st.table(pd.DataFrame(st.session_state, index=[0]))
+            display_session_state()
             image_qa_app(st.session_state['kbvqa'])
 
     else:
@@ -177,6 +177,22 @@ def display_model_settings():
     st.write("### Current Model Settings:")
     st.table(pd.DataFrame(st.session_state['model_settings']))
 
+def display_session_state():
+    def flatten_dict(d, parent_key='', sep='_'):
+        items = []
+        for k, v in d.items():
+            new_key = f"{parent_key}{sep}{k}" if parent_key else k
+            if isinstance(v, dict):
+                items.extend(flatten_dict(v, new_key, sep=sep).items())
+            else:
+                items.append((new_key, v))
+        return dict(items)
+
+    st.write("### Current Session State:")
+    flattened_state = flatten_dict(st.session_state)
+    st.table(pd.DataFrame([flattened_state]))
+
+