Add model inference for file

app.py

@@ -1,3 +1,4 @@
+import csv
 import gradio as gr
 import pandas as pd
 from sentiment_analyser import RandomAnalyser, RoBERTaAnalyser, ChatGPTAnalyser
@@ -49,6 +50,21 @@ def analysis(Text):
     return pd.DataFrame([values], columns=keys)
 
 
+def analyse_file(file):
+    output_name = 'output.csv'
+    with open(output_name, mode='w', newline='') as output:
+        writer = csv.writer(output)
+        header = ['Text', 'Label']
+        writer.writerow(header)
+        model = MODEL_MAPPING[OUR_MODEL]
+        with open(file.name) as f:
+            for line in f:
+                text = line[:-1]
+                sentiment = model.predict([text])
+                writer.writerow([text, sentiment[0]])
+    return output_name
+
+
 MODEL_MAPPING = {
     'Random': RandomAnalyser(),
     'RoBERTa': RoBERTaAnalyser(),
@@ -91,6 +107,18 @@ with gr.Blocks(title=TITLE) as demo:
                                         headers=list(MODEL_MAPPING.keys()), interactive=False)
     text_input.submit(analysis, inputs=text_input, outputs=senti_output, show_progress=True)
 
+    gr.Markdown((
+        "在左侧文件框中上传 txt/csv 文件，模型会对输入文本的每一行当作一个文本进行情感分析。"
+        "可以在右侧下载输出文件，输出文件为两列 csv 格式，第一列为原始文本，第二列为分类结果。"
+    ))
+    with gr.Row():
+        with gr.Column():
+            file_input = gr.File(label='File',
+                                 file_types=['.txt', '.csv'])
+        with gr.Column():
+            file_output = gr.File(label='Output')
+    file_input.upload(analyse_file, inputs=file_input, outputs=file_output)
+
     gr.HTML("<H2>Model Evaluation</H2>")
     gr.Markdown((
         "这里是在 StackOverflow4423 数据集上评估我们的模型。"