Youmnaaaa commited on
Commit
e9ce216
·
verified ·
1 Parent(s): 606a704

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -0
app.py CHANGED
@@ -475,6 +475,29 @@ def normalize_rating(r):
475
  r = float(r)
476
  return round(r/2, 1) if r > 5 else round(r, 1) if r > 0 else 0.0
477
  except: return 0.0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
478
 
479
  """TIME UTILS
480
 
@@ -2202,6 +2225,7 @@ def handle_search_by_item(text: str, session, user_lat=None, user_lon=None) -> d
2202
  3. Return formatted response
2203
  """
2204
  item_query = extract_item_query(text)
 
2205
  if not item_query:
2206
  return {
2207
  "reply": CFG.MSG_NO_PLACE_NAME,
 
475
  r = float(r)
476
  return round(r/2, 1) if r > 5 else round(r, 1) if r > 0 else 0.0
477
  except: return 0.0
478
+ def clean_item_query(text: str) -> str:
479
+ if not text:
480
+ return ""
481
+
482
+ q = str(text)
483
+
484
+ remove_words = [
485
+ "عايزه", "عايزة", "عايز", "عاوز", "عاوزه", "عاوزة",
486
+ "محتاجه", "محتاجة", "محتاج",
487
+ "نفسي", "في",
488
+ "مطعم", "مطاعم", "كافيه", "مكان",
489
+ "بيعمل", "يعمل", "بتعمل",
490
+ "عنده", "عندها",
491
+ "فيه", "فيها",
492
+ "بيقدم", "بتقدم"
493
+ ]
494
+
495
+ for w in remove_words:
496
+ q = q.replace(w, " ")
497
+
498
+ q = re.sub(r"\s+", " ", q).strip()
499
+
500
+ return q
501
 
502
  """TIME UTILS
503
 
 
2225
  3. Return formatted response
2226
  """
2227
  item_query = extract_item_query(text)
2228
+ item_query = clean_item_query(item_query)
2229
  if not item_query:
2230
  return {
2231
  "reply": CFG.MSG_NO_PLACE_NAME,