Update app.py

app.py

@@ -1,33 +1,51 @@
+# app.py
 import gradio as gr
 from search import search_google
 from scraper import scrape_url
+from rag import VectorStore
 from llm import generate_answer
+from summarizer import summarize_text
+
+# initialize vector store once
+vs = VectorStore()
 
 def ask_agent(question):
+    # Step 1: Search Google for top 5 URLs
     urls = search_google(question, num_results=5)
-    
     if not urls:
         return "❗ No search results found."
 
-    first_url = urls[0]
-    context = scrape_url(first_url)
-    
-    answer = generate_answer(context, question)
-    
-    # Show other high quality links as sources (skip first scraped one)
-    better_links = "\n".join([f"- [{u}]({u})" for u in urls[1:]])
-    
-    return f"""### 🧠 **Answer**  
+    # Step 2: Scrape text from each URL
+    texts = [scrape_url(url) for url in urls]
+
+    # Step 3: Add scraped texts to vector store
+    vs.add_texts(texts)
+
+    # Step 4: Retrieve top 3 most relevant texts to the question
+    relevant_texts = vs.retrieve(question, top_k=3)
+    context = "\n\n".join(relevant_texts)
+
+    # Step 5: Summarize the context to keep it concise
+    summary = summarize_text(context, max_length=100)
+
+    # Step 6: Generate final answer using LLM
+    answer = generate_answer(summary, question)
+
+    # Step 7: Format other URLs as markdown links
+    better_links = "\n".join([f"- [{u}]({u})" for u in urls])
+
+    return f"""### 🧠 **Answer**
 {answer}
 
-✅ **Context scraped from:** [{first_url}]({first_url})
+✅ **Context summarized from top sites**
 
 📚 **Other useful links:**  
-{better_links if better_links else 'No other links'}
+{better_links}
 """
 
+# Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("# 🔍 Dynamic RAG agent\nScrapes *one* top site, answers, and shows more trusted links.")
+    gr.Markdown("# 🔍 **Dynamic RAG Agent + Summarizer**\nGoogle search → scrape → embed → retrieve → summarize → answer!")
     inp = gr.Textbox(label="Ask your question")
     out = gr.Markdown()
     btn = gr.Button("Ask")