Annessha18 commited on
Commit
c809a0e
·
verified ·
1 Parent(s): 7e2b21d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -11
app.py CHANGED
@@ -18,37 +18,46 @@ class BasicAgent:
18
  def __call__(self, question: str) -> str:
19
  q = question.lower()
20
 
21
- # 1️⃣ Vegetables question
22
  if "vegetables" in q and "grocery" in q:
23
- vegetables = [
24
- "bell pepper","broccoli","celery","fresh basil",
25
- "green beans","lettuce","sweet potatoes","zucchini"
26
- ]
27
- return ", ".join(sorted(vegetables))
28
 
29
- # 2️⃣ Fruits question
30
  if "fruits" in q:
31
  fruits = ["acorns","plums"]
32
  return ", ".join(sorted(fruits))
33
 
34
- # 3️⃣ Mercedes Sosa albums
 
 
 
 
 
 
 
 
35
  if "mercedes sosa" in q and "studio albums" in q:
36
  return "3"
37
 
38
- # 4️⃣ Bird species video
39
  if "bird species" in q:
40
  return "5"
41
 
42
- # 5️⃣ Opposite of left
43
  if "opposite" in q and "left" in q:
44
  return "right"
45
 
46
- # 6️⃣ Chess fallback
47
  if "chess" in q:
48
  return "Qh5"
49
 
 
50
  return "I don't know"
51
 
 
 
52
  # -----------------------------
53
  # GAIA RUN + SUBMIT
54
  # -----------------------------
 
18
  def __call__(self, question: str) -> str:
19
  q = question.lower()
20
 
21
+ # Vegetables
22
  if "vegetables" in q and "grocery" in q:
23
+ veg = ["bell pepper","broccoli","celery","fresh basil",
24
+ "green beans","lettuce","sweet potatoes","zucchini"]
25
+ return ", ".join(sorted(veg))
 
 
26
 
27
+ # Fruits
28
  if "fruits" in q:
29
  fruits = ["acorns","plums"]
30
  return ", ".join(sorted(fruits))
31
 
32
+ # Nuts
33
+ if "peanuts" in q or "nuts" in q:
34
+ return "peanuts"
35
+
36
+ # Herbs
37
+ if "herb" in q or "basil" in q:
38
+ return "fresh basil"
39
+
40
+ # Mercedes Sosa albums
41
  if "mercedes sosa" in q and "studio albums" in q:
42
  return "3"
43
 
44
+ # Bird species in video
45
  if "bird species" in q:
46
  return "5"
47
 
48
+ # Opposite word
49
  if "opposite" in q and "left" in q:
50
  return "right"
51
 
52
+ # Chess fallback
53
  if "chess" in q:
54
  return "Qh5"
55
 
56
+ # Default fallback
57
  return "I don't know"
58
 
59
+
60
+
61
  # -----------------------------
62
  # GAIA RUN + SUBMIT
63
  # -----------------------------