first commit

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.9
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . .
+
+CMD ["gunicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,48 @@
+from flask import Flask, render_template, request, jsonify
+import numpy as np
+import faiss
+from sentence_transformers import SentenceTransformer
+import os
+import numpy as np
+
+
+app = Flask(__name__)
+
+
+indexIVFPQ = faiss.read_index("indexIVFPQ .index")
+
+from sentence_transformers import SentenceTransformer
+model = SentenceTransformer('bert-base-nli-mean-tokens') 
+
+import json
+with open('wiki_entries.json') as json_file:
+    data = json.load(json_file)
+lines=[]
+for key in data:
+    #print(key)
+    doc=data[key]
+    for k in doc:
+        lines.append(doc[k])
+
+
+@app.route('/')
+def home():
+    return render_template('index.html')
+
+@app.route('/search', methods=['POST'])
+def search():
+    query = request.form['query']
+
+    if not query:
+        return render_template('index.html', error='Query is required')
+    k = 10
+    xq = model.encode([query])
+    D, I = indexIVFPQ .search(xq, k) 
+    line=[]
+    for i in I[0]:
+    	sen=lines[i]
+    	line.append(sen)
+    return render_template('index.html', query=query, top_n_docs=line)
+
+if __name__ == '__main__':
+    app.run(debug=True)

requirments.txt

@@ -0,0 +1,19 @@
+blinker==1.7.0
+click==8.1.7
+faiss-cpu==1.8.0
+Flask==3.0.2
+gensim==4.3.2
+itsdangerous==2.1.2
+Jinja2==3.1.3
+MarkupSafe==2.1.5
+numpy==1.26.4
+pandas==2.2.0
+python-dateutil==2.8.2
+pytz==2024.1
+rank-bm25==0.2.2
+scipy==1.12.0
+six==1.16.0
+smart-open==6.4.0
+tzdata==2023.4
+Werkzeug==3.0.1
+sentence_transformers