Update app.py

app.py

@@ -1,116 +1,73 @@
+import os
 import streamlit as st
-from streamlit.components.v1 import html
 from groq import Groq
-import os
+from streamlit.components.v1 import html
 
 # Load Groq API key from environment variable
 groq_api_key = os.getenv('api_key')
 if not groq_api_key:
     raise ValueError("GROQ_API_KEY environment variable is not set.")
 
-# Initialize Groq client with API key
+# Initialize Groq client
 groq_client = Groq(api_key=groq_api_key)
 
-# Set the title of the app
-st.title("Social Media Post Maker for Data Science")
+# Few-shot examples for inspiration
+few_shot_examples = """
+🎢 **Data Scaling Showdown: Normalization vs Standardization!** 📊
 
-# Custom CSS for a professional look
-css = """
-    <style>
-        body {
-            background-color: #f0f2f6;
-            font-family: 'Arial', sans-serif;
-        }
-        .stTextInput, .stButton {
-            margin-top: 20px;
-        }
-        .custom-box {
-            padding: 20px;
-            background-color: #ffffff;
-            border-radius: 8px;
-            box-shadow: 0px 4px 10px rgba(0, 0, 0, 0.1);
-        }
-        .error-msg {
-            color: red;
-            font-weight: bold;
-        }
-        .copy-btn {
-            background-color: #4CAF50;
-            color: white;
-            border: none;
-            padding: 10px 20px;
-            border-radius: 5px;
-            cursor: pointer;
-        }
-        .copy-btn:hover {
-            background-color: #45a049;
-        }
-    </style>
-"""
+Ever felt like your data is on a rollercoaster? 🎡 Some values reaching for the stars, others barely off the ground? Time to level the playing field with the dynamic duo of data scaling: **Normalization** and **Standardization!** 💪
 
-html(css)
+🌈 **Normalization: The Range Ranger**
+🔸 What: Squeezes your data into a cozy 0-1 range.
+🔸 Why: Perfect for when your features are as different as apples and skyscrapers! 🍎🏙️
+🔸 Best for: k-NN, neural networks, and data that's living its best non-normal life.
+🔸 Superpower: Taming wild data ranges without breaking a sweat!
 
-st.markdown(
-    """
-    <div class='custom-box'>
-    <h2>Generate Data Science Posts</h2>
-    <p>Create engaging social media content with AI! 🚀💻</p>
-    </div>
-    """, unsafe_allow_html=True
-)
-
-# Select the type of post to generate
-post_type = st.selectbox(
-    "What type of post would you like to create?",
-    (
-        "Template-Based Post Generator",
-        "Data Science Meme/Quote Bot",
-        "Trend Explainer Bot",
-        "Data Science Tip Bot",
-        "Hashtag Suggestion Bot",
-        "Poll/Question Maker",
-        "Project Showcase Post Generator",
-        "Data Visual Explanation Bot"
-    )
-)
-
-# Input for user queries
-user_input = st.text_input("Describe your data science topic or idea:")
-
-# Function to handle Groq API call for different tasks
-def get_groq_response(post_type, user_input):
+📊 **Standardization: The Gaussian Don**
+🔷 What: Centers your data around 0, with a standard deviation of 1.
+🔷 Why: Ideal when your data follows the bell curve or your algorithm is a normal distribution fan.
+🔷 Best for: Logistic regression, SVM, and PCA - the cool kids of the ML world.
+🔷 Superpower: Handling outliers like a boss!
+
+🥊 **The Face-Off**  
+**Normalization:** "Let's all fit in this box!" 📦  
+**Standardization:** "Let's dance around the mean!" 💃
+
+Use Normalization for diverse data like a candy store 🍭, and Standardization when your data fits the normal crowd 🎭.
+
+#DataScience #NormalizationVsStandardization #AITricks
+
+---
+🎯 **Binary Classification or Multiclass? Here's How to Choose!** 🎯
+
+Choosing between **binary** and **multiclass classification** can feel like picking the perfect dessert—do you want the classic **chocolate cake** or an **exotic fruit tart**?
+
+**Binary Classification:**
+🔸 Focuses on **two classes**: cat or dog, pass or fail, true or false.
+🔸 **Simpler** to model, quick to train, and widely used when the task is a straightforward **this or that** decision.
+
+**Multiclass Classification:**
+🔷 Handles **more than two classes**: apple, orange, banana, or even dragon fruit!
+🔷 Complex but powerful when you need **multiple predictions** from a single model, often used in tasks like image recognition.
+
+---
+
+🧠 **How to Ace Hyperparameter Tuning!**
+
+Tuning hyperparameters can feel like adjusting the knobs on an old radio—get it just right and the music plays beautifully. 🎶 
+
+#AI #DataScienceTips #MLHacks
+"""
+
+# Function to get Groq response with few-shot examples
+def get_groq_response_with_few_shot(post_type, user_input):
     if not user_input:
         return "Please enter a valid query."
     
