feat: switch to Groq LLaMA 3 free API

app.py

@@ -4,31 +4,24 @@ import tempfile
 import gc
 import base64
 import time
-import json
 from datetime import datetime
 
 from crewai import Agent, Crew, Process, Task
 from crewai_tools import SerperDevTool
+from langchain_groq import ChatGroq
 from src.agentic_rag.tools.custom_tool import DocumentSearchTool
 
-# ===========================
-#   Page Config — must be first
-# ===========================
 st.set_page_config(
     page_title="NeuralDocs — Agentic RAG",
-    page_icon="⬡",
+    page_icon="🧠",
     layout="wide",
     initial_sidebar_state="expanded"
 )
 
-# ===========================
-#   Global CSS — Unified Dark Theme
-# ===========================
 st.markdown("""
 <style>
 @import url('https://fonts.googleapis.com/css2?family=Syne:wght@400;600;700;800&family=JetBrains+Mono:wght@400;500&family=Figtree:wght@300;400;500;600&display=swap');
 
-/* ── ROOT TOKENS ── */
 :root {
     --bg-base:       #080c12;
     --bg-surface:    #0d1520;
@@ -53,29 +46,23 @@ st.markdown("""
     --radius-sm:     6px;
     --radius-md:     12px;
     --radius-lg:     18px;
-    --shadow-card:   0 4px 24px rgba(0,0,0,0.4);
-    --shadow-glow:   0 0 32px rgba(0,229,255,0.08);
 }
 
-/* ── GLOBAL RESET ── */
 html, body, [class*="css"] {
     font-family: var(--font-body);
     background-color: var(--bg-base) !important;
     color: var(--text-primary) !important;
 }
 
-/* ── HIDE STREAMLIT CHROME ── */
 #MainMenu, footer, header { visibility: hidden; }
 .stDeployButton { display: none; }
 
-/* ── MAIN CONTAINER ── */
 .main .block-container {
     background: var(--bg-base);
     padding: 1.5rem 2rem 3rem;
     max-width: 1100px;
 }
 
