s481825/PJN-PROJEKT
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request 'fixed bug caused by empty user input, added full names of models and Alibaba recommended' (#17) from UI_Adjustments into master

Browse Source 
Reviewed-on: #17
This commit is contained in:
s481854 2025-01-14 15:51:44 +01:00
commit 7cb7881e01
2 changed files with 22 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/app/main.py
View File

@ -33,11 +33,17 @@ alibaba_embeddings=load_embeddings(alibaba_embeddings_path)
@app.route("/", methods=["GET", "POST"])
def index():
if request.method == "POST":
user_input = request.form["description"]
model_type = request.form["model_type"]
performance_choice = request.form["preference_choice"]
similarity_metric = request.form["similarity_metric"]
user_input = request.form.get("description", "").strip()
if not user_input:
return render_template(
"index.html",
error="Please provide a description to get movie recommendations."
)
model_type = request.form.get("model_type", "")
performance_choice = request.form.get("preference_choice", "")
similarity_metric = request.form.get("similarity_metric", "")
if model_type == "bert":
model = bert_model
tokenizer = bert_tokenizer

18
src/app/templates/index.html
View File

@ -42,10 +42,16 @@
<h1 class="text-center mb-4">Filmosfera</h1>
<div class="card shadow-sm p-4">
<div id="content">
{% if error %}
<div class="alert alert-danger" role="alert">
{{ error }}
</div>
{% endif %}
<form method="POST">
<div class="mb-3">
<label for="description" class="form-label">Describe the kind of movie you want:</label>
<textarea id="description" name="description" class="form-control" rows="3" required></textarea>
<textarea id="description" name="description" class="form-control" rows="3"></textarea>
</div>
<div class="mb-3">
<label for="preference_choice" class="form-label">Choose performance preference:</label>
@ -65,10 +71,10 @@
<div class="mb-3">
<label for="model_type" class="form-label">Choose a model:</label>
<select id="model_type" name="model_type" class="form-select">
<option value="bert">BERT</option>
<option value="sentence-transformer">SentenceTransformer</option>
<option value="sentence-bert">Sentence-BERT</option>
<option value="alibaba">Alibaba</option>
<option value="alibaba">Alibaba-NLP/gte-large-en-v1.5 (recommended)</option>
<option value="bert">google-bert/bert-base-uncased</option>
<option value="sentence-transformer">sentence-transformers/all-MiniLM-L6-v2</option>
<option value="sentence-bert">sentence-transformers/paraphrase-mpnet-base-v2</option>
</select>
</div>
<div class="mb-3">
@ -94,4 +100,4 @@
</div>
</body>
</html>
</html>