Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import pandas as pd
 from transformers import pipeline
 
 # transformers パイプラインのインポート
@@ -29,11 +30,16 @@ text = st.text_area("入力テキスト", default_text)
 
 # 翻訳ボタンが押されたときの処理
 if st.button("翻訳する"):
-    # Create an empty element for displaying the result
-    result_container = st.empty()
-    
     # Perform translation
     result = st.session_state.session_models[model](text)[0]['translation_text']
     
-    # Update the result_container with the translated text
-    result_container.write(f"翻訳結果: {result}")
+    # Save the data to a CSV file
+    data = {'ID': [1], 'Original Text': [text], 'Result': [result]}
+    df = pd.DataFrame(data)
+    df.to_csv('translation_data.csv', mode='a', header=not st.session_state.csv_created, index=False)
+    
+    # Update the CSV creation flag
+    st.session_state.csv_created = True
+    
+    # Display the result
+    st.write(f"翻訳結果: {result}")