-/* ── APP HEADER ── */
 .app-header {
     display: flex;
     align-items: center;
@@ -89,43 +76,27 @@ html, body, [class*="css"] {
     background: linear-gradient(135deg, var(--accent), var(--accent2));
     border-radius: 10px;
     display: flex; align-items: center; justify-content: center;
-    font-size: 18px; font-weight: 800;
-    color: #000;
-    flex-shrink: 0;
-    box-shadow: 0 0 20px var(--accent-glow);
+    font-size: 18px; font-weight: 800; color: #000; flex-shrink: 0;
 }
 .app-title {
     font-family: var(--font-display);
-    font-size: 1.5rem;
-    font-weight: 800;
-    letter-spacing: -0.5px;
+    font-size: 1.5rem; font-weight: 800; letter-spacing: -0.5px;
     background: linear-gradient(135deg, var(--accent) 0%, var(--accent2) 100%);
-    -webkit-background-clip: text;
-    -webkit-text-fill-color: transparent;
-    background-clip: text;
+    -webkit-background-clip: text; -webkit-text-fill-color: transparent; background-clip: text;
     line-height: 1;
 }
 .app-subtitle {
-    font-size: 0.7rem;
-    color: var(--text-muted);
-    letter-spacing: 2px;
-    text-transform: uppercase;
-    font-family: var(--font-mono);
-    margin-top: 3px;
+    font-size: 0.7rem; color: var(--text-muted);
+    letter-spacing: 2px; text-transform: uppercase;
+    font-family: var(--font-mono); margin-top: 3px;
 }
 .header-badge {
     margin-left: auto;
-    background: var(--accent-glow);
-    border: 1px solid var(--accent-dim);
-    color: var(--accent);
-    font-family: var(--font-mono);
-    font-size: 0.65rem;
-    padding: 4px 10px;
-    border-radius: 20px;
-    letter-spacing: 1px;
+    background: var(--accent-glow); border: 1px solid var(--accent-dim);
+    color: var(--accent); font-family: var(--font-mono);
+    font-size: 0.65rem; padding: 4px 10px; border-radius: 20px; letter-spacing: 1px;
 }
 
-/* ── SIDEBAR ── */
 [data-testid="stSidebar"] {
     background: var(--bg-surface) !important;
     border-right: 1px solid var(--border) !important;
@@ -135,377 +106,98 @@ html, body, [class*="css"] {
     padding: 1.5rem 1.2rem;
 }
 .sidebar-section-label {
-    font-family: var(--font-mono);
-    font-size: 0.6rem;
-    letter-spacing: 3px;
-    text-transform: uppercase;
-    color: var(--text-muted);
-    margin: 1.2rem 0 0.6rem;
-    padding-bottom: 0.4rem;
-    border-bottom: 1px solid var(--border);
-}
-.sidebar-title {
-    font-family: var(--font-display);
-    font-size: 1.1rem;
-    font-weight: 700;
-    color: var(--text-primary);
-    margin-bottom: 0.2rem;
+    font-family: var(--font-mono); font-size: 0.6rem;
+    letter-spacing: 3px; text-transform: uppercase;
+    color: var(--text-muted); margin: 1.2rem 0 0.6rem;
+    padding-bottom: 0.4rem; border-bottom: 1px solid var(--border);
 }
 
-/* ── STATUS PILLS ── */
 .status-pill {
-    display: inline-flex;
-    align-items: center;
-    gap: 6px;
-    padding: 5px 12px;
-    border-radius: 20px;
-    font-family: var(--font-mono);
-    font-size: 0.7rem;
-    letter-spacing: 0.5px;
-    margin: 6px 0;
+    display: inline-flex; align-items: center; gap: 6px;
+    padding: 5px 12px; border-radius: 20px;
+    font-family: var(--font-mono); font-size: 0.7rem; letter-spacing: 0.5px; margin: 6px 0;
 }
-.status-ready   { background: rgba(0,200,150,0.1);  border: 1px solid var(--success); color: var(--success); }
-.status-waiting { background: rgba(255,183,0,0.1);  border: 1px solid var(--warning); color: var(--warning); }
-.status-error   { background: rgba(255,77,109,0.1); border: 1px solid var(--danger);  color: var(--danger); }
+.status-ready   { background: rgba(0,200,150,0.1); border: 1px solid var(--success); color: var(--success); }
+.status-waiting { background: rgba(255,183,0,0.1); border: 1px solid var(--warning); color: var(--warning); }
 .status-dot { width: 6px; height: 6px; border-radius: 50%; background: currentColor; }
 .status-dot.pulse { animation: pulse 1.5s infinite; }
 @keyframes pulse { 0%,100%{opacity:1;} 50%{opacity:0.3;} }
 
-/* ── METRIC CARDS ── */
-.metrics-row {
-    display: grid;
-    grid-template-columns: 1fr 1fr;
-    gap: 8px;
-    margin: 10px 0;
-}
-.metric-card {
-    background: var(--bg-elevated);
-    border: 1px solid var(--border);
-    border-radius: var(--radius-md);
-    padding: 10px 14px;
-}
-.metric-value {
-    font-family: var(--font-display);
-    font-size: 1.3rem;
-    font-weight: 700;
-    color: var(--accent);
-    line-height: 1;
-}
-.metric-label {
-    font-family: var(--font-mono);
-    font-size: 0.6rem;
-    color: var(--text-muted);
-    letter-spacing: 1px;
-    text-transform: uppercase;
-    margin-top: 3px;
-}
+.metrics-row { display: grid; grid-template-columns: 1fr 1fr; gap: 8px; margin: 10px 0; }
+.metric-card { background: var(--bg-elevated); border: 1px solid var(--border); border-radius: var(--radius-md); padding: 10px 14px; }
+.metric-value { font-family: var(--font-display); font-size: 1.3rem; font-weight: 700; color: var(--accent); line-height: 1; }
+.metric-label { font-family: var(--font-mono); font-size: 0.6rem; color: var(--text-muted); letter-spacing: 1px; text-transform: uppercase; margin-top: 3px; }
 
-/* ── SOURCE BADGE ── */
 .source-badge {
-    display: inline-flex;
-    align-items: center;
-    gap: 5px;
-    background: var(--bg-elevated);
-    border: 1px solid var(--border);
-    border-radius: 4px;
-    padding: 2px 8px;
-    font-family: var(--font-mono);
-    font-size: 0.65rem;
-    color: var(--text-secondary);
-    margin: 2px 3px;
+    display: inline-flex; align-items: center; gap: 5px;
+    background: var(--bg-elevated); border: 1px solid var(--border);
+    border-radius: 4px; padding: 2px 8px;
+    font-family: var(--font-mono); font-size: 0.65rem; color: var(--text-secondary); margin: 2px 3px;
 }
-.source-badge.pdf  { border-color: var(--accent2);  color: #b39ddb; background: rgba(124,92,252,0.08); }
-.source-badge.web  { border-color: var(--accent);   color: var(--accent); background: var(--accent-glow); }
+.source-badge.pdf  { border-color: var(--accent2); color: #b39ddb; background: rgba(124,92,252,0.08); }
+.source-badge.web  { border-color: var(--accent); color: var(--accent); background: var(--accent-glow); }
 
-/* ── CHAT MESSAGES ── */
 .stChatMessage {
-    background: var(--bg-surface) !important;
-    border: 1px solid var(--border) !important;
-    border-radius: var(--radius-lg) !important;
-    padding: 1rem 1.2rem !important;
-    margin-bottom: 10px !important;
-}
-[data-testid="stChatMessageContent"] p {
-    font-family: var(--font-body);
-    font-size: 0.9rem;
-    line-height: 1.7;
-    color: var(--text-primary);
-}
-[data-testid="stChatMessageContent"] code {
-    font-family: var(--font-mono);
-    background: var(--bg-elevated);
-    border: 1px solid var(--border);
-    border-radius: 4px;
-    padding: 1px 6px;
-    font-size: 0.82rem;
-    color: var(--accent);
+    background: var(--bg-surface) !important; border: 1px solid var(--border) !important;
+    border-radius: var(--radius-lg) !important; padding: 1rem 1.2rem !important; margin-bottom: 10px !important;
 }
+[data-testid="stChatMessageContent"] p { font-family: var(--font-body); font-size: 0.9rem; line-height: 1.7; color: var(--text-primary); }
 
-/* ── FOLLOW-UP QUESTION CHIPS ── */
-.followup-container {
-    margin: 12px 0 6px;
-}
-.followup-label {
-    font-family: var(--font-mono);
-    font-size: 0.6rem;
-    letter-spacing: 2px;
-    text-transform: uppercase;
-    color: var(--text-muted);
-    margin-bottom: 6px;
-}
-.followup-chips {
-    display: flex;
-    flex-wrap: wrap;
-    gap: 6px;
-}
-.followup-chip {
-    background: var(--bg-elevated);
-    border: 1px solid var(--border-bright);
-    border-radius: 20px;
-    padding: 5px 12px;
-    font-size: 0.78rem;
-    color: var(--text-secondary);
-    cursor: pointer;
-    transition: all 0.2s;
-    font-family: var(--font-body);
-}
-.followup-chip:hover {
-    border-color: var(--accent);
-    color: var(--accent);
-    background: var(--accent-glow);
-}
-
-/* ── ANSWER METADATA BAR ── */
 .answer-meta {
-    display: flex;
-    align-items: center;
-    gap: 12px;
-    padding: 8px 0 4px;
-    border-top: 1px solid var(--border);
-    margin-top: 10px;
-    flex-wrap: wrap;
-}
-.answer-meta-item {
-    display: flex;
-    align-items: center;
-    gap: 4px;
-    font-family: var(--font-mono);
-    font-size: 0.65rem;
-    color: var(--text-muted);
-}
-.confidence-bar-wrap {
-    display: flex;
-    align-items: center;
-    gap: 6px;
-}
-.confidence-bar {
-    width: 60px;
-    height: 4px;
-    background: var(--border);
-    border-radius: 2px;
-    overflow: hidden;
-}
-.confidence-fill {
-    height: 100%;
-    border-radius: 2px;
-    background: linear-gradient(90deg, var(--accent2), var(--accent));
+    display: flex; align-items: center; gap: 12px;
+    padding: 8px 0 4px; border-top: 1px solid var(--border); margin-top: 10px; flex-wrap: wrap;
 }
+.answer-meta-item { display: flex; align-items: center; gap: 4px; font-family: var(--font-mono); font-size: 0.65rem; color: var(--text-muted); }
+.confidence-bar-wrap { display: flex; align-items: center; gap: 6px; }
+.confidence-bar { width: 60px; height: 4px; background: var(--border); border-radius: 2px; overflow: hidden; }
+.confidence-fill { height: 100%; border-radius: 2px; background: linear-gradient(90deg, var(--accent2), var(--accent)); }
 
-/* ── CHAT INPUT ── */
 [data-testid="stChatInput"] {
-    background: var(--bg-elevated) !important;
-    border: 1px solid var(--border-bright) !important;
-    border-radius: var(--radius-lg) !important;
-}
-[data-testid="stChatInput"]:focus-within {
-    border-color: var(--accent) !important;
-    box-shadow: 0 0 0 3px var(--accent-glow) !important;
-}
-[data-testid="stChatInput"] textarea {
-    color: var(--text-primary) !important;
-    font-family: var(--font-body) !important;
-    font-size: 0.9rem !important;
-    background: transparent !important;
+    background: var(--bg-elevated) !important; border: 1px solid var(--border-bright) !important; border-radius: var(--radius-lg) !important;
 }
+[data-testid="stChatInput"]:focus-within { border-color: var(--accent) !important; }
+[data-testid="stChatInput"] textarea { color: var(--text-primary) !important; font-family: var(--font-body) !important; font-size: 0.9rem !important; background: transparent !important; }
 
-/* ── BUTTONS ── */
 .stButton > button {
-    background: var(--bg-elevated) !important;
-    border: 1px solid var(--border-bright) !important;
-    color: var(--text-secondary) !important;
-    border-radius: var(--radius-sm) !important;
-    font-family: var(--font-mono) !important;
-    font-size: 0.72rem !important;
-    letter-spacing: 0.5px !important;
-    transition: all 0.2s !important;
-    padding: 0.35rem 0.8rem !important;
-}
-.stButton > button:hover {
-    border-color: var(--accent) !important;
-    color: var(--accent) !important;
-    background: var(--accent-glow) !important;
+    background: var(--bg-elevated) !important; border: 1px solid var(--border-bright) !important;
+    color: var(--text-secondary) !important; border-radius: var(--radius-sm) !important;
+    font-family: var(--font-mono) !important; font-size: 0.72rem !important; transition: all 0.2s !important;
 }
+.stButton > button:hover { border-color: var(--accent) !important; color: var(--accent) !important; background: var(--accent-glow) !important; }
 
-/* ── FILE UPLOADER ── */
 [data-testid="stFileUploader"] {
-    background: var(--bg-elevated) !important;
-    border: 1px dashed var(--border-bright) !important;
-    border-radius: var(--radius-md) !important;
-    padding: 8px !important;
-}
-[data-testid="stFileUploader"]:hover {
-    border-color: var(--accent) !important;
-}
-[data-testid="stFileUploader"] label {
-    color: var(--text-secondary) !important;
-    font-size: 0.8rem !important;
+    background: var(--bg-elevated) !important; border: 1px dashed var(--border-bright) !important; border-radius: var(--radius-md) !important;
 }
 
-/* ── SPINNER ── */
-.stSpinner > div {
-    border-top-color: var(--accent) !important;
+.history-item {
+    background: var(--bg-elevated); border: 1px solid var(--border);
+    border-left: 3px solid var(--accent-dim); border-radius: var(--radius-sm);
+    padding: 8px 10px; margin: 4px 0;
 }
+.history-q { font-size: 0.78rem; color: var(--text-secondary); white-space: nowrap; overflow: hidden; text-overflow: ellipsis; }
+.history-time { font-family: var(--font-mono); font-size: 0.6rem; color: var(--text-muted); margin-top: 2px; }
 
-/* ── DIVIDER ── */
-hr { border-color: var(--border) !important; }
+.empty-state { display: flex; flex-direction: column; align-items: center; justify-content: center; padding: 3rem 1rem; text-align: center; gap: 12px; }
+.empty-icon { font-size: 2.5rem; opacity: 0.3; }
+.empty-text { font-family: var(--font-body); color: var(--text-muted); font-size: 0.9rem; max-width: 320px; line-height: 1.6; }
 
-/* ── SCROLLBAR ── */
+hr { border-color: var(--border) !important; }
 ::-webkit-scrollbar { width: 4px; }
 ::-webkit-scrollbar-track { background: var(--bg-base); }
 ::-webkit-scrollbar-thumb { background: var(--border-bright); border-radius: 2px; }
-::-webkit-scrollbar-thumb:hover { background: var(--accent-dim); }
-
-/* ── EMPTY STATE ── */
-.empty-state {
-    display: flex;
-    flex-direction: column;
-    align-items: center;
-    justify-content: center;
-    padding: 3rem 1rem;
-    text-align: center;
-    gap: 12px;
-}
-.empty-icon {
-    font-size: 2.5rem;
-    opacity: 0.3;
-}
-.empty-text {
-    font-family: var(--font-body);
-    color: var(--text-muted);
-    font-size: 0.9rem;
-    max-width: 320px;
-    line-height: 1.6;
-}
-.empty-hint {
-    font-family: var(--font-mono);
-    font-size: 0.68rem;
-    color: var(--text-muted);
-    letter-spacing: 1px;
-}
-
-/* ── SUGGESTION PROMPTS ── */
-.suggestion-grid {
-    display: grid;
-    grid-template-columns: 1fr 1fr;
-    gap: 8px;
-    margin-top: 1rem;
-    max-width: 560px;
-}
-.suggestion-card {
-    background: var(--bg-surface);
-    border: 1px solid var(--border);
-    border-radius: var(--radius-md);
-    padding: 10px 14px;
-    cursor: pointer;
-    transition: all 0.2s;
-}
-.suggestion-card:hover {
-    border-color: var(--accent);
-    background: var(--accent-glow);
-}
-.suggestion-icon { font-size: 1rem; margin-bottom: 4px; }
-.suggestion-text {
-    font-size: 0.78rem;
-    color: var(--text-secondary);
-    font-family: var(--font-body);
-    line-height: 1.4;
-}
-
-/* ── HISTORY ITEM ── */
-.history-item {
-    background: var(--bg-elevated);
-    border: 1px solid var(--border);
-    border-left: 3px solid var(--accent-dim);
-    border-radius: var(--radius-sm);
-    padding: 8px 10px;
-    margin: 4px 0;
-    cursor: pointer;
-    transition: all 0.15s;
-}
-.history-item:hover {
-    border-left-color: var(--accent);
-    background: var(--bg-hover);
-}
-.history-q {
-    font-size: 0.78rem;
-    color: var(--text-secondary);
-    white-space: nowrap;
-    overflow: hidden;
-    text-overflow: ellipsis;
-    font-family: var(--font-body);
-}
-.history-time {
-    font-family: var(--font-mono);
-    font-size: 0.6rem;
-    color: var(--text-muted);
-    margin-top: 2px;
-}
-
-/* ── SUCCESS / INFO ALERTS ── */
-.stSuccess, .stInfo, .stWarning {
-    background: var(--bg-elevated) !important;
-    border-radius: var(--radius-md) !important;
-}
-
-/* ── SELECTBOX ── */
-[data-testid="stSelectbox"] > div {
-    background: var(--bg-elevated) !important;
-    border: 1px solid var(--border-bright) !important;
-    border-radius: var(--radius-sm) !important;
-    color: var(--text-primary) !important;
-}
-
-/* ── PROGRESS ── */
-.stProgress > div > div {
-    background: linear-gradient(90deg, var(--accent2), var(--accent)) !important;
-}
 </style>
 """, unsafe_allow_html=True)
 
-
-# ===========================
-#   Session State Init
-# ===========================
+# Session State
 defaults = {
-    "messages": [],
-    "pdf_tool": None,
-    "crew": None,
-    "pdf_name": None,
-    "total_queries": 0,
-    "pdf_hits": 0,
-    "web_hits": 0,
-    "query_log": [],          # [{q, ts, source}]
-    "pending_followups": [],  # follow-up suggestions for latest answer
-    "pending_prompt": None,   # queued prompt from chip click
+    "messages": [], "pdf_tool": None, "crew": None, "pdf_name": None,
+    "total_queries": 0, "pdf_hits": 0, "web_hits": 0,
+    "query_log": [], "pending_followups": [], "pending_prompt": None,
 }
 for k, v in defaults.items():
     if k not in st.session_state:
         st.session_state[k] = v
 
-
-# ===========================
-#   Helpers
-# ===========================
 def reset_chat():
     st.session_state.messages = []
     st.session_state.pending_followups = []
@@ -516,128 +208,85 @@ def full_reset():
         st.session_state[k] = defaults[k]
     gc.collect()
 
-def display_pdf(file_bytes: bytes, file_name: str):
+def display_pdf(file_bytes, file_name):
     b64 = base64.b64encode(file_bytes).decode("utf-8")
-    st.markdown(f"""
-    <iframe src="data:application/pdf;base64,{b64}"
-        width="100%" height="480px" type="application/pdf"
-        style="border:1px solid var(--border); border-radius:var(--radius-md); background:var(--bg-elevated);">
-    </iframe>
-    """, unsafe_allow_html=True)
+    st.markdown(f'<iframe src="data:application/pdf;base64,{b64}" width="100%" height="480px" style="border:1px solid var(--border);border-radius:var(--radius-md);"></iframe>', unsafe_allow_html=True)
 
-def detect_source(result_text: str) -> str:
-    """Heuristic: guess whether answer came primarily from PDF or web."""
+def detect_source(result_text):
     web_signals = ["according to", "website", "http", "search result", "online", "found on"]
-    if any(s in result_text.lower() for s in web_signals):
-        return "web"
-    return "pdf"
-
-def estimate_confidence(result_text: str) -> int:
-    """Simple heuristic confidence 0–100."""
-    if "i'm sorry" in result_text.lower() or "couldn't find" in result_text.lower():
-        return 20
-    if len(result_text) > 400:
-        return 88
-    if len(result_text) > 150:
-        return 72
+    return "web" if any(s in result_text.lower() for s in web_signals) else "pdf"
+
+def estimate_confidence(result_text):
+    if "i'm sorry" in result_text.lower() or "couldn't find" in result_text.lower(): return 20
+    if len(result_text) > 400: return 88
+    if len(result_text) > 150: return 72
     return 55
 
-def generate_followups(query: str, answer: str) -> list[str]:
-    """Generate 3 contextual follow-up questions from the answer."""
-    lines = answer.strip().split(".")
-    followups = []
-    # Simple keyword-driven suggestions
+def generate_followups(query, answer):
     keywords = [w for w in query.lower().split() if len(w) > 4]
-    starters = [
+    return [
         f"Can you elaborate on {keywords[0] if keywords else 'this'} in more detail?",
-        f"What are the practical applications of this?",
-        f"How does this compare to similar approaches?",
-        f"What are the limitations or challenges here?",
-        f"Can you give a concrete example?",
+        "What are the practical applications of this?",
+        "What are the limitations or challenges here?",
     ]
-    return starters[:3]
 
+def get_llm():
+    return ChatGroq(
+        api_key=os.environ.get("GROQ_API_KEY"),
+        model="llama3-70b-8192",
+        temperature=0.1,
+    )
 
-# ===========================
-#   Crew Factory
-# ===========================
 def create_crew(pdf_tool):
+    llm = get_llm()
     web_search_tool = SerperDevTool()
     tools = [t for t in [pdf_tool, web_search_tool] if t]
 
     retriever = Agent(
         role="Retrieve relevant information to answer the user query: {query}",
-        goal=(
-            "Retrieve the most relevant information from the available sources "
-            "for the user query: {query}. Always try the PDF search tool first. "
-            "Fall back to web search only if the PDF search yields nothing useful."
-        ),
-        backstory=(
-            "You're a meticulous analyst with a keen eye for detail. "
-            "You understand user queries: {query} and retrieve knowledge "
-            "from the most suitable knowledge base."
-        ),
-        verbose=True,
-        tools=tools,
+        goal="Retrieve the most relevant information from available sources for: {query}. Try PDF first, fall back to web search.",
+        backstory="You are a meticulous analyst with a keen eye for detail.",
+        verbose=True, tools=tools, llm=llm,
     )
 
     synthesizer = Agent(
-        role="Response synthesizer for the user query: {query}",
-        goal=(
-            "Synthesize retrieved information into a concise, well-structured response "
-            "for the query: {query}. If nothing was found, say so clearly."
-        ),
-        backstory="You're a skilled communicator who turns complex information into clear answers.",
-        verbose=True,
+        role="Response synthesizer for: {query}",
+        goal="Synthesize retrieved information into a concise, well-structured response for: {query}.",
+        backstory="You are a skilled communicator who turns complex information into clear answers.",
+        verbose=True, llm=llm,
     )
 
     retrieval_task = Task(
-        description="Retrieve the most relevant information for the user query: {query}",
-        expected_output="The most relevant information as retrieved from the sources.",
+        description="Retrieve the most relevant information for: {query}",
+        expected_output="The most relevant information retrieved from available sources.",
         agent=retriever,
     )
 
     response_task = Task(
-        description="Synthesize the final response for the user query: {query}",
-        expected_output=(
-            "A concise, coherent response based on retrieved information for: {query}. "
-            "If nothing was found, say: \"I'm sorry, I couldn't find the information you're looking for.\""
-        ),
+        description="Synthesize the final response for: {query}",
+        expected_output="A concise, coherent response based on retrieved information for: {query}.",
         agent=synthesizer,
     )
 
     return Crew(
         agents=[retriever, synthesizer],
         tasks=[retrieval_task, response_task],
-        process=Process.sequential,
-        verbose=True,
+        process=Process.sequential, verbose=True,
     )
 
-
-# ===========================
-#   SIDEBAR
-# ===========================
+# Sidebar
 with st.sidebar:
-    # Brand
     st.markdown("""
-    <div style="padding: 0.2rem 0 1rem;">
-        <div style="display:flex; align-items:center; gap:10px; margin-bottom:4px;">
-            <div style="width:30px;height:30px;background:linear-gradient(135deg,#00e5ff,#7c5cfc);
-                        border-radius:8px;display:flex;align-items:center;justify-content:center;
-                        font-weight:800;color:#000;font-size:14px;">⬡</div>
-            <span style="font-family:'Syne',sans-serif;font-size:1rem;font-weight:700;
-                         background:linear-gradient(135deg,#00e5ff,#7c5cfc);
-                         -webkit-background-clip:text;-webkit-text-fill-color:transparent;">NeuralDocs</span>
-        </div>
-        <div style="font-family:'JetBrains Mono',monospace;font-size:0.6rem;
-                    letter-spacing:2px;color:#3d5a75;text-transform:uppercase;">
-            Agentic RAG · v2.0
+    <div style="padding:0.2rem 0 1rem;">
+        <div style="display:flex;align-items:center;gap:10px;margin-bottom:4px;">
+            <div style="width:30px;height:30px;background:linear-gradient(135deg,#00e5ff,#7c5cfc);border-radius:8px;display:flex;align-items:center;justify-content:center;font-weight:800;color:#000;font-size:14px;">🧠</div>
+            <span style="font-family:'Syne',sans-serif;font-size:1rem;font-weight:700;background:linear-gradient(135deg,#00e5ff,#7c5cfc);-webkit-background-clip:text;-webkit-text-fill-color:transparent;">NeuralDocs</span>
         </div>
+        <div style="font-family:'JetBrains Mono',monospace;font-size:0.6rem;letter-spacing:2px;color:#3d5a75;text-transform:uppercase;">Agentic RAG · Groq + LLaMA 3</div>
     </div>
     """, unsafe_allow_html=True)
 
     st.markdown('<div class="sidebar-section-label">📄 Document</div>', unsafe_allow_html=True)
-
     uploaded_file = st.file_uploader("Upload PDF", type=["pdf"], label_visibility="collapsed")
 
     if uploaded_file is not None:
@@ -649,28 +298,16 @@ with st.sidebar:
                 with st.spinner("Indexing document…"):
                     st.session_state.pdf_tool = DocumentSearchTool(file_path=path)
                     st.session_state.pdf_name = uploaded_file.name
-                    st.session_state.crew = None  # force rebuild
+                    st.session_state.crew = None
             st.session_state.messages = []
             st.session_state.pending_followups = []
 
-        st.markdown(f"""
-        <div class="status-pill status-ready">
-            <span class="status-dot"></span>
-            {uploaded_file.name[:28]}{'…' if len(uploaded_file.name)>28 else ''}
-        </div>
-        """, unsafe_allow_html=True)
-
+        st.markdown(f'<div class="status-pill status-ready"><span class="status-dot"></span>{uploaded_file.name[:28]}{"…" if len(uploaded_file.name)>28 else ""}</div>', unsafe_allow_html=True)
         with st.expander("Preview PDF", expanded=False):
             display_pdf(uploaded_file.getvalue(), uploaded_file.name)
     else:
-        st.markdown("""
-        <div class="status-pill status-waiting">
-            <span class="status-dot pulse"></span>
-            No document loaded
-        </div>
-        """, unsafe_allow_html=True)
+        st.markdown('<div class="status-pill status-waiting"><span class="status-dot pulse"></span>No document loaded</div>', unsafe_allow_html=True)
 
-    # ── METRICS (Extended Feature #1) ──
     st.markdown('<div class="sidebar-section-label">📊 Session Stats</div>', unsafe_allow_html=True)
     total = st.session_state.total_queries or 0
     pdf_h = st.session_state.pdf_hits or 0
@@ -680,209 +317,117 @@ with st.sidebar:
 
     st.markdown(f"""
     <div class="metrics-row">
-        <div class="metric-card">
-            <div class="metric-value">{total}</div>
-            <div class="metric-label">Queries</div>
-        </div>
-        <div class="metric-card">
-            <div class="metric-value" style="color:var(--success)">{pdf_pct}%</div>
-            <div class="metric-label">PDF hits</div>
-        </div>
+        <div class="metric-card"><div class="metric-value">{total}</div><div class="metric-label">Queries</div></div>
+        <div class="metric-card"><div class="metric-value" style="color:var(--success)">{pdf_pct}%</div><div class="metric-label">PDF hits</div></div>
     </div>
     <div class="metrics-row">
-        <div class="metric-card">
-            <div class="metric-value" style="color:var(--accent2)">{web_pct}%</div>
-            <div class="metric-label">Web hits</div>
-        </div>
-        <div class="metric-card">
-            <div class="metric-value" style="color:var(--warning)">{len(st.session_state.messages)//2}</div>
-            <div class="metric-label">Exchanges</div>
-        </div>
+        <div class="metric-card"><div class="metric-value" style="color:var(--accent2)">{web_pct}%</div><div class="metric-label">Web hits</div></div>
+        <div class="metric-card"><div class="metric-value" style="color:var(--warning)">{len(st.session_state.messages)//2}</div><div class="metric-label">Exchanges</div></div>
     </div>
     """, unsafe_allow_html=True)
 
-    # ── QUERY HISTORY (Extended Feature #2) ──
     if st.session_state.query_log:
         st.markdown('<div class="sidebar-section-label">🕓 Query History</div>', unsafe_allow_html=True)
         for item in reversed(st.session_state.query_log[-6:]):
             src_icon = "🌐" if item["source"] == "web" else "📄"
-            st.markdown(f"""
-            <div class="history-item">
-                <div class="history-q">{src_icon} {item['q'][:48]}{'…' if len(item['q'])>48 else ''}</div>
-                <div class="history-time">{item['ts']}</div>
-            </div>
-            """, unsafe_allow_html=True)
+            st.markdown(f'<div class="history-item"><div class="history-q">{src_icon} {item["q"][:48]}{"…" if len(item["q"])>48 else ""}</div><div class="history-time">{item["ts"]}</div></div>', unsafe_allow_html=True)
 
     st.markdown('<div class="sidebar-section-label">⚙ Controls</div>', unsafe_allow_html=True)
     col1, col2 = st.columns(2)
-    with col1:
-        st.button("Clear Chat", on_click=reset_chat, use_container_width=True)
-    with col2:
-        st.button("Full Reset", on_click=full_reset, use_container_width=True)
-
+    with col1: st.button("Clear Chat", on_click=reset_chat, use_container_width=True)
+    with col2: st.button("Full Reset", on_click=full_reset, use_container_width=True)
 
-# ===========================
-#   MAIN AREA
-# ===========================
-
-# Header
+# Main
 st.markdown("""
 <div class="app-header">
-    <div class="app-logo">⬡</div>
+    <div class="app-logo">🧠</div>
     <div>
         <div class="app-title">NeuralDocs</div>
         <div class="app-subtitle">Agentic RAG · PDF + Web Intelligence</div>
     </div>
-    <div class="header-badge">CREWAI POWERED</div>
+    <div class="header-badge">GROQ · LLAMA 3</div>
 </div>
 """, unsafe_allow_html=True)
 
-# ── Empty state ──
 if not st.session_state.messages:
     st.markdown("""
     <div class="empty-state">
-        <div class="empty-icon">⬡</div>
-        <div class="empty-text">
-            Upload a PDF in the sidebar, then ask anything.<br>
-            The agent searches your document first — then the web.
-        </div>
-        <div class="empty-hint">PDF · WEB SEARCH · SYNTHESIS</div>
-    </div>
-    <div style="display:flex;justify-content:center;">
-    <div class="suggestion-grid">
-        <div class="suggestion-card">
-            <div class="suggestion-icon">📖</div>
-            <div class="suggestion-text">Summarize the key points of this document</div>
-        </div>
-        <div class="suggestion-card">
-            <div class="suggestion-icon">🔍</div>
-            <div class="suggestion-text">What are the main findings or conclusions?</div>
-        </div>
-        <div class="suggestion-card">
-            <div class="suggestion-icon">🌐</div>
-            <div class="suggestion-text">Search the web for the latest news on this topic</div>
-        </div>
-        <div class="suggestion-card">
-            <div class="suggestion-icon">⚖️</div>
-            <div class="suggestion-text">Compare this with current industry standards</div>
-        </div>
-    </div>
+        <div class="empty-icon">🧠</div>
+        <div class="empty-text">Upload a PDF in the sidebar, then ask anything.<br>The agent searches your document first — then the web.</div>
     </div>
     """, unsafe_allow_html=True)
 
-# ── Chat history ──
 for i, message in enumerate(st.session_state.messages):
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
-
-        # Show metadata + follow-ups only on last assistant message
-        if (message["role"] == "assistant"
-                and i == len(st.session_state.messages) - 1
-                and st.session_state.pending_followups):
-
+        if message["role"] == "assistant" and i == len(st.session_state.messages) - 1 and st.session_state.pending_followups:
             conf = estimate_confidence(message["content"])
-            src  = detect_source(message["content"])
-            src_label = "Web Search" if src == "web" else "PDF Document"
-            src_class = src
-
+            src = detect_source(message["content"])
             st.markdown(f"""
             <div class="answer-meta">
-                <span class="source-badge {src_class}">
-                    {'🌐' if src == 'web' else '📄'} {src_label}
-                </span>
+                <span class="source-badge {src}">{'🌐 Web' if src == 'web' else '📄 PDF'}</span>
                 <div class="answer-meta-item confidence-bar-wrap">
                     <span>Confidence</span>
-                    <div class="confidence-bar">
-                        <div class="confidence-fill" style="width:{conf}%"></div>
-                    </div>
+                    <div class="confidence-bar"><div class="confidence-fill" style="width:{conf}%"></div></div>
                     <span>{conf}%</span>
                 </div>
-                <div class="answer-meta-item">
-                    🕐 {st.session_state.query_log[-1]['ts'] if st.session_state.query_log else '—'}
-                </div>
             </div>
             """, unsafe_allow_html=True)
-
-            # Follow-up chips
-            st.markdown("""
-            <div class="followup-label">💡 FOLLOW-UP SUGGESTIONS</div>
-            """, unsafe_allow_html=True)
-
+            st.markdown('<div style="font-family:var(--font-mono);font-size:0.6rem;color:var(--text-muted);margin:8px 0 4px;">💡 FOLLOW-UP SUGGESTIONS</div>', unsafe_allow_html=True)
             cols = st.columns(len(st.session_state.pending_followups))
             for j, (col, q) in enumerate(zip(cols, st.session_state.pending_followups)):
                 with col:
                     if st.button(q, key=f"followup_{i}_{j}", use_container_width=True):
                         st.session_state.pending_prompt = q
 
-# ───────────────────────────────────────────
-#   Handle queued follow-up prompt
-# ───────────────────────────────────────────
 if st.session_state.pending_prompt:
     prompt = st.session_state.pending_prompt
     st.session_state.pending_prompt = None
     st.session_state.pending_followups = []
-
     st.session_state.messages.append({"role": "user", "content": prompt})
-    with st.chat_message("user"):
-        st.markdown(prompt)
-
-    if st.session_state.crew is None:
-        st.session_state.crew = create_crew(st.session_state.pdf_tool)
-
+    with st.chat_message("user"): st.markdown(prompt)
+    if st.session_state.crew is None: st.session_state.crew = create_crew(st.session_state.pdf_tool)
     with st.chat_message("assistant"):
         placeholder = st.empty()
-        full_response = ""
         with st.spinner("Researching…"):
             result = st.session_state.crew.kickoff(inputs={"query": prompt}).raw
-
+        full_response = ""
         for i, line in enumerate(result.split("\n")):
             full_response += line + ("\n" if i < len(result.split("\n")) - 1 else "")
             placeholder.markdown(full_response + "▌")
-            time.sleep(0.12)
+            time.sleep(0.08)
         placeholder.markdown(full_response)
-
     src = detect_source(result)
-    ts  = datetime.now().strftime("%H:%M")
+    ts = datetime.now().strftime("%H:%M")
     st.session_state.total_queries += 1
     if src == "pdf": st.session_state.pdf_hits += 1
-    else:            st.session_state.web_hits += 1
+    else: st.session_state.web_hits += 1
     st.session_state.query_log.append({"q": prompt, "ts": ts, "source": src})
     st.session_state.messages.append({"role": "assistant", "content": result})
     st.session_state.pending_followups = generate_followups(prompt, result)
     st.rerun()
 
-# ───────────────────────────────────────────
-#   Primary chat input
-# ───────────────────────────────────────────
 prompt = st.chat_input("Ask anything about your document or search the web…")
-
 if prompt:
     st.session_state.pending_followups = []
     st.session_state.messages.append({"role": "user", "content": prompt})
-
-    with st.chat_message("user"):
-        st.markdown(prompt)
-
-    if st.session_state.crew is None:
-        st.session_state.crew = create_crew(st.session_state.pdf_tool)
-
+    with st.chat_message("user"): st.markdown(prompt)
+    if st.session_state.crew is None: st.session_state.crew = create_crew(st.session_state.pdf_tool)
     with st.chat_message("assistant"):
         placeholder = st.empty()
-        full_response = ""
         with st.spinner("Researching…"):
             result = st.session_state.crew.kickoff(inputs={"query": prompt}).raw
-
+        full_response = ""
         for i, line in enumerate(result.split("\n")):
             full_response += line + ("\n" if i < len(result.split("\n")) - 1 else "")
             placeholder.markdown(full_response + "▌")
-            time.sleep(0.12)
+            time.sleep(0.08)
         placeholder.markdown(full_response)
-
     src = detect_source(result)
-    ts  = datetime.now().strftime("%H:%M")
+    ts = datetime.now().strftime("%H:%M")
     st.session_state.total_queries += 1
     if src == "pdf": st.session_state.pdf_hits += 1
-    else:            st.session_state.web_hits += 1
+    else: st.session_state.web_hits += 1
     st.session_state.query_log.append({"q": prompt, "ts": ts, "source": src})
     st.session_state.messages.append({"role": "assistant", "content": result})
     st.session_state.pending_followups = generate_followups(prompt, result)

requirements.txt

@@ -1,8 +1,9 @@
 streamlit==1.32.0
-crewai==0.80.0
-crewai-tools==0.14.0
+crewai==0.86.0
+crewai-tools>=0.17.0
 markitdown>=0.0.1
 chonkie>=0.1.0
 qdrant-client>=1.7.3
 openai>=1.30.0
-pydantic>=2.0.0
+langchain-groq>=0.1.0
+groq>=0.4.0