Update app.py

app.py

@@ -1,76 +1,32 @@
-# app.py
-import os, requests
 import gradio as gr
-
-# === 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()
+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()