Added elapsed time
Browse files
app.py
CHANGED
@@ -6,6 +6,7 @@ from opentelemetry import trace
|
|
6 |
from evaluator import Evaluator
|
7 |
from runner import Runner
|
8 |
from settings import Settings
|
|
|
9 |
import os
|
10 |
import pandas as pd
|
11 |
import gradio as gr
|
@@ -45,8 +46,13 @@ EMPTY_RESULTS_TABLE = pd.DataFrame(columns=['task_id', 'question', 'answer'])
|
|
45 |
def run_one(profile: gr.OAuthProfile | None) -> pd.DataFrame:
|
46 |
if not user_logged_in(profile):
|
47 |
return LOGIN_MESSAGE, EMPTY_RESULTS_TABLE
|
|
|
48 |
questions = [evaluator.get_one_question()]
|
49 |
-
|
|
|
|
|
|
|
|
|
50 |
|
51 |
def run_all(profile: gr.OAuthProfile | None) -> pd.DataFrame:
|
52 |
if not user_logged_in(profile):
|
|
|
6 |
from evaluator import Evaluator
|
7 |
from runner import Runner
|
8 |
from settings import Settings
|
9 |
+
import time
|
10 |
import os
|
11 |
import pandas as pd
|
12 |
import gradio as gr
|
|
|
46 |
def run_one(profile: gr.OAuthProfile | None) -> pd.DataFrame:
|
47 |
if not user_logged_in(profile):
|
48 |
return LOGIN_MESSAGE, EMPTY_RESULTS_TABLE
|
49 |
+
start_time = time.time()
|
50 |
questions = [evaluator.get_one_question()]
|
51 |
+
question_answer_pairs = runner.run_agent(questions, profile.username)
|
52 |
+
end_time = time.time()
|
53 |
+
elapsed_time = end_time - start_time
|
54 |
+
message = f"Complete. Elapsed time: {elapsed_time} seconds"
|
55 |
+
return message, question_answer_pairs
|
56 |
|
57 |
def run_all(profile: gr.OAuthProfile | None) -> pd.DataFrame:
|
58 |
if not user_logged_in(profile):
|