Update app.py

app.py

@@ -61,7 +61,11 @@ def get_takeaways(article : str):
                     {
                         "role": "system",
                         "content": f" The article you have written is as follows: {article}.",
-                    }
+                    },
+                    {
+                    "role": "system",
+                    "content": f"Schreibe mir zu diesen Artikel Key Takeaways nach folgenden Regeln {takeaway_query}.",
+                    },
                 ],
             )
             st.session_state["takeaways"] = res["choices"][0]["message"]["content"]
@@ -88,7 +92,11 @@ def get_faq(article : str):
                     {
                         "role": "system",
                         "content": f" The article you have written is as follows: {article}.",
-                    }
+                    },
+                    {
+                    "role": "system",
+                    "content": f"Schreibe mir zu diesen Artikel Frequently Asked Questions nach folgenden Regeln {faq_query}.",
+                    },
                 ],
             )
             st.session_state["faq"] = res["choices"][0]["message"]["content"]
@@ -735,6 +743,17 @@ else:
             on_click=get_related_studies,
             args=(st.session_state["generated_article"],),
         )
+        if "takeaways" in st.session_state:
+            st.write("Hier sind einige Takeaways die wichtig sein könnten:")
+            st.write(st.session_state["takeaways"])
+        
+        if "faq" in st.session_state:
+            st.write("Hier sind FAQs zu dem Artikel:")
+            st.write(st.session_state["faq"])
+        
+        st.button("Key Takeaways generieren", on_click=lambda: get_takeaways(st.session_state["generated_article"]))
+        st.button("FAQ generieren", on_click=lambda: get_faq(st.session_state["generated_article"]))
+
         st.button(
             "Neuen Artikel generieren", key="reset_btn", on_click=reset_session_state
         )