Update app.py

app.py

@@ -1,14 +1,163 @@
+import os
+import requests
 import gradio as gr
+from openai import OpenAI
 
-def generate(keyword):
-    return f"你输入的关键词是：{keyword}"
+# =====================
+# 配置
+# =====================
 
-demo = gr.Interface(
-    fn=generate,
-    inputs="text",
-    outputs="text",
-    title="头条AI Agent",
-    description="输入关键词，自动生成头条文章"
-)
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+PEXELS_API_KEY = os.getenv("PEXELS_API_KEY")
+
+client = OpenAI(api_key=OPENAI_API_KEY)
+
+# =====================
+# AI文章生成
+# =====================
+
+def generate_article(keyword):
+
+    prompt = f"""
+请根据关键词《{keyword}》写一篇适合今日头条发布的文章。
+
+要求：
+1. 标题吸引人
+2. 正文800字左右
+3. 分段清晰
+4. 结尾总结
+"""
+
+    try:
+        response = client.chat.completions.create(
+            model="gpt-4.1-mini",
+            messages=[
+                {
+                    "role": "user",
+                    "content": prompt
+                }
+            ],
+            temperature=0.8
+        )
+
+        return response.choices[0].message.content
+
+    except Exception as e:
+        return f"文章生成失败：{str(e)}"
+
+
+# =====================
+# 搜图
+# =====================
+
+def search_images(keyword):
+
+    if not PEXELS_API_KEY:
+        return []
+
+    url = "https://api.pexels.com/v1/search"
+
+    headers = {
+        "Authorization": PEXELS_API_KEY
+    }
+
+    params = {
+        "query": keyword,
+        "per_page": 6
+    }
+
+    try:
+        res = requests.get(
+            url,
+            headers=headers,
+            params=params,
+            timeout=20
+        )
+
+        data = res.json()
+
+        images = []
+
+        for photo in data.get("photos", []):
+            images.append(photo["src"]["medium"])
+
+        return images
+
+    except:
+        return []
+
+
+# =====================
+# 视频搜索（示例）
+# =====================
+
+def search_videos(keyword):
+
+    links = [
+        f"https://www.youtube.com/results?search_query={keyword}"
+    ]
+
+    return "\n".join(
+        [f"- {link}" for link in links]
+    )
+
+
+# =====================
+# 主函数
+# =====================
+
+def main(keyword):
+
+    article = generate_article(keyword)
+
+    images = search_images(keyword)
+
+    videos = search_videos(keyword)
+
+    return article, images, videos
+
+
+# =====================
+# Gradio界面
+# =====================
+
+with gr.Blocks() as demo:
+
+    gr.Markdown("# 头条 AI Agent")
+
+    gr.Markdown(
+        "输入关键词，自动生成文章、搜索图片和视频"
+    )
+
+    keyword = gr.Textbox(
+        label="关键词",
+        placeholder="例如：人工智能"
+    )
+
+    btn = gr.Button("生成")
+
+    article_output = gr.Textbox(
+        label="生成文章",
+        lines=20
+    )
+
+    gallery_output = gr.Gallery(
+        label="相关图片",
+        columns=3
+    )
+
+    video_output = gr.Markdown(
+        label="相关视频"
+    )
+
+    btn.click(
+        fn=main,
+        inputs=keyword,
+        outputs=[
+            article_output,
+            gallery_output,
+            video_output
+        ]
+    )
 
 demo.launch()