Update app.py

app.py

@@ -13,7 +13,7 @@ from docx.shared import Pt
 APP_NAME = "ScholarLens"
 TAGLINE = "Query your literature, get page-level proof"
 
-# ---------- Config (same engine, nicer UI) ----------
+# ---------- Config ----------
 EMBED_MODEL_NAME = "intfloat/multilingual-e5-small"
 CHUNK_SIZE = 1200
 CHUNK_OVERLAP = 200
@@ -23,9 +23,8 @@ MAX_CONTEXT_CHARS = 16000
 INDEX_PATH = "rag_index.faiss"
 STORE_PATH = "rag_store.pkl"
 
-# You can edit the default model here. All are selectable in the UI.
 MODEL_CHOICES = [
-    "llama-3.3-70b-versatile",
+    "llama-3.1-70b-versatile",
     "llama-3.1-8b-instant",
     "mixtral-8x7b-32768",
 ]
@@ -44,7 +43,10 @@ def extract_text_from_pdf(pdf_path: str) -> List[Tuple[int, str]]:
             if not txt.strip():
                 blocks = page.get_text("blocks")
                 if isinstance(blocks, list):
-                    txt = "\n".join(b[4] for b in blocks if isinstance(b, (list, tuple)) and len(b) > 4)
+                    txt = "\n".join(
+                        b[4] for b in blocks
+                        if isinstance(b, (list, tuple)) and len(b) > 4
+                    )
             pages.append((i, txt or ""))
     return pages
 
@@ -182,7 +184,7 @@ def retrieve(query: str, top_k=5, must_contain: str = ""):
     return hits
 
 # ---------- Groq LLM ----------
-def groq_answer(query: str, contexts, model_name="llama-3.3-70b-versatile", temperature=0.2, max_tokens=1000):
+def groq_answer(query: str, contexts, model_name="llama-3.1-70b-versatile", temperature=0.2, max_tokens=1000):
     try:
         if not os.environ.get("GROQ_API_KEY"):
             return "GROQ_API_KEY is not set. Add it in your Space secrets or the key box."
@@ -236,7 +238,6 @@ def export_answer_to_docx(question: str, answer_md: str, rows: List[List[str]])
     doc.add_paragraph(f"Question: {question}")
 
     doc.add_heading("Answer", level=2)
-    # Write as plain text to keep it simple in Word
     for line in answer_md.splitlines():
         doc.add_paragraph(line)
 
@@ -278,13 +279,11 @@ def ask_rag(question: str, top_k, model_name: str, temperature: float, must_cont
         ctx = retrieve(question, top_k=int(top_k) if top_k else TOP_K_DEFAULT, must_contain=must_contain)
         ans = groq_answer(question, ctx, model_name=model_name, temperature=temperature)
 
-        # sources table
         rows = []
         for c in ctx:
             preview = c["text"][:200].replace("\n"," ") + ("..." if len(c["text"])>200 else "")
             rows.append([c["source"], str(c["page_start"]), f"{c['score']:.3f}", preview])
 
-        # snippets pretty print
         details = []
         for c in ctx:
             details.append(f"**{c['source']} p.{c['page_start']}**\n> {c['text'].strip()[:1000]}")
@@ -321,28 +320,32 @@ def do_export_docx(question, answer_md, sources_rows):
     except Exception:
         return None
 
-# ---------- Theme ----------
+# ---------- Theme (no .set used; styling via CSS) ----------
 theme = gr.themes.Soft(
     primary_hue="indigo",
     secondary_hue="blue",
     neutral_hue="slate",
-).set(
-    body_background_fill="#0B1220",  # dark-friendly hero
-    block_background_fill="#0F172A",
-    block_shadow="*shadow-lg",
-    radius_size="8px",
 )
 
 # ---------- Gradio UI ----------
-with gr.Blocks(title=f"{APP_NAME} | RAG over PDFs", theme=theme, css="""
+with gr.Blocks(
+    title=f"{APP_NAME} | RAG over PDFs",
+    theme=theme,
+    css="""
 #hero {
   background: radial-gradient(1200px 600px at 20% -10%, rgba(99,102,241,.25), transparent),
               radial-gradient(1000px 500px at 120% 10%, rgba(14,165,233,.20), transparent);
   border: 1px solid rgba(99,102,241,.20);
+  border-radius: 12px;
+  padding: 14px 16px;
 }
 .kpi {text-align:center;padding:12px;border-radius:10px;border:1px solid rgba(255,255,255,.08);}
 .footer {opacity:.8;}
-""") as demo:
+/* Dark-friendly background */
+body, .gradio-container { background: #0B1220 !important; }
+.gradio-container .block, .gradio-container .tabs { background: #0F172A !important; }
+"""
+) as demo:
     # --- Header / Hero ---
     with gr.Group(elem_id="hero"):
         gr.Markdown(
@@ -396,7 +399,7 @@ Upload your papers, build an index, and ask research questions with verifiable,
         with gr.Tab("2) Ask Questions"):
             with gr.Row():
                 with gr.Column(scale=1):
-                    q = gr.Textbox(label="Your question", lines=3, placeholder="e.g., Compare GTAW experimental parameters with citations")
+                    q = gr.Textbox(label="Your question", lines=3, placeholder="e.g., Compare GTAW parameters with citations")
                     must = gr.Textbox(label="Must contain (comma-separated keywords)", placeholder="camera, CMOS, frame rate")
                     with gr.Accordion("Advanced settings", open=False):
                         topk = gr.Slider(1, 20, value=TOP_K_DEFAULT, step=1, label="Top-K passages")
@@ -446,7 +449,7 @@ Upload your papers, build an index, and ask research questions with verifiable,
                 """
             )
 
-# broad compatibility for Spaces
+# Broad compatibility for Spaces
 demo.queue()
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)))