Update app.py

app.py

@@ -175,20 +175,17 @@ def ask_paper(question, mode):
     elif mode == "表形式分析":
 
         instruction = """
-論文から情報を抽出し、
-必ずMarkdown表のみで回答してください。
+論文から情報を抽出し、結果をCSV形式のみで出力してください。
 
-列例
+例
 
-|項目|内容|
-|---|---|
-|データセット|xxx|
-|評価指標|xxx|
-|精度|xxx|
-|提案手法|xxx|
-|比較手法|xxx|
+項目,内容
+データセット,MNIST
+評価指標,Accuracy
+精度,98.3%
 
-文章は書かないこと。
+説明文は禁止です。
+CSVのみ出力してください。
 """
 
     else:
@@ -321,7 +318,19 @@ def run_analysis(question, mode):
 
     if mode == "表形式分析":
 
-        df = markdown_table_to_df(result)
+        try:
+
+            df = pd.read_csv(
+                StringIO(result)
+            )
+
+        except Exception as e:
+
+            print(e)
+
+            df = pd.DataFrame(
+                {"エラー": ["表に変換できませんでした"]}
+            )
 
         return (
             "",
@@ -454,12 +463,12 @@ with gr.Blocks() as app:
                 ],
                 outputs=[
                     answer,
-                    table_output
+                    table_output = gr.Dataframe(visible=False,label="分析表")
                 ]
             )
 
             download_btn = gr.Button(
-                "💾 この対話履歴を .txt で保存する",
+                "💾 この対話をテキストファイル(.txt)で保存する",
                 variant="secondary"
             )