Update app.py

app.py

@@ -10,7 +10,6 @@ from googleapiclient.discovery import build
 import urllib.request
 import urllib.error
 import datetime
-import html
 
 @st.cache_resource
 def get_top_urls_and_keyword(keyword):
@@ -27,7 +26,6 @@ def get_top_urls_and_keyword(keyword):
     ).execute()
 
     urls = [item['link'] for item in response["items"][:3]]
-
     return urls, keyword
 
 def get_valid_url(urls):
@@ -64,8 +62,10 @@ db.remove(Query().timestamp.test(lambda x: datetime.datetime.fromisoformat(x) <=
 
 st.title("記事生成ウェブアプリ")
 st.write("このアプリは、与えられたキーワードを使用して記事を生成します。")
+
 new_keyword = st.text_input("キーワード:")
 keyword_id = re.sub(r"\W+", "", new_keyword) if new_keyword else None
+
 last_keyword = db.search(Query().keyword_id.exists())
 
 if new_keyword and (not last_keyword or last_keyword[0]['keyword_id'] != keyword_id):
@@ -84,12 +84,12 @@ if new_keyword and (not last_keyword or last_keyword[0]['keyword_id'] != keyword
 
 if new_keyword:
     urls, keyword = get_top_urls_and_keyword(new_keyword)
-    if len(urls) < 3:  
+    if len(urls) < 3:
         st.error("Google検索の結果が3つ未満です。別のキーワードを試してみてください。")
     else:
         url1, url2, url3 = urls
 
-if keyword_id:  
+if keyword_id:
     output1 = st.empty()
     output2 = st.empty()
     output3 = st.empty()
@@ -101,20 +101,29 @@ if keyword_id:
         editable_output2 = ""
 
     if st.button("記事構成作成", key=f"run_button_{keyword_id}"):
-        with st.spinner("タイトル・見出し作成中..."):
-            urls, keyword = get_top_urls_and_keyword(new_keyword)
-            url1, url2, url3 = urls
-            parsed_urls = [urlparse(url) for url in urls]
-            if len(urls) != len(set(urls)):
-                st.error("異なるURLを入力してください。")
-                st.stop()
-            elif len(set([url.netloc for url in parsed_urls])) != len(urls):
-                st.error("異なるサイトのURLを入力してください。")
-                st.stop()
-            subprocess.run(["python3", "first.py", url1, url2, url3])
-
-        with st.spinner("関連コンテンツ抽出中..."):  # 新しいスピナーを追加
-            subprocess.run(["python3", "second.py", keyword])
+        try:
+            with st.spinner("タイトル・見出し作成中..."):
+                urls, keyword = get_top_urls_and_keyword(new_keyword)
+                url1, url2, url3 = urls
+
+                parsed_urls = [urlparse(url) for url in urls]
+                if len(urls) != len(set(urls)):
+                    st.error("異なるURLを入力してください。")
+                    st.stop()
+                elif len(set([url.netloc for url in parsed_urls])) != len(urls):
+                    st.error("異なるサイトのURLを入力してください。")
+                    st.stop()
+
+                subprocess.run(["python3", "first.py", url1, url2, url3], check=True)
+
+            with open("output1.txt", "r", encoding="utf-8") as f:
+                content = f.read()
+            content = re.sub(r"\n関連するテキスト部分:.*", "", content, flags=re.DOTALL)
+            output1.text(content)
+            db.upsert({"name": "output1.txt", "content": content, "keyword_id": keyword_id},
+                      (Query().name == "output1.txt") & (Query().keyword_id == keyword_id))
+
+            subprocess.run(["python3", "second.py", keyword], check=True)
             with open("output2.txt", "r", encoding="utf-8") as f:
                 editable_output2 = f.read()
                 soup = BeautifulSoup(editable_output2, "html.parser")
@@ -131,23 +140,36 @@ if keyword_id:
                         {"name": "output2.txt", "content": editable_output2, "tags": str(h_tags),
                          "timestamp": current_time.isoformat(), "keyword_id": keyword_id}
                     )
-        st.success("処理が完了しました。")
+
+            st.success("処理が完了しました。")
+        except subprocess.CalledProcessError:
+            st.error("記事の構成作成中にエラーが発生しました。もう一度お試しください。")
+        except Exception as e:
+            st.error(f"予期せぬエラーが発生しました：{str(e)}")
 
     editable_output2 = st.text_area("output2.txtを編集してください:", value=editable_output2)
 
     if st.button("記事作成"):
-        with st.spinner("記事作成中..."):
-            subprocess.run(["python3", "run_third.py", editable_output2, keyword_id])
+        try:
+            with st.spinner("記事作成中..."):
+                subprocess.run(["python3", "run_third.py", editable_output2, keyword_id], check=True)
+
             with open("output3.txt", "r", encoding="utf-8") as f:
                 content = f.read()
             output3.text(content)
+        except subprocess.CalledProcessError:
+            st.error("エラーです。やり直してください。")
+        except Exception as e:
+            st.error(f"予期せぬエラーが発生しました：{str(e)}")
 
     if st.button("保存"):
         h2_limit = 5
         h3_limit = 10
+
         soup = BeautifulSoup(editable_output2, "html.parser")
         h2_count = len(soup.find_all("h2"))
         h3_count = len(soup.find_all("h3"))
+
         if h2_count > h2_limit or h3_count > h3_limit:
             st.error(f"h2タグの数が{h2_limit}を、h3タグの数が{h3_limit}を超えています。")
         elif not is_valid_html(editable_output2):
@@ -157,31 +179,11 @@ if keyword_id:
             with open("output2.txt", "w", encoding="utf-8") as f:
                 f.write(content)
             db.upsert({"name": "output2.txt", "content": content, "timestamp": current_time.isoformat(),
-            "keyword_id": keyword_id}, (Query().name == "output2.txt") & (Query().keyword_id == keyword_id))
+                       "keyword_id": keyword_id}, (Query().name == "output2.txt") & (Query().keyword_id == keyword_id))
             st.write("output2.txt に変更が保存されました。")
 
     if st.button("データクリア"):
         db.remove(Query().keyword_id == keyword_id)
         st.write("データベースがクリアされました。")
-
-    # コピー機能の追加
-    if os.path.exists("output3.txt"):
-        with open("output3.txt", "r", encoding="utf-8") as f:
-            content = f.read()
-        
-        sanitized_content = html.escape(content)
-        st.text_area("コピーするテキスト:", value=sanitized_content, height=200, key="copy_area")
-        
-        copy_button = """
-            <button id="copyButton">
-                テキストをコピー
-            </button>
-            <script>
-                document.getElementById('copyButton').addEventListener('click', function() {
-                    navigator.clipboard.writeText(document.getElementById('copy_area').value);
-                });
-            </script>
-        """
-        st.markdown(copy_button, unsafe_allow_html=True)
 else:
     st.warning("キーワードを入力してください。")