Update app.py

app.py

@@ -190,13 +190,33 @@ if query:
         
         # --- AI SUMMARY SECTION ---
         with st.container():
-            st.markdown("### 🤖 Executive Summary")
+            st.markdown("### 🤖 Intelligence Hub")
             st.caption(f"Analyzing primary source: {top_match['source']}")
             
+            # LAYOUT: Two columns for controls
+            col1, col2 = st.columns(2)
+            
+            with col1:
+                # The "Deep Dive" Selector
+                analysis_mode = st.selectbox(
+                    "Select Analysis Type:", 
+                    ["Executive Summary", "Action Plan", "Risk Assessment", "Socratic Review"]
+                )
+            
+            with col2:
+                # The "Brain" Selector
+                model_choice = st.selectbox(
+                    "Select Model:",
+                    ["Gemini (Cloud - Smartest)", "Granite (Private Space)"]
+                )
+                # Map the UI text to the backend key
+                provider = "Gemini" if "Gemini" in model_choice else "Granite"
+
             if st.button("✨ Generate Assessment"):
-                with st.spinner("Sending Data to Selected Model..."):
-                    # Call our separated LLM client
-                    response = ask_granite(query, full_doc_text)
+                with st.spinner(f"Consulting {provider} via {analysis_mode}..."):
+                    
+                    # Call the client with the new parameters
+                    response = ask_llm(query, full_doc_text, mode=analysis_mode, model_provider=provider)
                     
                     st.markdown("---")
                     st.markdown(response)