Update app.py

app.py

@@ -19,6 +19,7 @@ import requests
 import asyncio
 import random
 import pandas as pd
+from docx import Document
 
 load_dotenv(verbose=True)
 
@@ -145,6 +146,7 @@ def summarize_text(long_text, username):
     #conn.close()
     return summary
 
+
 def respond(ctype, msg, username):
     conn = psycopg2.connect(
         dbname="smair",
@@ -344,6 +346,7 @@ with gr.Blocks(title="Fund Manager Buddy", css="""footer {visibility: hidden;} #
                             summary_submit_btn = gr.Button("要約")
 
                         with gr.TabItem("ファンド投資家向け最新情報"):
+                            report_file = gr.Textbox(label="ファイル名")
                             report_input = gr.Textbox(label="投資家向けのアップデート情報を入力", lines=10, value='''
 自分のプロフィル
 山田 太郎
@@ -377,7 +380,7 @@ with gr.Blocks(title="Fund Manager Buddy", css="""footer {visibility: hidden;} #
                             
     運用資産総額 (AUM): 125.5 億ドル
     ファンド設定日: 2025年1月1日''')
-                            report_output = gr.Textbox(label="レポート", show_copy_button=True)
+                            report_output = gr.File(label="ダウンロード")
                             report_submit_btn = gr.Button("生成")
 
             logout_btn = gr.Button("ログアウト")
@@ -484,7 +487,7 @@ with gr.Blocks(title="Fund Manager Buddy", css="""footer {visibility: hidden;} #
         chat_history.append([msg, bot_message])
         return chat_history, "" # Clear the input after sending
     
-    def generate_report(report_input):
+    def generate_report(report_file, report_input):
         if not report_input:
             return "アップデートがありません。"
 
@@ -532,7 +535,13 @@ with gr.Blocks(title="Fund Manager Buddy", css="""footer {visibility: hidden;} #
             contents=[prompt]
         )
         report_update = gresponse.text
-        return report_update
+
+        doc = Document()
+        doc.add_paragraph(report_update)
+        file_path = report_file
+        doc.save(file_path)
+        return file_path
+        #return report_update
         
     dialog_submit_btn.click(
         generate_response,
@@ -542,7 +551,7 @@ with gr.Blocks(title="Fund Manager Buddy", css="""footer {visibility: hidden;} #
 
     report_submit_btn.click(
         generate_report,
-        inputs=[report_input],
+        inputs=[report_file, report_input],
         outputs=[report_output]
     )