Upload folder using huggingface_hub

app.py

@@ -1,54 +1,69 @@
-import gradio as gr
+import os
 
+import gradio as gr
+from huggingface_hub import InferenceClient
 
-def _qwen(system: str, user: str) -> str:
-    return (
-        f"<|im_start|>system\n{system}<|im_end|>\n"
-        f"<|im_start|>user\n{user}<|im_end|>\n"
-        "<|im_start|>assistant\n"
-    )
+TEXT_MODEL = "Qwen/Qwen2.5-72B-Instruct"
 
 
-def scene_prompt_input(moment: str) -> str:
-    return _qwen(
-        "You write precise FLUX image generation prompts. Output ONE sentence only, "
-        "starting with 'World Cup 2026 stadium,'",
-        f"Write an image generation prompt for this match moment: {moment}",
+def _chat(client: InferenceClient, system: str, user: str, max_tokens: int = 300) -> str:
+    resp = client.chat.completions.create(
+        model=TEXT_MODEL,
+        messages=[
+            {"role": "system", "content": system},
+            {"role": "user", "content": user},
+        ],
+        max_tokens=max_tokens,
     )
+    return (resp.choices[0].message.content or "").strip()
+
 
+def create_content(
+    moment: str, oauth_token: gr.OAuthToken | None = None
+) -> tuple[str, str, str, str]:
+    token = (oauth_token.token if oauth_token else None) or os.environ.get("HF_TOKEN")
+    client = InferenceClient(token=token)
 
-def tweet_input(moment: str) -> str:
-    return _qwen(
+    tweet = _chat(
+        client,
         "You write punchy viral sports tweets. Max 240 chars. 2-3 emojis. "
-        "End with #WorldCup2026. Output only the tweet text, nothing else.",
-        f"Write a tweet about this World Cup moment: {moment}",
+        "End with #WorldCup2026. Output ONLY the tweet, no commentary.",
+        f"World Cup moment: {moment}",
+        max_tokens=120,
     )
 
-
-def instagram_input(moment: str) -> str:
-    return _qwen(
-        "You write engaging Instagram captions for sports moments. "
-        "3-5 sentences, storytelling tone, end with 10 relevant hashtags.",
-        f"Write an Instagram caption about this World Cup moment: {moment}",
+    instagram = _chat(
+        client,
+        "You write high-engagement Instagram sports captions. "
+        "Open with a punchy hook. 3-4 vivid sentences building the story. "
+        "End with 12-15 hashtags on a new line.",
+        f"World Cup moment: {moment}",
+        max_tokens=350,
     )
 
+    linkedin = _chat(
+        client,
+        "You write professional LinkedIn posts that use a sports moment as a lens for "
+        "leadership or teamwork. Bold opening statement. 150-200 words. No hashtags. "
+        "End with a question to drive comments.",
+        f"World Cup moment: {moment}",
+        max_tokens=450,
+    )
 
-def linkedin_input(moment: str) -> str:
-    return _qwen(
-        "You write professional LinkedIn posts about sports as a lens for leadership "
-        "and teamwork. 150-200 words, compelling hook.",
-        f"Write a LinkedIn post inspired by this World Cup moment: {moment}",
+    scene_prompt = _chat(
+        client,
+        "Write a single FLUX image editing prompt starting with "
+        "'Transform into a cinematic World Cup 2026 poster,' — one sentence only.",
+        f"Match moment: {moment}",
+        max_tokens=60,
     )
 
+    return tweet, instagram, linkedin, scene_prompt
+
 
 demo = gr.Workflow(
     graph="workflow.json",
-    bind={
-        "Scene Prompt Input": scene_prompt_input,
-        "Tweet Input": tweet_input,
-        "Instagram Input": instagram_input,
-        "LinkedIn Input": linkedin_input,
-    },
+    bind={"Content Creator": create_content},
 )
 
 if __name__ == "__main__":

workflow.json

@@ -21,33 +21,28 @@
       "asset_type": "text",
       "inputs":  [{ "id": "in",  "label": "Text", "type": "text" }],
       "outputs": [{ "id": "out", "label": "Text", "type": "text" }],
-      "x": 80, "y": 300, "width": 220, "height": 161,
+      "x": 80, "y": 280, "width": 220, "height": 200,
       "data": { "in": "USA score in 94th minute to beat Mexico 2-1 in the group stage, LA crowd erupts" }
     }
   ],
 
   "operators": [
     {
-      "id": "op_scene_fn",
-      "label": "Scene Prompt Input",
+      "id": "op_content",
+      "label": "Content Creator",
       "role": "operator",
       "kind": "fn",
-      "fn": "Scene Prompt Input",
-      "inputs":  [{ "id": "in_moment", "label": "moment", "type": "text", "required": true }],
-      "outputs": [{ "id": "out_0", "label": "output", "type": "text", "output_index": 0 }],
-      "x": 380, "y": 80, "width": 220, "height": 121,
-      "data": {}
-    },
-    {
-      "id": "op_scene_model",
-      "label": "Generate Scene Prompt",
-      "role": "operator",
-      "kind": "model",
-      "model_id": "Qwen/Qwen2.5-72B-Instruct",
-      "pipeline_tag": "text-generation",
-      "inputs":  [{ "id": "in_0", "label": "inputs", "type": "text", "required": true }],
-      "outputs": [{ "id": "out_0", "label": "output", "type": "text", "output_index": 0 }],
-      "x": 650, "y": 80, "width": 220, "height": 121,
+      "fn": "Content Creator",
+      "inputs": [
+        { "id": "in_moment", "label": "moment", "type": "text", "required": true }
+      ],
+      "outputs": [
+        { "id": "out_tweet",    "label": "tweet",         "type": "text", "output_index": 0 },
+        { "id": "out_ig",       "label": "instagram",     "type": "text", "output_index": 1 },
+        { "id": "out_linkedin", "label": "linkedin",      "type": "text", "output_index": 2 },
+        { "id": "out_scene",    "label": "scene_prompt",  "type": "text", "output_index": 3 }
+      ],
+      "x": 380, "y": 280, "width": 240, "height": 200,
       "data": {}
     },
     {
@@ -59,81 +54,12 @@
       "endpoint": "/infer",
       "inputs": [
         { "id": "in_image",  "label": "input_image", "type": "image", "required": true },
-        { "id": "in_prompt", "label": "prompt",       "type": "text",  "required": true }
+        { "id": "in_prompt", "label": "prompt",      "type": "text",  "required": true }
       ],
       "outputs": [
         { "id": "out_0", "label": "Result", "type": "image", "output_index": 0 }
       ],
-      "x": 920, "y": 80, "width": 220, "height": 121,
-      "data": {}
-    },
-    {
-      "id": "op_tweet_fn",
-      "label": "Tweet Input",
-      "role": "operator",
-      "kind": "fn",
-      "fn": "Tweet Input",
-      "inputs":  [{ "id": "in_moment", "label": "moment", "type": "text", "required": true }],
-      "outputs": [{ "id": "out_0", "label": "output", "type": "text", "output_index": 0 }],
-      "x": 380, "y": 260, "width": 220, "height": 121,
-      "data": {}
-    },
-    {
-      "id": "op_tweet_model",
-      "label": "Generate Tweet",
-      "role": "operator",
-      "kind": "model",
-      "model_id": "Qwen/Qwen2.5-72B-Instruct",
-      "pipeline_tag": "text-generation",
-      "inputs":  [{ "id": "in_0", "label": "inputs", "type": "text", "required": true }],
-      "outputs": [{ "id": "out_0", "label": "output", "type": "text", "output_index": 0 }],
-      "x": 650, "y": 260, "width": 220, "height": 121,
-      "data": {}
-    },
-    {
-      "id": "op_ig_fn",
-      "label": "Instagram Input",
-      "role": "operator",
-      "kind": "fn",
-      "fn": "Instagram Input",
-      "inputs":  [{ "id": "in_moment", "label": "moment", "type": "text", "required": true }],
-      "outputs": [{ "id": "out_0", "label": "output", "type": "text", "output_index": 0 }],
-      "x": 380, "y": 420, "width": 220, "height": 121,
-      "data": {}
-    },
-    {
-      "id": "op_ig_model",
-      "label": "Generate Instagram Caption",
-      "role": "operator",
-      "kind": "model",
-      "model_id": "Qwen/Qwen2.5-72B-Instruct",
-      "pipeline_tag": "text-generation",
-      "inputs":  [{ "id": "in_0", "label": "inputs", "type": "text", "required": true }],
-      "outputs": [{ "id": "out_0", "label": "output", "type": "text", "output_index": 0 }],
-      "x": 650, "y": 420, "width": 220, "height": 121,
-      "data": {}
-    },
-    {
-      "id": "op_linkedin_fn",
-      "label": "LinkedIn Input",
-      "role": "operator",
-      "kind": "fn",
-      "fn": "LinkedIn Input",
-      "inputs":  [{ "id": "in_moment", "label": "moment", "type": "text", "required": true }],
-      "outputs": [{ "id": "out_0", "label": "output", "type": "text", "output_index": 0 }],
-      "x": 380, "y": 580, "width": 220, "height": 121,
-      "data": {}
-    },
-    {
-      "id": "op_linkedin_model",
-      "label": "Generate LinkedIn Post",
-      "role": "operator",
-      "kind": "model",
-      "model_id": "Qwen/Qwen2.5-72B-Instruct",
-      "pipeline_tag": "text-generation",
-      "inputs":  [{ "id": "in_0", "label": "inputs", "type": "text", "required": true }],
-      "outputs": [{ "id": "out_0", "label": "output", "type": "text", "output_index": 0 }],
-      "x": 650, "y": 580, "width": 220, "height": 121,
+      "x": 700, "y": 80, "width": 220, "height": 121,
       "data": {}
     }
   ],
@@ -144,9 +70,9 @@
       "label": "Match Scene",
       "role": "subject",
       "asset_type": "image",
-      "inputs":  [{ "id": "in",  "label": "Image", "type": "image" }],
+      "inputs":  [{ "id": "in", "label": "Image", "type": "image" }],
       "outputs": [{ "id": "out", "label": "Image", "type": "image" }],
-      "x": 1190, "y": 80, "width": 220, "height": 107,
+      "x": 980, "y": 80, "width": 220, "height": 107,
       "data": {}
     },
     {
@@ -154,9 +80,9 @@
       "label": "Tweet",
       "role": "subject",
       "asset_type": "text",
-      "inputs":  [{ "id": "in",  "label": "Text", "type": "text" }],
+      "inputs":  [{ "id": "in", "label": "Text", "type": "text" }],
       "outputs": [{ "id": "out", "label": "Text", "type": "text" }],
-      "x": 920, "y": 260, "width": 220, "height": 107,
+      "x": 700, "y": 260, "width": 220, "height": 107,
       "data": {}
     },
     {
@@ -164,9 +90,9 @@
       "label": "Instagram Caption",
       "role": "subject",
       "asset_type": "text",
-      "inputs":  [{ "id": "in",  "label": "Text", "type": "text" }],
+      "inputs":  [{ "id": "in", "label": "Text", "type": "text" }],
       "outputs": [{ "id": "out", "label": "Text", "type": "text" }],
-      "x": 920, "y": 420, "width": 220, "height": 107,
+      "x": 700, "y": 400, "width": 220, "height": 107,
       "data": {}
     },
     {
@@ -174,28 +100,21 @@
       "label": "LinkedIn Post",
       "role": "subject",
       "asset_type": "text",
-      "inputs":  [{ "id": "in",  "label": "Text", "type": "text" }],
+      "inputs":  [{ "id": "in", "label": "Text", "type": "text" }],
       "outputs": [{ "id": "out", "label": "Text", "type": "text" }],
-      "x": 920, "y": 580, "width": 220, "height": 107,
+      "x": 700, "y": 540, "width": 220, "height": 107,
       "data": {}
     }
   ],
 
   "edges": [
-    { "id": "e1",  "from_node_id": "ref_moment",       "from_port_id": "out",    "to_node_id": "op_scene_fn",      "to_port_id": "in_moment", "type": "text"  },
-    { "id": "e2",  "from_node_id": "op_scene_fn",      "from_port_id": "out_0",  "to_node_id": "op_scene_model",   "to_port_id": "in_0",      "type": "text"  },
-    { "id": "e3",  "from_node_id": "op_scene_model",   "from_port_id": "out_0",  "to_node_id": "op_flux",          "to_port_id": "in_prompt", "type": "text"  },
-    { "id": "e4",  "from_node_id": "ref_image",        "from_port_id": "out",    "to_node_id": "op_flux",          "to_port_id": "in_image",  "type": "image" },
-    { "id": "e5",  "from_node_id": "op_flux",          "from_port_id": "out_0",  "to_node_id": "sub_image",        "to_port_id": "in",        "type": "image" },
-    { "id": "e6",  "from_node_id": "ref_moment",       "from_port_id": "out",    "to_node_id": "op_tweet_fn",      "to_port_id": "in_moment", "type": "text"  },
-    { "id": "e7",  "from_node_id": "op_tweet_fn",      "from_port_id": "out_0",  "to_node_id": "op_tweet_model",   "to_port_id": "in_0",      "type": "text"  },
-    { "id": "e8",  "from_node_id": "op_tweet_model",   "from_port_id": "out_0",  "to_node_id": "sub_tweet",        "to_port_id": "in",        "type": "text"  },
-    { "id": "e9",  "from_node_id": "ref_moment",       "from_port_id": "out",    "to_node_id": "op_ig_fn",         "to_port_id": "in_moment", "type": "text"  },
-    { "id": "e10", "from_node_id": "op_ig_fn",         "from_port_id": "out_0",  "to_node_id": "op_ig_model",      "to_port_id": "in_0",      "type": "text"  },
-    { "id": "e11", "from_node_id": "op_ig_model",      "from_port_id": "out_0",  "to_node_id": "sub_instagram",    "to_port_id": "in",        "type": "text"  },
-    { "id": "e12", "from_node_id": "ref_moment",       "from_port_id": "out",    "to_node_id": "op_linkedin_fn",   "to_port_id": "in_moment", "type": "text"  },
-    { "id": "e13", "from_node_id": "op_linkedin_fn",   "from_port_id": "out_0",  "to_node_id": "op_linkedin_model","to_port_id": "in_0",      "type": "text"  },
-    { "id": "e14", "from_node_id": "op_linkedin_model","from_port_id": "out_0",  "to_node_id": "sub_linkedin",     "to_port_id": "in",        "type": "text"  }
+    { "id": "e1", "from_node_id": "ref_moment",  "from_port_id": "out",        "to_node_id": "op_content", "to_port_id": "in_moment",  "type": "text"  },
+    { "id": "e2", "from_node_id": "ref_image",   "from_port_id": "out",        "to_node_id": "op_flux",    "to_port_id": "in_image",   "type": "image" },
+    { "id": "e3", "from_node_id": "op_content",  "from_port_id": "out_scene",  "to_node_id": "op_flux",    "to_port_id": "in_prompt",  "type": "text"  },
+    { "id": "e4", "from_node_id": "op_flux",     "from_port_id": "out_0",      "to_node_id": "sub_image",  "to_port_id": "in",         "type": "image" },
+    { "id": "e5", "from_node_id": "op_content",  "from_port_id": "out_tweet",  "to_node_id": "sub_tweet",  "to_port_id": "in",         "type": "text"  },
+    { "id": "e6", "from_node_id": "op_content",  "from_port_id": "out_ig",     "to_node_id": "sub_instagram","to_port_id": "in",        "type": "text"  },
+    { "id": "e7", "from_node_id": "op_content",  "from_port_id": "out_linkedin","to_node_id": "sub_linkedin","to_port_id": "in",        "type": "text"  }
   ],
 
   "view": { "default": "canvas" }