Create app.py

app.py

@@ -0,0 +1,86 @@
+import streamlit as st
+import atproto as atp
+import markdown
+from langchain_cohere import ChatCohere
+from langchain_core.prompts import ChatPromptTemplate
+import os
+
+api_key = st.secrets["COHERE_API_KEY"]
+llm = ChatCohere(cohere_api_key=api_key)
+template = ChatPromptTemplate([
+    (
+        "system",
+        "You are a helpful and efficient feed summarization assistant. You should provide a coincise and well-style, polished overview of what the post authors are telling"
+    ),
+    (
+        "human",
+        "{message}"
+    )
+])
+
+chain = template | llm
+client = atp.Client()
+
+
+def text_inference(message):
+    res = chain.invoke({"message": message})
+    ret = res.content
+    return ret
+
+def get_feed(username: str, password: str):
+    try:
+        client.login(username, password)
+    except Exception as e:
+        return "### Your username or password are wrong :("
+    strtort = ""
+    strsum = ""
+    # Get "Home" page. Use pagination (cursor + limit) to fetch all posts
+    timeline = client.get_timeline(algorithm='reverse-chronological')
+    c = 0
+    for feed_view in timeline.feed:
+        c+=1
+        action = 'New Post🆕'
+        if feed_view.reason:
+            action_by = feed_view.reason.by.handle
+            action = f'Reposted by @{action_by}🔁'
+
+        post = feed_view.post.record
+        author = feed_view.post.author
+
+        strtort += f'<h3>{action}</h3>\n<img alt="Avatar for {author.display_name}" src="{author.avatar}" width=50> <b>{author.display_name}<b>:<br><p>{post.text}</p><hr>'
+        strsum += f'Author: {author.display_name} - Post content: {post.text}'
+        if c>=25:
+            break
+    res = text_inference(strsum)
+    reshtml = markdown.markdown(res)
+    smry = f"\n\n<details>\n\t<summary><b>Feed Summary</b></summary>\n\n{reshtml}\n\n</details>\n\n"
+    strtort = f"{smry}\n\n{strtort}"
+    return strtort
+
+
+if __name__ == "__main__":
+    # Title of the web app
+    st.title("BlueSky User Feed🦋")
+    st.subheader("Your home feed and a summary of it, all in one placr")
+    st.image("https://upload.wikimedia.org/wikipedia/commons/thumb/7/7a/Bluesky_Logo.svg/1920px-Bluesky_Logo.svg.png", width=150)
+    # Input text box for the search query
+    username = st.text_input("You BlueSky username/handle", placeholder="user.bsky.social")
+    password = st.text_input("Password", type="password")
+
+    # Button to initiate search
+    if st.button("Show my feed!"):
+        if username != "" and password!="":
+            results = get_feed(username, password)
+            if results != "### Your username or password are wrong :(":
+                st.write(f"## Home Feed (Following)🏠\n\n----------------------\n\n")
+                st.html(results)
+            else:
+                st.write(results)
+        else:
+            if username == "" and password!="":
+                st.write("### Please enter a username")
+            elif password == "" and username!="":
+                st.write("### Please enter a password")
+            else:
+                st.write("### Please enter a username and a password")
+