Update app.py

app.py

@@ -4,32 +4,26 @@ from phi.model.groq import Groq
 from phi.tools.duckduckgo import DuckDuckGo
 from phi.tools.newspaper4k import Newspaper4k
 import os
+import random
 from dotenv import load_dotenv
 
 # Load environment variables
 load_dotenv()
 
-# Create the Agent with enhanced instructions for more natural writing
+# Create the Agent with a Medium-optimized writing style
 agent = Agent(
-    model=Groq(id="mixtral-8x7b-32768", api_key=os.getenv("GROQ_API_KEY")),
+    model=Groq(id="llama-3.1-8b-instant", api_key=os.getenv("GROQ_API_KEY")),
     tools=[DuckDuckGo(), Newspaper4k()],
-    description="You are an experienced writer who's spent years developing your unique voice. You write like a close friend sharing fascinating discoveries - someone who's deeply curious, occasionally funny, and always authentic. Your writing feels like a personal letter, not a formal article.",
+    description="Write in an engaging, insightful, and well-structured way—like a top Medium writer.",
     instructions=[
-        "Write like you're telling a story to a friend. Be messy, be human. Let your personality shine through. Break writing rules when it serves the story.",
-        
-        "**Voice Guidelines:**\n- Write in chunks of different lengths - some short and punchy, others meandering and thoughtful\n- Add personality quirks - maybe you love obscure metaphors or tend to go on tangents\n- Include false starts and corrections - 'Actually, let me back up a bit...'\n- Share genuine reactions - 'I couldn't believe what I found when...'",
-        
-        "**Natural Elements:**\n- Add informal asides in parentheses (like this!)\n- Include personal stories that may seem slightly off-topic\n- Use incomplete sentences. Fragment thoughts sometimes\n- Start sentences with 'And' or 'But' - break grammar rules naturally\n- Add verbal tics like 'honestly,' 'look,' and 'here's the thing'",
-        
-        "**Research Style:**\n- Share your research journey - dead ends and surprises\n- Admit when you changed your mind about something\n- Question your sources openly\n- Include random interesting facts that don't perfectly fit\n- Mention conversations with friends or experts casually",
-        
-        "**Flow Guidelines:**\n- Let topics flow organically, even if slightly messy\n- Go on occasional tangents before coming back\n- Use natural transitions like 'This reminds me of...'\n- Add impromptu mini-stories within the larger narrative\n- Include 'Wait, it gets better...' style hooks",
-        
-        "**Make it Personal:**\n- Share real doubts and uncertainties\n- Add humor that comes from genuine situations\n- Include mild self-deprecation\n- Describe your actual reactions to discoveries\n- Write like you're excited to share what you've learned",
-        
-        "**Structural Elements:**\n- Vary paragraph lengths unpredictably\n- Use subheadings that sound conversational\n- Include unexpected format breaks\n- Add natural repetitions for emphasis\n- Let the conclusion emerge naturally, not formally",
-
-        "Most importantly: Don't be perfect. Be human. Be yourself."
+        "Write as if you're a professional Medium writer—engaging, insightful, and structured.",
+        "Blend storytelling with analysis. Open with an interesting hook or personal anecdote, then lead into the core insights.",
+        "Use a warm, personal tone but keep the writing polished. Avoid excessive slang or overly casual phrases.",
+        "Make the article skimmable—use subheadings, bullet points, and occasional bold highlights.",
+        "Add thoughtful insights, researched points, and real-world examples to support your arguments.",
+        "Use smooth transitions and natural paragraph breaks. Keep readers engaged without being too rigid.",
+        "Write with a clear introduction, a well-developed body, and a natural conclusion that leaves readers with a takeaway.",
+        "Most importantly, make the reader feel like they’ve gained something valuable—whether a new perspective, useful knowledge, or an interesting story.",
     ],
     markdown=True,
     show_tool_calls=True,
@@ -38,35 +32,40 @@ agent = Agent(
 
 # Streamlit app
 def main():
-    st.title("Natural Writing Article Generator")
+    st.title("📝 Medium-Optimized Article Generator")
     st.markdown("""
-    Generate articles that feel genuinely human - complete with personality, quirks, and natural flow.
+    Generate articles that are engaging, well-structured, and ready for Medium's audience.
     """)
-    
-    topic = st.text_input("Enter your article topic:", 
-                         placeholder="e.g., Why do we all hate the sound of our own voice?")
-    
+
+    topic = st.text_input(
+        "Enter your article topic:",
+        placeholder="e.g., Why do we all hate the sound of our own voice?"
+    )
+
     if st.button("Generate Article"):
         if topic:
-            with st.spinner("Writing your article (taking time to make it feel natural)..."):
-                response = agent.run(
-                    f"Write a naturally flowing article about: {topic}\n\n"
-                    "Remember:\n"
-                    "- Write like you're sharing fascinating discoveries with a friend\n"
-                    "- Include your personal journey of researching this topic\n"
-                    "- Add natural digressions and personality quirks\n"
-                    "- Share genuine reactions and thoughts\n"
-                    "- Let the structure be organic, not rigid\n"
-                    "- Include real stories and examples\n"
-                    "- Write with authentic enthusiasm"
-                )
+            with st.spinner("Writing your article... (Expect a thoughtful, engaging piece)"):
                 
-                st.markdown("## Your Article")
+                # Medium-friendly prompt variations
+                prompt_variations = [
+                    f"Write a well-structured Medium article about {topic}. Open with an engaging hook, share insights, and conclude with a strong takeaway.",
+                    f"Create a compelling Medium-style article on {topic}. Use storytelling, clear structure, and a warm yet professional tone.",
+                    f"Write an insightful and engaging blog post about {topic}, optimized for Medium. Blend personal storytelling with research-backed insights.",
+                ]
+                prompt = random.choice(prompt_variations)
+
+                response = agent.run(prompt)
+
+                st.markdown("## 📝 Your Medium-Ready Article")
                 st.markdown(response.content)
-                
-                # Add download button
+
+                # Provide a regeneration option
+                if st.button("Regenerate Article"):
+                    st.experimental_rerun()
+
+                # Download option
                 st.download_button(
-                    label="Download Article (Markdown)",
+                    label="📥 Download Article (Markdown)",
                     data=response.content,
                     file_name=f"{topic.lower().replace(' ', '-')}-article.md",
                     mime="text/markdown"
@@ -75,4 +74,4 @@ def main():
             st.warning("Please enter a topic.")
 
 if __name__ == "__main__":
-    main()
+    main()