Update app.py

app.py

@@ -67,6 +67,23 @@ def init_chain():
 
 def main():
     init_page()
+    if __name__ == '__main__':
+    main()
+    # Style adjustments (optional, remove if not needed)
+    st.markdown(
+    """
+    <style>
+    /* Custom style adjustments */
+    .st-emotion-cache-15ecox0 { display: none !important; }
+    @media (max-width: 50.5rem) {
+        .st-emotion-cache-13ln4jf {
+            max-width: calc(0rem + 100vw);
+        }
+    }
+    </style>
+    """,
+        unsafe_allow_html=True,
+    )
     chain = init_chain()
     if chain:
         sender = st.selectbox("送信者",("同僚", "上司", "クライアント", "友人"),key="sender")
@@ -77,22 +94,3 @@ def main():
             result = chain.stream({"sender": sender, "subject": subject, "message": message, "reply": reply})
             st.write(result)   
       
-
-if __name__ == '__main__':
-    main()
-
-# Style adjustments (optional, remove if not needed)
-st.markdown(
-"""
-<style>
-/* Custom style adjustments */
-.st-emotion-cache-15ecox0 { display: none !important; }
-@media (max-width: 50.5rem) {
-    .st-emotion-cache-13ln4jf {
-        max-width: calc(0rem + 100vw);
-    }
-}
-</style>
-""",
-    unsafe_allow_html=True,
-)