Searching

app.py

@@ -1,19 +1,42 @@
+import sqlite3
+import pandas as pd
+
 from fastapi import FastAPI
 from fastapi.staticfiles import StaticFiles
 from starlette.responses import FileResponse, RedirectResponse
 
+from paths import INDEX_PATH
+
 app = FastAPI()
 
 
 @app.get("/search")
 def search(s: str):
-    return RedirectResponse(f'https://www.google.com/search?q={s}')
+    if '—' in s:
+        url = s.split('—')[1].strip()
+    else:
+        url = f'https://www.google.com/search?q={s}'
+    return RedirectResponse(url)
 
 
 @app.get("/complete")
 def complete(q: str):
-    all_titles = ['some', 'nice results', 'here']
-    return [q, all_titles]
+    terms = [x.lower() for x in q.split()]
+    con = sqlite3.connect(INDEX_PATH)
+    in_part = ','.join('?'*len(terms))
+    query = f"""
+        SELECT title, url, count(*)
+        FROM terms INNER JOIN pages
+        ON terms.page_id = pages.id
+        WHERE term IN ({in_part})
+        GROUP BY title, url
+        ORDER BY 3 DESC
+    """
+
+    data = pd.read_sql(query, con, params=terms)
+    results = data.apply(lambda row: f'{row.title} — {row.url}', axis=1)
+    print("Results", results)
+    return [q, results.to_list()[:5]]
 
 
 @app.get('/')