Update app.py

app.py

@@ -10,12 +10,38 @@ import matplotlib.pyplot as plt
 import numpy as np
 from datetime import datetime
 import scipy
+import shutil
 # os.system('git clone https://github.com/EleutherAI/lm-evaluation-harness')
 # os.system('cd lm-evaluation-harness')
 # os.system('pip install -e .')
 # -i https://pypi.tuna.tsinghua.edu.cn/simple
 # 第一个功能：基于输入文本和对应的损失值对文本进行着色展示
 
+
+csv_file_path = 'data.csv'
+
+def save_and_share_csv():
+    src_path = './data/0309_merge_gjo.csv'
+    dest_dir = './save/'
+    if not os.path.exists(dest_dir):
+        os.makedirs(dest_dir)
+    dest_path = os.path.join(dest_dir, '0309_merge_gjo_shared.csv')
+    shutil.copy(src_path, dest_path)
+    return """
+    <script>
+    alert('Data shared successfully! CSV saved to ./save/ directory.');
+    </script>
+    """
+    #弹窗没有但反正能保存
+
+
+def plot_ppl():
+    df = pd.read_csv(csv_file_path)
+    # 假设df已经有适当的列用于绘图
+    fig = px.line(df, x='date', y='loss_mean_at_1000', color='model', title='PPL with Time')
+    return fig
+
+
 def color_text(text_list=["hi", "FreshEval","!"], loss_list=[0.1,0.7]):
     """
     根据损失值为文本着色。
@@ -454,6 +480,18 @@ with gr.Blocks() as demo:
             see the questions
             '''
         
+    with gr.Row():
+        plot_btn = gr.Button("Generate Plot")
+        share_btn = gr.Button("Share Data")
+    with gr.Row():
+        plot_space = gr.Plot()
+        share_result = gr.Textbox(visible=False)
+    
+    # 当点击“Generate Plot”按钮时，调用plotly_plot_question函数并在plot_space显示结果
+    plot_btn.click(fn=plotly_plot_question, inputs=[], outputs=plot_space)
+    
+    # 当点击“Share Data”按钮时，调用save_and_share_csv函数并在share_result显示结果
+    share_btn.click(fn=save_and_share_csv, inputs=[], outputs=share_result)
 
-demo.launch(debug=True)
 
+demo.launch(share=True,debug=True)