First commit

main.py

@@ -0,0 +1,65 @@
+from email import message
+import streamlit as st
+from writer import write_article, incorporate_feedback, _template
+
+st.title("Linkedin shill")
+
+#! I hate this
+if "feedback_interface" not in st.session_state:
+    st.session_state.feedback_interface = 0
+
+    st.session_state.content = (
+        ":heart_eyes: **Start writing today with the power of AI** :hugging_face:"
+    )
+
+with st.container(border=True):
+    container = st.empty()
+container.markdown(st.session_state.content)
+
+progress_text = "Operation in progress. Please wait."
+
+with st.container():
+    feedback_container = st.empty()
+
+with st.sidebar:
+    with st.form("my_form"):
+        topic = st.text_input("topic", "Oil future")
+        template = st.text_area("template", _template, height=600)
+        purpose = st.text_input("purpose", "Informative")
+        audience = st.text_input("audience", "Linkedin")
+        tone_style = st.text_input("tone style", "engaging and informative")
+        key_points = st.text_input(
+            "key points", "efficiency, climate change, energy dependance"
+        )
+        num_words = st.text_input("number of words", "600")
+        language = st.text_input("language", "english")
+        if st.form_submit_button(label="Submit"):
+            user_inputs = {
+                "topic": topic,
+                "template": template,
+                "purpose": purpose,
+                "audience": audience,
+                "tone_style": tone_style,
+                "key_points": key_points,
+                "num_words": num_words,
+                "language": language,
+            }
+            bar = 1
+            with st.status(progress_text) as status:
+                container.status("Sit back and relax! AI is doing the job for you!")
+                st.session_state.content = write_article(user_inputs)
+                st.session_state.feedback_interface = 1
+                container.markdown(st.session_state.content)
+                status.update(label="Writing complete!", state="complete")
+
+if st.session_state.feedback_interface:
+    with st.container(border=True):
+        messages = st.container(height=300)
+        if feedback := st.chat_input("What do you want to change?"):
+            messages.chat_message("user").write(feedback)
+            container.status("Sit back and relax! AI is doing the job for you!")
+            st.session_state.content = incorporate_feedback(
+                st.session_state.content, feedback
+            )
+            messages.chat_message("ai").write("Done!")
+            container.markdown(st.session_state.content, height=800)

writer.py

@@ -0,0 +1,145 @@
+import dspy
+from dspy import Signature, InputField, Module, ChainOfThought, Predict
+import streamlit as st
+
+dspy.settings.configure(
+    lm=dspy.OpenAI(
+        model="gpt-4-0125-preview",
+        max_tokens=4096,
+        api_key=st.secrets["OpenAI"],
+    )
+)
+
+
+# refin_lm = dspy.OpenAI(
+#     model="gpt-3.5-turbo",
+#     max_tokens=4096,
+#     api_key="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",
+# )
+
+_template = """A few controversial things I believe about {Topic}:
+
+1) {StrongOpinion1}
+
+I’m not saying {SubtleDistinction}.
+
+I’m just saying {ClarifyingStatement}.
+
+- 
+
+2) {StrongOpinion2}
+
+Contrary to popular belief, {ClicheAction} is a myth.
+
+- {FailureExample1}
+- {FailureExample2}
+- {FailureExample3}
+
+{ClicheAction} is a fast-track to {UndesirableOutcome}.
+
+- 
+
+3) {StrongOpinion3}
+
+“Most people” do what “most people” do.
+
+Which leads to…
+
+- {NegativeStat1}
+- {NegativeStat2}
+- {NegativeStat3}
+
+Clearly what “most people do” doesn’t work.
+
+So do something else.
+
+- 
+
+4) {StrongOpinion4}
+
+99% of the time, this is a giant mistake.
+
+Instead, {Name} has a great framework for achieving {Outcome} without falling into {Trap}.
+
+- {Step1}
+- {Step2}
+- {Step3}
+- 
+
+5) {StrongOpinion5}
+
+Pursuing {Outcome} is a silly goal.
+
+Because {UnconventionalReason}.
+
+{DifferentOutcome} is a better pursuit in life.
+
+- {Benefit1}
+- {Benefit2}
+- {Benefit3}"""
+
+
+class GenerateLinkedinArticle(Signature):
+    topic = InputField(desc="Title or brief description of content")
+    template = InputField(prefix="template to follow")
+    purpose = InputField()
+    audience = InputField()
+    tone_style = InputField()
+    key_points = InputField()
+    num_words = InputField()
+    language = InputField()
+    article = dspy.OutputField(desc=f"Linkedin article of {num_words} words")
+
+
+class LinkedinArticle(Module):
+    def __init__(self):
+        super().__init__()
+        self.generate_article = ChainOfThought(GenerateLinkedinArticle)
+
+    def forward(
+        self,
+        topic,
+        template,
+        purpose,
+        audience,
+        tone_style,
+        key_points,
+        num_words,
+        language,
+    ):
+        article_prediction = self.generate_article(
+            topic=topic,
+            template=template,
+            purpose=purpose,
+            audience=audience,
+            tone_style=tone_style,
+            key_points=key_points,
+            num_words=num_words,
+            language=language,
+        )
+        return article_prediction.article
+
+
+class Feedback(Module):
+    def __init__(self):
+        super().__init__()
+        self.feedback = ChainOfThought("original_content, feedback -> refined_content")
+
+    def forward(self, original_content, feedback):
+        corrected_article = self.feedback(
+            original_content=original_content, feedback=feedback
+        ).refined_content
+        return corrected_article
+
+
+def write_article(user_inputs: dict) -> str:
+    article = LinkedinArticle()
+    content = article.forward(**user_inputs)
+    return content
+
+
+def incorporate_feedback(original_content, feedback):
+    refined_content = Feedback().forward(
+        original_content=original_content, feedback=feedback
+    )
+    return refined_content