Update app.py

app.py

@@ -1,8 +1,7 @@
+import time
 import random
 from sympy import symbols, expand
 import streamlit as st
-from docx import Document as DocxDocument
-from docx.shared import Pt
 
 def generate_problem(min_value, max_value):
     x = symbols('x')
@@ -23,24 +22,14 @@ def generate_problem(min_value, max_value):
     return equation_str
 
 def save_to_text(problems):
-    text_content = ""
-    for index, problem in enumerate(problems, start=1):
-        text_content += f"Problem {index}:\n{problem}\n\n"
-    return text_content.encode("utf-8")
+    timestamp = int(time.time())
+    filename = f"generated_problems_{timestamp}.txt"
 
-def save_to_docx(problems, filename):
-    document = DocxDocument()
+    with open(filename, 'w', encoding='utf-8') as file:
+        for index, problem in enumerate(problems, start=1):
+            file.write(f"Problem {index}:\n{problem}\n\n")
 
-    # 設定したフォントとサイズで新しいスタイルを作成
-    style = document.styles['Normal']
-    font = style.font
-    font.name = 'Arial'
-    font.size = Pt(12)
-
-    for index, problem in enumerate(problems, start=1):
-        document.add_paragraph(f"Problem {index}:\n{problem}\n")
-
-    document.save(filename)
+    return filename
 
 def main():
     st.title("数学問題生成アプリ")
@@ -61,16 +50,9 @@ def main():
             st.write(f"Problem {index}:\n{problem}\n")
 
         # ダウンロードボタン
-        text_filename = st.text_input("テキストファイルのファイル名を入力してください", "generated_problems.txt")
-        docx_filename = st.text_input("Word文書のファイル名を入力してください", "generated_problems.docx")
-
-        # ダウンロードボタンを生成
-        text_button = st.download_button("テキストファイルをダウンロード", save_to_text(generated_problems), key="text")
-        docx_button = st.download_button("Word文書をダウンロード", save_to_docx(generated_problems, docx_filename), key="docx")
-
-        # ボタンが押されたときにメッセージを表示
-        if text_button or docx_button:
-            st.success(f"問題をファイルに保存しました: {text_filename}, {docx_filename}")
+        if st.button("テキストファイルをダウンロード"):
+            filename = save_to_text(generated_problems)
+            st.success(f"問題をテキストファイルに保存しました: {filename}")
 
 if __name__ == "__main__":
     main()