Update app.py

app.py

@@ -1,13 +1,44 @@
 import gradio as gr
+from transformers import pipeline
 
-def greet(name):
-    return f"Hello {name}, welcome to my first Space."
-
-demo = gr.Interface(
-    fn=greet,
-    inputs="text",
-    outputs="text",
-    title="My First Hugging Face Space"
+# 使用一个轻量文本生成模型（CPU 可跑）
+generator = pipeline(
+    "text-generation",
+    model="distilgpt2"
 )
 
+def rewrite(text, style):
+    if style == "More formal":
+        prompt = f"Rewrite this sentence in a more formal tone:\n{text}"
+    elif style == "More friendly":
+        prompt = f"Rewrite this sentence in a friendly tone:\n{text}"
+    else:
+        prompt = f"Rewrite this sentence to be shorter:\n{text}"
+
+    result = generator(prompt, max_length=100, num_return_sequences=1)
+    return result[0]["generated_text"]
+
+with gr.Blocks(title="AI Text Rewriter") as demo:
+    gr.Markdown(
+        "### ✨ AI Text Rewriter\n"
+        "Paste a sentence and let AI rewrite it for you."
+    )
+
+    text_input = gr.Textbox(
+        label="Your text",
+        lines=3,
+        placeholder="Type something here..."
+    )
+
+    style = gr.Radio(
+        ["More formal", "More friendly", "Shorter"],
+        label="Rewrite style",
+        value="More friendly"
+    )
+
+    output = gr.Textbox(label="AI Result", lines=4)
+
+    btn = gr.Button("Rewrite with AI")
+    btn.click(fn=rewrite, inputs=[text_input, style], outputs=output)
+
 demo.launch()