Update app.py

app.py

@@ -123,66 +123,45 @@ def chat_with_file_contents(prompt, file_content, model_choice='gpt-3.5-turbo'):
     return response['choices'][0]['message']['content']
 
 
-
-
-
 def link_button_with_emoji(url, title, emoji_summary):
-    emojis = ["💉", "🏥", "🌡️", "🩺", "🌡️", "🔬", "💊", "🧪", "👨‍⚕️", "👩‍⚕️"]
+    emojis = ["💉", "🏥", "🌡️", "🩺", "🔬", "💊", "🧪", "👨‍⚕️", "👩‍⚕️"]
     random_emoji = random.choice(emojis)
     st.markdown(f"[{random_emoji} {emoji_summary} - {title}]({url})")
 
-# MemGPT Paper
-url_memgpt = "https://arxiv.org/abs/2310.08560"
-title_memgpt = "MemGPT"
-emoji_summary_memgpt = "🧠💾 Memory OS"
-link_button_with_emoji(url_memgpt, title_memgpt, emoji_summary_memgpt)
-
-# AutoGen Paper
-url_autogen = "https://arxiv.org/abs/2308.08155"
-title_autogen = "AutoGen"
-emoji_summary_autogen = "🤖🔗 Multi-Agent LLM"
-link_button_with_emoji(url_autogen, title_autogen, emoji_summary_autogen)
-
-# Whisper Paper
-url_whisper = "https://arxiv.org/abs/2212.04356"
-title_whisper = "Whisper"
-emoji_summary_whisper = "🔊🧑‍🚀 Robust STT"
-link_button_with_emoji(url_whisper, title_whisper, emoji_summary_whisper)
-
-# ChatDev Paper
-url_chatdev = "https://arxiv.org/pdf/2307.07924.pdf"
-title_chatdev = "ChatDev"
-emoji_summary_chatdev = "💬💻 Comm. Agents"
-link_button_with_emoji(url_chatdev, title_chatdev, emoji_summary_chatdev)
-
-# 2. Paper Interface with LLM
-def add_paper_buttons():
-    # Expander for MemGPT
-    with st.expander("MemGPT 🧠💾", expanded=False):
-        if st.button("Discuss MemGPT Features"):
-            outline_memgpt = "Memory Hierarchy, Context Paging, Self-directed Memory Updates, Memory Editing, Memory Retrieval, Semantic & Episodic Memory, Emotional Contextual Understanding"
-            chat_with_model("Discuss the key features of MemGPT: " + outline_memgpt, "MemGPT")
-
-    # Expander for AutoGen
-    with st.expander("AutoGen 🤖🔗", expanded=False):
-        if st.button("Explore AutoGen Multi-Agent LLM"):
-            outline_autogen = "Cooperative Conversations, Combining Capabilities, Complex Task Solving, Divergent Thinking, Factuality, Highly Capable Agents, Generic Abstraction"
-            chat_with_model("Explore the key features of AutoGen: " + outline_autogen, "AutoGen")
-
-    # Expander for Whisper
-    with st.expander("Whisper 🔊🧑‍🚀", expanded=False):
-        if st.button("Learn About Whisper STT"):
-            outline_whisper = "Scaling, Deep Learning Approaches, Weak Supervision, Zero-shot Transfer Learning, Accuracy & Robustness, Pre-training Techniques, Broad Environments"
-            chat_with_model("Learn about the key features of Whisper: " + outline_whisper, "Whisper")
-
-    # Expander for ChatDev
-    with st.expander("ChatDev 💬💻", expanded=False):
-        if st.button("Deep Dive into ChatDev"):
+# Define function to add paper buttons and links
+def add_paper_buttons_and_links():
+    col1, col2, col3, col4 = st.columns(4)
+
+    with col1:
+        with st.expander("MemGPT 🧠💾", expanded=False):
+            link_button_with_emoji("https://arxiv.org/abs/2310.08560", "MemGPT", "🧠💾 Memory OS")
+            outline_memgpt = "Memory Hierarchy, Context Paging, Self-directed Memory Updates, Memory Editing, Memory Retrieval, Preprompt Instructions, Semantic Memory, Episodic Memory, Emotional Contextual Understanding"
+            if st.button("Discuss MemGPT Features"):
+                chat_with_model("Discuss the key features of MemGPT: " + outline_memgpt, "MemGPT")
+
+    with col2:
+        with st.expander("AutoGen 🤖🔗", expanded=False):
+            link_button_with_emoji("https://arxiv.org/abs/2308.08155", "AutoGen", "🤖🔗 Multi-Agent LLM")
+            outline_autogen = "Cooperative Conversations, Combining Capabilities, Complex Task Solving, Divergent Thinking, Factuality, Highly Capable Agents, Generic Abstraction, Effective Implementation"
+            if st.button("Explore AutoGen Multi-Agent LLM"):
+                chat_with_model("Explore the key features of AutoGen: " + outline_autogen, "AutoGen")
+
+    with col3:
+        with st.expander("Whisper 🔊🧑‍🚀", expanded=False):
+            link_button_with_emoji("https://arxiv.org/abs/2212.04356", "Whisper", "🔊🧑‍🚀 Robust STT")
+            outline_whisper = "Scaling, Deep Learning Approaches, Weak Supervision, Zero-shot Transfer Learning, Accuracy & Robustness, Pre-training Techniques, Broad Range of Environments, Combining Multiple Datasets"
+            if st.button("Learn About Whisper STT"):
+                chat_with_model("Learn about the key features of Whisper: " + outline_whisper, "Whisper")
+
+    with col4:
+        with st.expander("ChatDev 💬💻", expanded=False):
+            link_button_with_emoji("https://arxiv.org/pdf/2307.07924.pdf", "ChatDev", "💬💻 Comm. Agents")
             outline_chatdev = "Effective Communication, Comprehensive Software Solutions, Diverse Social Identities, Tailored Codes, Environment Dependencies, User Manuals"
-            chat_with_model("Deep dive into the features of ChatDev: " + outline_chatdev, "ChatDev")
+            if st.button("Deep Dive into ChatDev"):
+                chat_with_model("Deep dive into the features of ChatDev: " + outline_chatdev, "ChatDev")
 
+add_paper_buttons_and_links()
 
-add_paper_buttons()
 
 def generate_filename_old(prompt, file_type):
     central = pytz.timezone('US/Central')