From 516db59d76afb788372b5383729b4e97d960d3dc Mon Sep 17 00:00:00 2001
From: felixwelter <felixwelter@gmail.com>
Date: Mon, 31 Aug 2020 11:45:24 +0200
Subject: [PATCH] Use title focused search index
---
app.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/app.py b/app.py
index 03df656..80a46a8 100644
--- a/app.py
+++ b/app.py
@@ -4,7 +4,8 @@ from pathlib import Path
import pdfplumber
from flask import Flask, render_template, request, redirect, send_file
from werkzeug.utils import secure_filename
-from search_index import SearchIndex
+from search_index import BasicSearchIndex
+from title_focus_search_index import TitleFocusSearchIndex
import os
app = Flask(__name__)
@@ -12,6 +13,7 @@ app = Flask(__name__)
SLIDE_DIR = "slides"
IMAGE_DIR = "img_cache"
+Index = TitleFocusSearchIndex
@app.route('/')
def index():
@@ -36,7 +38,7 @@ def upload():
file_path = os.path.join(Path(SLIDE_DIR), filename)
file.save(file_path)
pdf = pdfplumber.open(file_path)
- index = SearchIndex()
+ index = Index()
for i, page in enumerate(pdf.pages):
text = page.extract_text()
index.add(str(file_path), i, text)
@@ -49,7 +51,7 @@ def upload():
@app.route("/search")
def query():
try:
- index = SearchIndex()
+ index = Index()
query = request.args.get("term")
result = index.search(query)
#pdf = pdfplumber.open(result["path"])
--
GitLab