Update app.py

app.py

@@ -0,0 +1,93 @@
+import gradio as gr
+import os
+from langchain_core.prompts import ChatPromptTemplate
+from langchain_groq import ChatGroq
+
+
+
+# ---------------------------
+# LangChain Setup
+# ---------------------------
+
+llm = ChatGroq(
+    temperature=0,
+    groq_api_key=os.getenv('groq_api_key'),
+    model_name="llama-3.3-70b-versatile"
+)
+
+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. 
+
+        User text : {text}
+
+        Output the result in the following format:
+
+        [
+            "niche": Fitness
+        ]
+    """
+
+    prompt = ChatPromptTemplate.from_messages([("system", system_prompt)])
+    chain = prompt | llm
+    output = chain.invoke({"text": text})
+
+    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
+
+        User text : {text}
+
+        Output the result in the following format:
+
+        ["content_styles": Educational]
+    """
+
+    prompt = ChatPromptTemplate.from_messages([("system", system_prompt)])
+    chain = prompt | llm
+    output = chain.invoke({"text": text})
+
+    try:
+        return output.content.split('"content_styles": ')[1].split(']')[0].strip()
+    except Exception:
+        return "Could not determine content style"
+
+# ---------------------------
+# Gradio Interface
+# ---------------------------
+
+def analyze_post(text):
+    niche = niche_detection(text)
+    style = content_styles(text)
+    return {
+        "Niche": niche,
+        "Content Style": style
+    }
+
+iface = gr.Interface(
+    fn=analyze_post,
+    inputs=gr.Textbox(lines=5, label="Instagram Post Text"),
+    outputs=[
+        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."
+)
+
+if __name__ == "__main__":
+    iface.launch(share=True)