added multiple sites

app.py

@@ -11,7 +11,7 @@ from st_copy_to_clipboard import st_copy_to_clipboard
 
 load_dotenv()
 
-def create_article(length_option, articles, params):
+def create_article(length_option, articles, params, web_page_option):
     
     article_string = "; ".join(f"Artikel {index + 1}: {artikel}" for index, artikel in enumerate(articles))
     
@@ -27,7 +27,10 @@ def create_article(length_option, articles, params):
     openai.api_type = os.environ.get("OPEN_API_TYPE")
     openai.api_version = os.environ.get("OPEN_API_VERSION")
 
-    writing_style = os.environ.get("WRITING_STYLE")
+    if web_page_option == "heute.at":
+        writing_style = os.environ.get("WRITING_STYLE_HEUTE")
+    else:
+        writing_style = os.environ.get("WRITING_STYLE_GESUND")
     
     try:
         res = openai.ChatCompletion.create(
@@ -57,13 +60,16 @@ def create_article(length_option, articles, params):
         print(f"Fehler beim erstellen des artikels: {str(e)}")
         st.error(f"Something went wrong: {str(e)}", icon="🚨")
 
-def create_headline(article):
+def create_headline(article, web_page_option):
     openai.api_key = os.environ.get("OPEN_API_KEY")
     openai.api_base = os.environ.get("OPEN_API_BASE")
     openai.api_type = os.environ.get("OPEN_API_TYPE")
     openai.api_version = os.environ.get("OPEN_API_VERSION")
 
-    writing_style = os.environ.get("WRITING_STYLE")
+    if web_page_option == "heute.at":
+        writing_style = os.environ.get("WRITING_STYLE_HEUTE")
+    else:
+        writing_style = os.environ.get("WRITING_STYLE_GESUND")
     
     try:
         res = openai.ChatCompletion.create(
@@ -177,8 +183,8 @@ def decrease_page():
         
 def on_click_handler_generate_article(**kwargs):
     with st.spinner("Generiere Artikel..."):
-        created_article = create_article(kwargs["length_option"],kwargs["final_articles"],kwargs["add_info"])   
-        headline = create_headline(created_article)
+        created_article = create_article(kwargs["length_option"],kwargs["final_articles"],kwargs["add_info"], kwargs["webpage_option"])   
+        headline = create_headline(created_article, kwargs["webpage_option"])
     st.session_state["generated_article"] = created_article
     st.session_state["generated_headline"] = headline
     if st.session_state["process_step"] < 3:
@@ -210,40 +216,47 @@ if "function_state" not in st.session_state:
     st.session_state["function_state"] = True
 if "generated_headline" not in st.session_state:
     st.session_state["generated_headline"] = ""
+if "webpage_option" not in st.session_state:
+    st.session_state["webpage_option"] = "heute.at"
 
 PROCESS_STEPS = ["Artikel Extraktion","Artikel Finalisierung","Artikel Generierung","Artikel Ausgabe"]
 
-def check_password():
-    """Returns `True` if the user had the correct password."""
+# def check_password():
+#     """Returns `True` if the user had the correct password."""
 
-    def password_entered():
-        """Checks whether a password entered by the user is correct."""
-        if hmac.compare_digest(st.session_state["password"], os.environ.get("PASSWORD")):
-            st.session_state["password_correct"] = True
-            del st.session_state["password"]  # Don't store the password.
-        else:
-            st.session_state["password_correct"] = False
+#     def password_entered():
+#         """Checks whether a password entered by the user is correct."""
+#         if hmac.compare_digest(st.session_state["password"], os.environ.get("PASSWORD")):
+#             st.session_state["password_correct"] = True
+#             del st.session_state["password"]  # Don't store the password.
+#         else:
+#             st.session_state["password_correct"] = False
 
-    # Return True if the password is validated.
-    if st.session_state.get("password_correct", False):
-        return True
+#     # Return True if the password is validated.
+#     if st.session_state.get("password_correct", False):
+#         return True
 
-    # Show input for password.
-    st.text_input(
-        "Password", type="password", on_change=password_entered, key="password"
-    )
-    if "password_correct" in st.session_state:
-        st.error("😕 Password incorrect")
-    return False
+#     # Show input for password.
+#     st.text_input(
+#         "Password", type="password", on_change=password_entered, key="password"
+#     )
+#     if "password_correct" in st.session_state:
+#         st.error("😕 Password incorrect")
+#     return False
 
 
-if not check_password():
-    st.stop()  # Do not continue if check_password is not True.
+# if not check_password():
+#     st.stop()  # Do not continue if check_password is not True.
 
 col1, col2 = st.columns([2, 1])
 
 col1.title("TINA")
-col2.image("heute_logo.png")
+if st.session_state["webpage_option"] == "heute.at":
+    col2.image("heute_logo.png")
+else:
+    col2.image("gesundheitstrends_logo.png")
+
+st.radio("Wähle die Webseite aus für die Du einen Artikel generieren möchtest",["heute.at","gesundheitstrends.com"], key="webpage_option")
 
 # with st.sidebar:
 #     st.title("Funktions Auswahl")
@@ -303,9 +316,9 @@ if st.session_state["function_state"]:
             st.text_area("Füge weitere Informationen für den Prompt hinzu, falls nötig:",key="add_info")
             st.write("Artikellänge")
             st.radio("Optionen",["Kurz", "Mittel", "Lang"], key="length_option")
-            st.button("Artikel generieren", key="article_btn", on_click=on_click_handler_generate_article, kwargs={"length_option":st.session_state["length_option"],"final_articles":st.session_state["final_articles"],"add_info":st.session_state["add_info"]})
+            st.button("Artikel generieren", key="article_btn", on_click=on_click_handler_generate_article, kwargs={"length_option":st.session_state["length_option"],"final_articles":st.session_state["final_articles"],"add_info":st.session_state["add_info"], "webpage_option":st.session_state["webpage_option"]})
     elif(st.session_state["selected_page"] == 3):
-        st.write(st.session_state["generated_headline"])
+        st.write(f"**{st.session_state['generated_headline']}**")
         st.write(st.session_state["generated_article"]) 
         st.write("Kopieren Sie den Artikel: ")
         st_copy_to_clipboard(st.session_state["generated_headline"]+"\n"+st.session_state["generated_article"])