Update app.py

app.py

@@ -326,7 +326,25 @@ query = st.text_input("What are you looking for?")
 
 if query:
     results = st.session_state.engine.search_documents(query, top_k=5)
+
+    # --- LLM INTEGRATION START ---
+    if results:
+        # We grab the text from the #1 result to feed the AI
+        top_match = results[0]
+        top_context = f"Source: {top_match['source']}\nContent: {top_match['snippet']}"
+        
+        # Create a container for the AI Answer
+        with st.container():
+            st.markdown("### 🤖 AI Summary")
+            if st.button("✨ Summarize Top Result"):
+                with st.spinner("Reading document..."):
+                    ai_response = ask_llm(query, top_context)
+                    st.success(ai_response)
+            st.divider()
+    # --- LLM INTEGRATION END ---
+    
     st.subheader("Top Relevant Documents")
+
     if not results: st.info("No documents found.")
     for res in results:
         score = res['score']