Thanh Vinh Vo commited on
Commit
99dd386
·
1 Parent(s): ed781c5
Files changed (2) hide show
  1. app.py +6 -7
  2. requirements.txt +2 -0
app.py CHANGED
@@ -370,6 +370,9 @@ class BasicAgent:
370
  "openpyxl",
371
  "json",
372
  "wikipediaapi",
 
 
 
373
  ],
374
  name="multimodal_agent",
375
  description="""
@@ -409,6 +412,9 @@ class BasicAgent:
409
  "openpyxl",
410
  "json",
411
  "wikipediaapi",
 
 
 
412
  ],
413
  planning_interval=2,
414
  max_steps=10,
@@ -425,15 +431,8 @@ class BasicAgent:
425
  - Please answer as concisely as possible.
426
  - If the question asks for a number, please return a numerical answer without unit (unless unit is specifically asked for). For example: 3 instead of three, 0 instead of None, 3 instead of $3.
427
  - If the question asks for a number with specific decimal places, please format the number into string with the same decimal places. For example: 3.00 instead of 3.
428
- - To get table data from Wikipedia or any HTML page content, please follow the following steps:
429
- - 1. Identify the section you want to extract.
430
- - 2.Use `pandas.read_html` to extract the table data from that section.
431
  - If the question asks for a list, please make sure that the elements are separated by a comma(`,`) and a space(` `). For example: `1, 2, 3` instead of `1,2,3`.
432
  - If the question asks for name without abbreviations, please ALWAYS ask `multimodal_agent` for the FULL name of final answer to ensure NO abbreviation is included in Final Answer. For example: `United States` instead of `US`.
433
- - Just give up and return a guess answer if the request requires you to do any of the following:
434
- - Watch Youtube videos.
435
- - Convert Youtube videos to audio.
436
- - Solve chess puzzle from an image.
437
  """
438
  if "food" in question.lower() or "drink" in question.lower():
439
  prompt = f"""
 
370
  "openpyxl",
371
  "json",
372
  "wikipediaapi",
373
+ "pytube",
374
+ "pytubefix",
375
+ "pytubefix.cli",
376
  ],
377
  name="multimodal_agent",
378
  description="""
 
412
  "openpyxl",
413
  "json",
414
  "wikipediaapi",
415
+ "pytube",
416
+ "pytubefix",
417
+ "pytubefix.cli",
418
  ],
419
  planning_interval=2,
420
  max_steps=10,
 
431
  - Please answer as concisely as possible.
432
  - If the question asks for a number, please return a numerical answer without unit (unless unit is specifically asked for). For example: 3 instead of three, 0 instead of None, 3 instead of $3.
433
  - If the question asks for a number with specific decimal places, please format the number into string with the same decimal places. For example: 3.00 instead of 3.
 
 
 
434
  - If the question asks for a list, please make sure that the elements are separated by a comma(`,`) and a space(` `). For example: `1, 2, 3` instead of `1,2,3`.
435
  - If the question asks for name without abbreviations, please ALWAYS ask `multimodal_agent` for the FULL name of final answer to ensure NO abbreviation is included in Final Answer. For example: `United States` instead of `US`.
 
 
 
 
436
  """
437
  if "food" in question.lower() or "drink" in question.lower():
438
  prompt = f"""
requirements.txt CHANGED
@@ -16,3 +16,5 @@ openai-whisper
16
  openpyxl
17
  youtube-transcript-api
18
  wikipedia-api
 
 
 
16
  openpyxl
17
  youtube-transcript-api
18
  wikipedia-api
19
+ pytube
20
+ pytubefix