Update src/app.py

src/app.py

@@ -329,8 +329,11 @@ with tab2:
     # --- CONTROLS AND METRICS ---
     c1, c2, c3 = st.columns([2, 1, 1])
     with c1:
-        # Use the global model_choice
-        selected_model_name = st.session_state.get('model_choice', 'Granite 4 (IBM)')
+        # FIX: Access the correct key from the sidebar widget
+        # We default to the global variable 'model_choice' if state is missing
+        selected_model_name = st.session_state.get('model_selector_radio', model_choice)
+        st.caption(f"Active Model: **{selected_model_name}**")
+        
     with c2:
         use_rag = st.toggle("🔌 Enable Knowledge Base", value=False)
     
@@ -426,9 +429,12 @@ with tab2:
                         model_id = val
                         break
                 
+                # ROUTING CHECK
                 if not model_id and "gpt" in selected_model_name.lower():
+                    # If it's the GPT model choice
                     response, usage = query_openai_model(messages_payload, max_len)
                 elif model_id:
+                    # If it's the local Ollama model
                     response, usage = query_local_model(messages_payload, max_len, model_id)
                 else:
                     response, usage = "Error: Could not determine model to use.", None
@@ -455,7 +461,7 @@ with tab2:
                     st.caption(f"Rank {i+1} (Source: {src}, Rel: {score})")
                     st.text(doc.page_content)
                     st.divider()
-
+                    
 # --- TAB 3: PROMPT ARCHITECT ---
 with tab3:
     st.header("🛠️ Mega-Prompt Factory")