Update app.py

app.py

@@ -17,8 +17,32 @@ llm = ChatGroq(
 
 def niche_detection(text):
     system_prompt = """
-        You are a social media niche expert. 
-        Analyze the following Instagram post captions and hashtags to determine the niches this content belongs to. 
+        You are a highly skilled social media strategist and niche classification expert. Your task is to analyze the following Instagram caption and hashtags to identify the **main niche** the content belongs to.
+
+        Examples of possible niches include:
+        - Fitness
+        - Nutrition  
+        - Mental Health
+        - Personal Finance
+        - Entrepreneurship
+        - Fashion
+        - Tech/Gadgets
+        - Parenting
+        - Travel
+        - Beauty/Skincare
+        - Relationships
+        - Digital Marketing
+        - Motivation
+        - Self-improvement
+        - Gaming
+        - Food/Cooking
+        - Home Decor
+        - Photography
+        - Music
+        - Art/Design
+
+        Analyze the content carefully and respond with ONLY the niche name (e.g., "Fitness" or "Travel"). 
+        ** Don't retunr - " Could not determine niche "
 
         User text : {text}
 
@@ -32,23 +56,29 @@ def niche_detection(text):
     prompt = ChatPromptTemplate.from_messages([("system", system_prompt)])
     chain = prompt | llm
     output = chain.invoke({"text": text})
+    return output.content
+
+
+        
 
-    try:
-        return output.content.split('"niche": "')[1].split('"')[0]
-    except Exception:
-        return "Could not determine niche"
 
 def content_styles(text):
     system_prompt = """
-        You are an expert in analyzing social media content styles. Review the following Instagram captions and hashtags and classify the dominant content style used in each post. Choose from the following styles:
-            - Educational
-            - Motivational
-            - Meme/Relatable
-            - Personal Story
-            - Tips/Hacks
-            - Promotional
-            - Inspirational
-            - Opinion/Rant
+        You are an expert in analyzing social media content styles. Review the following Instagram caption and hashtags and classify the dominant content style.
+
+        Choose from these content styles:
+        - Educational (teaching something new, how-to content)
+        - Motivational (encouraging action, inspiring change)
+        - Meme/Relatable (funny, relatable content)
+        - Personal Story (sharing personal experiences)
+        - Tips/Hacks (quick tips, life hacks)
+        - Promotional (selling products/services)
+        - Inspirational (uplifting quotes, positive messages)
+        - Opinion/Rant (personal opinions, controversial takes)
+        - Behind-the-Scenes (showing process, daily life)
+        - Question/Poll (asking for engagement)
+
+        Analyze the content and respond with ONLY the content style name (e.g., "Educational" or "Motivational").
 
         User text : {text}
 
@@ -60,11 +90,8 @@ def content_styles(text):
     prompt = ChatPromptTemplate.from_messages([("system", system_prompt)])
     chain = prompt | llm
     output = chain.invoke({"text": text})
+    return output.content
 
-    try:
-        return output.content.split('"content_styles": ')[1].split(']')[0].strip()
-    except Exception:
-        return "Could not determine content style"
 
 # ---------------------------
 # Gradio Interface
@@ -75,6 +102,7 @@ def analyze_post(text):
     style = content_styles(text)
     return niche, style  
 
+
 iface = gr.Interface(
     fn=analyze_post,
     inputs=gr.Textbox(lines=5, label="Instagram Post Text"),
@@ -82,7 +110,6 @@ iface = gr.Interface(
         gr.Textbox(label="Detected Niche"),
         gr.Textbox(label="Detected Content Style")
     ],
-    title="Instagram Post Analyzer",
     description="Enter an Instagram post (caption + hashtags) to detect its niche and content style."
 )