Update utils/utils.py
Browse files- utils/utils.py +4 -4
utils/utils.py
CHANGED
|
@@ -3,19 +3,19 @@ import pandas as pd
|
|
| 3 |
from docx import Document
|
| 4 |
from pptx import Presentation
|
| 5 |
|
| 6 |
-
def get_questions(file_path, level=
|
| 7 |
df = pd.read_json(file_path, lines=True)
|
| 8 |
|
| 9 |
-
if level
|
| 10 |
df = df[df["Level"] == level]
|
| 11 |
|
| 12 |
result=[]
|
| 13 |
|
| 14 |
for index, row in df.iterrows():
|
| 15 |
-
if level
|
| 16 |
result.append([row["Question"], row["Level"], row["Final answer"], row["file_name"]])
|
| 17 |
else:
|
| 18 |
-
result.append([row["Question"], "", row["Final answer"], row["file_name"]])
|
| 19 |
|
| 20 |
return result
|
| 21 |
|
|
|
|
| 3 |
from docx import Document
|
| 4 |
from pptx import Presentation
|
| 5 |
|
| 6 |
+
def get_questions(file_path, level=0):
|
| 7 |
df = pd.read_json(file_path, lines=True)
|
| 8 |
|
| 9 |
+
if level > 0:
|
| 10 |
df = df[df["Level"] == level]
|
| 11 |
|
| 12 |
result=[]
|
| 13 |
|
| 14 |
for index, row in df.iterrows():
|
| 15 |
+
if level > 0:
|
| 16 |
result.append([row["Question"], row["Level"], row["Final answer"], row["file_name"]])
|
| 17 |
else:
|
| 18 |
+
result.append([row["Question"], "0", row["Final answer"], row["file_name"]])
|
| 19 |
|
| 20 |
return result
|
| 21 |
|