Update app.py

app.py

@@ -1,32 +1,76 @@
+# app.py
+import os, requests
 import gradio as gr
-from transformers import pipeline
-
-# 🔹 Polski model GPT-2 (mały, działa na CPU w Hugging Face Spaces)
-generator = pipeline("text-generation", model="flax-community/polish-gpt2-small")
-
-def generate_post(prompt, platform):
-    style = ""
-    if platform == "TikTok":
-        style = "Napisz krótki, dynamiczny opis z emoji i hashtagiem."
-    elif platform == "Instagram":
-        style = "Napisz inspirujący post z emoji i hashtagiem."
-    elif platform == "LinkedIn":
-        style = "Napisz profesjonalny, motywacyjny post."
-    elif platform == "Twitter":
-        style = "Napisz krótki tweet z emoji i hashtagiem."
-
-    input_text = f"{style}\nTemat: {prompt}\nPost:"
-    result = generator(input_text, max_length=120, num_return_sequences=1, do_sample=True, temperature=0.9)
-    return result[0]["generated_text"]
-
-with gr.Blocks() as demo:
-    gr.Markdown("## 🤖 Social Media Generator (Polski) – Twój darmowy agent AI")
-    with gr.Row():
-        prompt = gr.Textbox(label="Podaj temat posta", placeholder="np. motywacja do działania")
-        platform = gr.Dropdown(["TikTok", "Instagram", "LinkedIn", "Twitter"], label="Platforma")
-    output = gr.Textbox(label="Wygenerowany post")
-
-    btn = gr.Button("Generuj post")
-    btn.click(generate_post, inputs=[prompt, platform], outputs=output)
-
-demo.launch()
+
+# === CONFIG ===
+HF_TOKEN = os.environ.get("HF_TOKEN") # ustaw w Hugging Face Spaces → Settings → Secrets
+MODEL = "google/flan-t5-base" # lekki i darmowy model do generacji tekstów
+
+headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+
+# ===== FUNKCJA DO WYWOŁYWANIA MODELU =====
+def call_model(prompt):
+    url = f"https://api-inference.huggingface.co/models/{MODEL}"
+    payload = {
+        "inputs": prompt,
+        "parameters": {
+            "max_new_tokens": 400,
+            "temperature": 0.9, # bardziej kreatywne treści
+            "top_p": 0.95
+        }
+    }
+    r = requests.post(url, headers=headers, json=payload, timeout=60)
+    data = r.json()
+
+    if isinstance(data, dict) and data.get("error"):
+        return "⚠️ Błąd modelu: " + data.get("error")
+    if isinstance(data, list) and data and isinstance(data[0], dict):
+        return data[0].get("generated_text", str(data))
+    if isinstance(data, dict):
+        return data.get("generated_text", str(data))
+    return str(data)
+
+# ===== FUNKCJA GŁÓWNA – GENERACJA POSTÓW =====
+def generate_posts(temat, ton, platforma):
+    prompt = f"""
+Jesteś ekspertem od pisania angażujących postów na social media w języku polskim.
+
+Temat: {temat}
+Platforma: {platforma}
+Ton: {ton}
+
+Twoje zadanie:
+Wygeneruj 3 różne warianty posta. Każdy wariant podaj w tym formacie:
+
+---
+Wariant X:
+1) Hook: (1 krótkie, chwytliwe zdanie)
+2) Treść: (dostosowana do platformy – 
+   • TikTok/Instagram: max 3 zdania, dynamiczne, z emoji
+   • Facebook: 3–5 zdań, angażujące, lekkie
+   • LinkedIn: 6–8 zdań, profesjonalne, z wartością merytoryczną)
+3) Hashtagi: (10 propozycji, oddzielone przecinkami, bez polskich znaków)
+4) CTA: (1 krótka linia zachęcająca do działania)
+5) Sugestia wizualna: (jedno zdanie, opis pomysłu na zdjęcie/wideo)
+---
+
+Pisz konkretnie, naturalnym językiem, bez powtarzania instrukcji.
+"""
+    out = call_model(prompt)
+    return out
+
+# ===== UI =====
+iface = gr.Interface(
+    fn=generate_posts,
+    inputs=[
+        gr.Textbox(label="🎯 Temat (np. Jak wstać o 5 rano)"),
+        gr.Dropdown(["motywacyjny","informacyjny","sprzedażowy","edukacyjny"], label="🎙 Ton"),
+        gr.Dropdown(["TikTok","Instagram","Facebook","LinkedIn"], label="📱 Platforma")
+    ],
+    outputs=gr.Textbox(label="✨ Wygenerowane posty", lines=25),
+    title="🤖 Social Media Post Generator",
+    description="Podaj temat, wybierz ton i platformę. Agent wygeneruje 3 gotowe propozycje postów (hook, treść, hashtagi, CTA i sugestia wizualna)."
+)
+
+if __name__ == "__main__":
+    iface.launch()