Update app.py

app.py

@@ -65,31 +65,12 @@ domain_names = list(DOMAINS.keys())
 domain_texts = list(DOMAINS.values())
 domain_embeddings = model.encode(domain_texts)
 
-# Semantic Scholar + arXiv wrapper
-sch = SemanticScholar()
-
-def fetch_semantic_and_arxiv(query, top_n=5):
-    papers = []
-
-    # Semantic Scholar
-    try:
-        sem_results = sch.search_paper(query, limit=top_n)
-        for p in sem_results:
-            papers.append({
-                "title": p.title,
-                "authors": ", ".join(a.name for a in p.authors[:3]),
-                "year": p.year,
-                "url": p.url,
-                "source": "Semantic Scholar"
-            })
-    except:
-        pass
-
-    # arXiv
+def fetch_arxiv_refs(query, max_results=3):
+    refs = []
     try:
-        search = arxiv.Search(query=query, max_results=top_n)
+        search = arxiv.Search(query=query, max_results=max_results)
         for r in search.results():
-            papers.append({
+            refs.append({
                 "title": r.title,
                 "authors": ", ".join(a.name for a in r.authors[:3]),
                 "year": r.published.year,
@@ -98,10 +79,9 @@ def fetch_semantic_and_arxiv(query, top_n=5):
             })
     except:
         pass
+    return refs
 
-    return papers
-
-def fetch_duckduckgo_links(query, max_results=5):
+def fetch_duckduckgo_links(query, max_results=3):
     links = []
     try:
         with DDGS() as ddgs:
@@ -129,19 +109,19 @@ def classify_math_question(question):
     out = f"<b>Major Domain:</b> {major}<br><i>Reason:</i> {major_reason}<br><br>"
     out += f"<b>Minor Domain:</b> {minor}<br><i>Reason:</i> {minor_reason}<br><br>"
 
-    refs = fetch_semantic_and_arxiv(question)
-    links = fetch_duckduckgo_links(question)
+    refs = fetch_arxiv_refs(question, max_results=3)
+    links = fetch_duckduckgo_links(question, max_results=3)
 
     if refs:
-        out += "<b>Top Academic References:</b><ul>"
+        out += "<b>Top Academic References (arXiv):</b><ul>"
         for p in refs:
-            out += f"<li><b>{p['title']}</b> ({p['year']}) - <i>{p['authors']}</i> [{p['source']}]<br><a href='{p['url']}' target='_blank'>{p['url']}</a></li>"
+            out += f"<li><b>{p['title']}</b> ({p['year']}) - <i>{p['authors']}</i><br><a href='{p['url']}' target='_blank'>{p['url']}</a></li>"
         out += "</ul>"
     else:
         out += "<i>No academic references found.</i><br>"
 
     if links:
-        out += "<b>Top Web Resources:</b><ul>"
+        out += "<b>Top Web Resources (DuckDuckGo):</b><ul>"
         for link in links:
             out += f"<li><b>{link['title']}</b><br>{link['snippet']}<br><a href='{link['url']}' target='_blank'>{link['url']}</a></li>"
         out += "</ul>"
@@ -154,8 +134,8 @@ iface = gr.Interface(
     fn=classify_math_question,
     inputs=gr.Textbox(lines=5, label="Enter Math Question (LaTeX supported)"),
     outputs=gr.HTML(label="Predicted Domains + References"),
-    title="📚 Math Domain Classifier with arXiv + Semantic Scholar + DuckDuckGo",
-    description="Paste any math problem or LaTeX expression and get its major/minor domains + academic references + web links."
+    title="⚡ Fast Math Domain Classifier with arXiv + DuckDuckGo",
+    description="Classifies math problems into major/minor domains and fetches fast references from arXiv + DuckDuckGo."
 )
 
 iface.launch()