Update app.py

app.py

@@ -3,14 +3,16 @@ from streamlit.components.v1 import html
 from groq import Groq
 import os
 
-# Use the Hugging Face secret API key
-api_key = os.getenv('api_key')
+# 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 the Hugging Face secret key
-client = Groq(api_key=api_key)
+# Initialize Groq client with API key
+groq_client = Groq(api_key=groq_api_key)
 
 # Set the title of the app
-st.title("AI Chatbot with Groq LLaMA Model")
+st.title("Social Media Post Maker for Data Science")
 
 # Custom CSS for a professional look
 css = """
@@ -40,37 +42,83 @@ html(css)
 st.markdown(
     """
     <div class='custom-box'>
-    <h2>Ask anything about Data Science</h2>
-    <p>Powered by Groq LLaMA 3.1 Model 🚀</p>
+    <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("Ask your question here:")
+user_input = st.text_input("Describe your data science topic or idea:")
 
-# Function to handle Groq API call
-def get_groq_response(query):
+# Function to handle Groq API call for different tasks
+def get_groq_response(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}."
+    
     try:
-        chat_completion = client.chat.completions.create(
-            messages=[{"role": "user", "content": query}],
+        # Make API call to Groq LLaMA model
+        chat_completion = groq_client.chat.completions.create(
+            messages=[{"role": "user", "content": query_content}],
             model="llama-3.1-70b-versatile",
         )
         return chat_completion.choices[0].message.content
     except Exception as e:
         return f"An error occurred: {str(e)}"
 
-# Submit button for the chatbot
+# Submit button to generate the post
 response = None
-if st.button("Submit"):
+if st.button("Generate Post"):
     if user_input:
-        with st.spinner("Fetching response..."):
-            response = get_groq_response(user_input)
+        with st.spinner("Generating post..."):
+            response = get_groq_response(post_type, user_input)
             st.success(response)
     else:
-        st.error("Please enter a question!")
+        st.error("Please enter a topic or idea!")
 
-# Copy Button for User to easily copy the result
+# Allow users to copy the generated post
 if response:
     st.code(response)
     st.button("Copy to clipboard", on_click=st.experimental_rerun)