Update app.py

app.py

@@ -43,6 +43,33 @@ def kousei(csv_file, input_text):
     # 各プロンプトをAPIに送信（この部分は実装に応じて修正が必要です）
     prompts['response'] = prompts['prompt_after'].apply(get_chatgpt_response)
     print(prompts['response'])
+
+    def check(csv_file, input_text):
+    prompt_text = input_text + "振り分けは全ての問題において、該当しない場合も含めて、必ず全問題、[id]に続けて結果を各フォーマットで返してください。[id]の後は改行しないで。[id]はリクエストと完全一致の形式で。（必ず[問題ID]が５つ表示されるはずです）指摘方法は、該当しない場合は「該当なし」、該当する場合は「該当あり」\n#リスト"
+    
+    # CSVファイルを読み込む
+    df = pd.read_csv(csv_file)
+    # 'id'列のデータ型を文字列に変換
+    df['id'] = df['id'].astype(str)
+    # DataFrameを5行ごとに処理するためのグループ化
+    df['group_id'] = df.index // 5
+    grouped = df.groupby('group_id')
+
+    # 各グループに対してフォーマットされたプロンプトを作成
+    def create_prompt(group, base_prompt):
+        prompt = base_prompt
+        for _, row in group.iterrows():
+            prompt += f"\n[{row['id']}]\n{row['原稿']}"
+            print(prompt)
+        return prompt
+    
+    # 各グループごとのプロンプトを生成
+    prompts = grouped.apply(lambda g: create_prompt(g, prompt_text))
+    prompts = prompts.reset_index(name='prompt_after')
+    
+    # 各プロンプトをAPIに送信（この部分は実装に応じて修正が必要です）
+    prompts['response'] = prompts['prompt_after'].apply(get_chatgpt_response)
+    print(prompts['response'])
     
     
 
@@ -121,7 +148,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             with gr.Row():
                 inputs=gr.File(label="CSVファイルをアップロード")
                 outputs=gr.File(label="ダウンロード", file_count="singular")
-            gr.Button("問題生成").click(
+            gr.Button("校正スタート").click(
                 kousei,
                 inputs=[inputs,input_text_kousei],
                 outputs=[outputs]
@@ -133,5 +160,16 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                  submit_button = gr.Button("音声ファイルを生成")
              file_output = gr.File(label="ダウンロード")
              submit_button.click(fn=text_to_speech, inputs=[file_input], outputs=[file_output])
+     with gr.Tab("カテゴリ分類分類"):
+        with gr.Column():
+            input_text_kousei = gr.Textbox(label="分類観点を入力してください。",value="xxxxxxxに関する内容の場合は「該当あり」と書いてください。")
+            with gr.Row():
+                inputs=gr.File(label="CSVファイルをアップロード")
+                outputs=gr.File(label="ダウンロード", file_count="singular")
+            gr.Button("分類スタート").click(
+                kousei,
+                inputs=[inputs,input_text_kousei],
+                outputs=[outputs]
+            )
 demo.launch(share=True)