Update app.py

app.py

@@ -46,6 +46,57 @@ def get_related_studies(article: str):
             st.error(f"Something went wrong: {str(e)}", icon="🚨")
 
 
+def get_takeaways(article : str):
+    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")
+    takeaway_query = os.environ.get("takeaway")
+    with st.spinner("Creating Takeaways"):
+        try:
+            res = openai.ChatCompletion.create(
+                engine="gpt-4-1106",
+                temperature=0.2,
+                messages=[
+                    {
+                        "role": "system",
+                        "content": f" The article you have written is as follows: {article}.",
+                    }
+                ],
+            )
+            st.session_state["takeaways"] = res["choices"][0]["message"]["content"]
+        except Exception as e:
+            print(f"Fehler beim extrahieren der Query: {str(e)}")
+            st.error(f"Something went wrong: {str(e)}", icon="🚨")
+            st.session_state["takeaways"] = None
+            
+
+    
+def get_faq(article : str):
+    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")
+    faq_query = os.environ.get("faq")
+
+    with st.spinner("Creating FAQ"):
+        try:
+            res = openai.ChatCompletion.create(
+                engine="gpt-4-1106",
+                temperature=0.2,
+                messages=[
+                    {
+                        "role": "system",
+                        "content": f" The article you have written is as follows: {article}.",
+                    }
+                ],
+            )
+            st.session_state["faq"] = res["choices"][0]["message"]["content"]
+        except Exception as e:
+            print(f"Fehler beim extrahieren der Query: {str(e)}")
+            st.error(f"Something went wrong: {str(e)}", icon="🚨")
+            st.session_state["faq"] = None
+
 def extract_scholar_query(article: str):
     openai.api_key = os.environ.get("OPEN_API_KEY")
     openai.api_base = os.environ.get("OPEN_API_BASE")
@@ -77,6 +128,10 @@ def create_article(length_option, articles, params, web_page_option):
         length = os.environ.get("MEDIUM_LENGTH")
     elif length_option == "Lang":
         length = os.environ.get("LONG_LENGTH")
+    elif length_option == "SEO":
+        length = os.environ.get("SEO_LENGTH")
+    elif length_option == "SEO Plus"
+        length = os.environ.get("SEO_PLUS_LENGTH")
 
     openai.api_key = os.environ.get("OPEN_API_KEY")
     openai.api_base = os.environ.get("OPEN_API_BASE")
@@ -123,7 +178,7 @@ def create_article(length_option, articles, params, web_page_option):
             messages = [
                 {
                     "role": "system",
-                    "content": f"You are a professional journalist whose task is to write an article based on your own notes. This article should be written in the following writing style: {writing_style} The length of your article should be {length} sentences long.",
+                    "content": f"You are a professional journalist whose task is to write an article based on your own notes. This article should be written in the following writing style: {writing_style} .It is important that the length of your article should be {length} words long.",
                 },
                 {
                     "role": "system",
@@ -296,7 +351,7 @@ def on_click_handler_generate_article(**kwargs):
             kwargs["length_option"],
             kwargs["final_articles"],
             kwargs["add_info"],
-            kwargs["webpage_option"],
+            kwargs["webpage_option"],            
         )
         headline = create_headline(created_article, kwargs["webpage_option"])
     db_analytics_item = {
@@ -586,7 +641,7 @@ if st.session_state["function_state"]:
                 key="add_info",
             )
             st.write("Artikellänge")
-            st.radio("Optionen", ["Kurz", "Mittel", "Lang"], key="length_option")
+            st.radio("Optionen", ["Kurz", "Mittel", "Lang", "SEO", "SEO Plus"], key="length_option")
             st.button(
                 "Artikel generieren",
                 key="article_btn",
@@ -596,6 +651,8 @@ if st.session_state["function_state"]:
                     "final_articles": st.session_state["final_articles"],
                     "add_info": st.session_state["add_info"],
                     "webpage_option": st.session_state["webpage_option"],
+                    "takeaways" : st.session_state["takeaways"]
+                    "faq" : st.session_state["faq"]
                 },
             )
     elif st.session_state["selected_page"] == 3:
@@ -615,11 +672,28 @@ if st.session_state["function_state"]:
                 st.write(f"- [{result['title']}]({result['link']})")
         else:
             st.write("Keine relevanten Studien gefunden.")
+            
+        if st.session_state["takeaways"]:
+            st.write("Hier sind einige Takeaways die wichtig sein könnten:")
+            st.write(st.session_state["takeaways"])
+        
+        if st.session_state["faq"]:
+            st.write("Hier sind FAQs zu dem Artikel:")
+            st.write(st.session_state["faq"])
+        
         st.button(
             "Relevante Studien finden",
             on_click=get_related_studies,
             args=(st.session_state["generated_article"],),
         )
+        st.button(
+            "Key Takeaways generieren",
+            onclick=get_takeaways(st.session_state["generated_article"])
+        )
+        st.button(
+            "FAQ generieren",
+            onclick=get_faq(st.session_state["generated_article"])
+        )
         st.button(
             "Neuen Artikel generieren", key="reset_btn", on_click=reset_session_state
         )
@@ -629,7 +703,7 @@ else:
     )
     st.text_area(label="Artikel input:", key="keyword_article_input")
     st.write("Artikellänge")
-    st.radio("Optionen", ["Kurz", "Mittel", "Lang"], key="length_option")
+    st.radio("Optionen", ["Kurz", "Mittel", "Lang", "SEO", "SEO Plus"], key="length_option")
     st.button(
         "Artikel generieren",
         key="article_btn",