-    query_content = ""
-    
-    # Match the post type with specific tasks
-    if post_type == "Template-Based Post Generator":
-        query_content = f"Generate a social media post about {user_input}."
-    
-    elif post_type == "Data Science Meme/Quote Bot":
-        query_content = f"Generate a data science meme or quote about {user_input}."
-    
-    elif post_type == "Trend Explainer Bot":
-        query_content = f"Explain the trend '{user_input}' in data science in a concise way."
-    
-    elif post_type == "Data Science Tip Bot":
-        query_content = f"Give a useful data science tip about {user_input}."
-    
-    elif post_type == "Hashtag Suggestion Bot":
-        query_content = f"Suggest relevant hashtags for {user_input}."
-    
-    elif post_type == "Poll/Question Maker":
-        query_content = f"Generate a poll or question about {user_input} for social media."
-    
-    elif post_type == "Project Showcase Post Generator":
-        query_content = f"Create a social media post to showcase my data science project on {user_input}."
-    
-    elif post_type == "Data Visual Explanation Bot":
-        query_content = f"Generate a caption or explanation for a data visualization about {user_input}."
+    query_content = f"Here are a few examples of fun and engaging social media posts about data science:\n{few_shot_examples}\nNow generate a {post_type} post about {user_input} in a similar style."
     
     try:
-        # Make API call to Groq LLaMA model
+        # Make the API call with few-shot examples included
         chat_completion = groq_client.chat.completions.create(
             messages=[{"role": "user", "content": query_content}],
             model="llama-3.1-70b-versatile",
@@ -119,17 +76,33 @@ def get_groq_response(post_type, user_input):
     except Exception as e:
         return f"An error occurred: {str(e)}"
 
-# Submit button to generate the post
-response = None
+# Streamlit UI
+st.title("Social Media Post Maker for Data Science")
+
+# User input
+post_type = st.selectbox("Select the type of post:", [
+    "Template-Based Post",
+    "Data Science Meme/Quote",
+    "Trend Explainer",
+    "Data Science Tip",
+    "Hashtag Suggestion",
+    "Poll/Question Maker",
+    "Project Showcase",
+    "Data Visual Explanation"
+])
+
+user_input = st.text_input("Enter the topic or idea for the post:")
+
+# Generate button
 if st.button("Generate Post"):
     if user_input:
         with st.spinner("Generating post..."):
-            response = get_groq_response(post_type, user_input)
+            response = get_groq_response_with_few_shot(post_type, user_input)
             st.success("Post generated successfully!")
     else:
         st.error("Please enter a topic or idea!")
 
-# Allow users to copy the generated post
+# Display the result
 if response:
     st.code(response)
 
@@ -148,6 +121,24 @@ if response:
         alert("Copied to clipboard!");
     }}
     </script>
+    <style>
+    .copy-btn {{
+        background-color: #007bff;
+        color: white;
+        border: none;
+        padding: 10px 20px;
+        text-align: center;
+        text-decoration: none;
+        display: inline-block;
+        font-size: 16px;
+        margin: 4px 2px;
+        cursor: pointer;
+        border-radius: 4px;
+    }}
+    .copy-btn:hover {{
+        background-color: #0056b3;
+    }}
+    </style>
     """
-
-    html(copy_button_code)  # Removed unsafe_allow_html
+    
+    html(copy_button_code)