다수의 사용자 고려

app.py

@@ -1,13 +1,21 @@
 import gradio as gr
 import pandas as pd
+import random
+
+USERS = ['user1', 'user2', 'user3']
+df_per_user = {}
+df = pd.read_csv("data.csv")
+df['score'] = None
+df['assignee'] = random.choices(USERS, k=len(df))
+for u in USERS:
+    df_per_user[u] = df[df.assignee == u].to_dict('records')
 
 with gr.Blocks() as demo:
     dataset_df = {}
-    state = gr.State(value=0)
+    state = gr.State(value=-1)
     with gr.Row():
         gr.Markdown("# Distributed Evaluation Parallel 😎")
     with gr.Row():
-        file = gr.File(label="Upload a file")
         prev = gr.Button(value="Previous")
         next = gr.Button(value="Next")
         download = gr.File(label="Download a file")
@@ -20,44 +28,41 @@ with gr.Blocks() as demo:
             prediction = gr.Textbox(label="Prediction")
             score = gr.Radio(choices=["Incorrect", "Correct"], label="Score")
     with gr.Row():
-        todos = gr.DataFrame()
-        done = gr.DataFrame()
-    
-
-    def csv2df(file):
-        df = pd.read_csv(file.name)
-        df['score'] = None
-        df_dict = df.to_dict('records')
-        dataset_df.update(dict(df=df, df_dict=df_dict))
-        return update()
+        with gr.Column():
+            gr.Markdown("## TODO")
+            todos = gr.DataFrame()
+        with gr.Column():
+            gr.Markdown("## DONE")
+            done = gr.DataFrame()
 
-    def prev_func(score):
-        df_dict = dataset_df['df_dict']
+    def prev_func(score, request: gr.Request):
+        df_dict = df_per_user[request.username]
         state.value = max(state.value - 1, 0)
         score = df_dict[state.value]['score']
-        gr.Info(f"총 {len(dataset_df['df'])}개 중에 {state.value + 1}번째 데이터입니다.")
-        return [*update(), score]
+        gr.Info(f"{request.username}님, 총 {len(df_dict)}개 중에 {state.value + 1}번째 데이터입니다.")
+        return [*update(request.username), score]
     
-    def next_func(score):
-        df_dict = dataset_df['df_dict']
+    def next_func(score, request: gr.Request):
+        df_dict = df_per_user[request.username]
         df_dict[state.value]['score'] = score
-        state.value = min(state.value + 1, len(dataset_df['df']) - 1)
+        state.value = min(state.value + 1, len(df_dict) - 1)
         score = df_dict[state.value]['score']
-        gr.Info(f"총 {len(dataset_df['df'])}개 중에 {state.value + 1}번째 데이터입니다.")
-        return [*update(), score]
+        gr.Info(f"{request.username}님, 총 {len(df_dict)}개 중에 {state.value + 1}번째 데이터입니다.")
+        return [*update(request.username), score]
 
-    def update():
-        df_dict = dataset_df['df_dict']
+    def update(username):
+        df_dict = df_per_user[username]
         q = df_dict[state.value]['question']
         g = df_dict[state.value]['answer']
         p = df_dict[state.value]['prediction']
         df = pd.DataFrame(df_dict)
         todos = df[df.score.isna()]
         done = df[df.score.isna() == False]
-        return q, g, p, todos, done, "data_backup.csv"
-    file.upload(csv2df, file, [question, ground_truth, prediction, todos, done, download])
+        filename = f"done_{username}.csv"
+        done.to_csv(filename, index=False)
+        return q, g, p, todos, done, filename
     prev.click(prev_func, [score], [question, ground_truth, prediction, todos, done, download, score])
     next.click(next_func, [score], [question, ground_truth, prediction, todos, done, download, score])
 
-demo.queue()
-demo.launch()
+demo.queue(concurrency_count=10)
+demo.launch(auth=[(u, u) for u in USERS])

data.csv

@@ -0,0 +1,25 @@
+,question,answer,prediction,score
+0,한국말로 인사 어떻게 해?,안녕하세요,반가워요1,
+1,한국말로 인사 어떻게 해?,안녕하세요,반가워요2,
+2,한국말로 인사 어떻게 해?,안녕하세요,반가워요3,
+3,한국말로 인사 어떻게 해?,안녕하세요,반가워요1,
+4,한국말로 인사 어떻게 해?,안녕하세요,반가워요2,
+5,한국말로 인사 어떻게 해?,안녕하세요,반가워요3,
+6,한국말로 인사 어떻게 해?,안녕하세요,반가워요1,
+7,한국말로 인사 어떻게 해?,안녕하세요,반가워요2,
+8,한국말로 인사 어떻게 해?,안녕하세요,반가워요3,
+9,한국말로 인사 어떻게 해?,안녕하세요,반가워요1,
+10,한국말로 인사 어떻게 해?,안녕하세요,반가워요2,
+11,한국말로 인사 어떻게 해?,안녕하세요,반가워요3,
+12,한국말로 인사 어떻게 해?,안녕하세요,반가워요1,
+13,한국말로 인사 어떻게 해?,안녕하세요,반가워요2,
+14,한국말로 인사 어떻게 해?,안녕하세요,반가워요3,
+15,한국말로 인사 어떻게 해?,안녕하세요,반가워요1,
+16,한국말로 인사 어떻게 해?,안녕하세요,반가워요2,
+17,한국말로 인사 어떻게 해?,안녕하세요,반가워요3,
+18,한국말로 인사 어떻게 해?,안녕하세요,반가워요1,
+19,한국말로 인사 어떻게 해?,안녕하세요,반가워요2,
+20,한국말로 인사 어떻게 해?,안녕하세요,반가워요3,
+21,한국말로 인사 어떻게 해?,안녕하세요,반가워요1,
+22,한국말로 인사 어떻게 해?,안녕하세요,반가워요2,
+23,한국말로 인사 어떻게 해?,안녕하세요,반가워요3,