Update app.py

app.py

@@ -27,6 +27,7 @@ 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):
@@ -63,10 +64,8 @@ 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):
@@ -102,23 +101,22 @@ if keyword_id:
         editable_output2 = ""
 
     if st.button("記事構成作成", key=f"run_button_{keyword_id}"):
-        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 st.spinner("関連コンテンツ抽出中..."):  
-                subprocess.run(["python3", "second.py", keyword], check=True)
-                with open("output2.txt", "r", encoding="utf-8") as f:
-                    editable_output2 = f.read()
+        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])
+            with open("output2.txt", "r", encoding="utf-8") as f:
+                editable_output2 = f.read()
                 soup = BeautifulSoup(editable_output2, "html.parser")
                 h_tags = soup.find_all(re.compile("^h[1-3]$"))
                 output2.text(editable_output2)
@@ -133,25 +131,16 @@ if keyword_id:
                         {"name": "output2.txt", "content": editable_output2, "tags": str(h_tags),
                          "timestamp": current_time.isoformat(), "keyword_id": keyword_id}
                     )
-            st.success("処理が完了しました。")
-        except subprocess.CalledProcessError:
-            st.error("記事の構成作成中にエラーが発生しました。もう一度お試しください。")
-        except Exception as e:
-            st.error(f"予期せぬエラーが発生しました：{str(e)}")
+        st.success("処理が完了しました。")
 
     editable_output2 = st.text_area("output2.txtを編集してください:", value=editable_output2)
 
     if st.button("記事作成"):
-        try:
-            with st.spinner("記事作成中..."):
-                subprocess.run(["python3", "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)}")
+        with st.spinner("記事作成中..."):
+            subprocess.run(["python3", "run_third.py", editable_output2, keyword_id])
+            with open("output3.txt", "r", encoding="utf-8") as f:
+                content = f.read()
+            output3.text(content)
 
     if st.button("保存"):
         h2_limit = 5