Update app.py

app.py

@@ -1,23 +1,23 @@
 import os
 import requests
 import gradio as gr
-from openai import OpenAI
+from huggingface_hub import InferenceClient
 
 # =====================
-# 配置
+# 配置 (从环境变量安全读取)
 # =====================
-
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+# 获取 Hugging Face 的 Access Token（在后台 Settings -> Secrets 里配置）
+HF_TOKEN = os.getenv("HF_TOKEN")
 PEXELS_API_KEY = os.getenv("PEXELS_API_KEY")
 
-client = OpenAI(api_key=OPENAI_API_KEY)
+# 初始化 Hugging Face 免费大模型客户端
+# 使用官方推荐且对中文支持极好的 Qwen 2.5 7B 模型，完全运行在 HF 云端，不吃本地内存
+client = InferenceClient("Qwen/Qwen-2.5-7B-Instruct", token=HF_TOKEN)
 
 # =====================
-# AI文章生成
+# AI文章生成（HF 免费版）
 # =====================
-
 def generate_article(keyword):
-
     prompt = f"""
 请根据关键词《{keyword}》写一篇适合今日头条发布的文章。
 
@@ -27,40 +27,34 @@ def generate_article(keyword):
 3. 分段清晰
 4. 结尾总结
 """
-
     try:
+        # 使用标准的 chat.completions 接口，完全免费
         response = client.chat.completions.create(
-            model="gpt-4.1-mini",
+            model="Qwen/Qwen-2.5-7B-Instruct",
             messages=[
                 {
                     "role": "user",
                     "content": prompt
                 }
             ],
-            temperature=0.8
+            temperature=0.8,
+            max_tokens=1500
         )
-
         return response.choices[0].message.content
-
     except Exception as e:
-        return f"文章生成失败：{str(e)}"
-
+        return f"文章生成失败，原因：{str(e)}\n\n💡 提示：如果提示无权限，请检查是否在 Space 的 Settings -> Secrets 中正确配置了 HF_TOKEN。"
 
 # =====================
-# 搜图
+# 搜图 (Pexels API)
 # =====================
-
 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
@@ -73,86 +67,87 @@ def search_images(keyword):
             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):
-
+    # 用标准 Markdown 语法包装超链接
     links = [
-        f"https://www.youtube.com/results?search_query={keyword}"
+        f"[在 YouTube 上搜索 《{keyword}》](https://www.youtube.com/results?search_query={keyword})"
     ]
-
-    return "\n".join(
-        [f"- {link}" for link in links]
-    )
-
+    
+    # 采用纯文本符号 !!! 代替复杂 emoji，防止系统字体缺失导致乱码
+    result = "### !!! 相关视频搜索链接\n"
+    result += "\n".join([f"- {link}" for link in links])
+    return result
 
 # =====================
 # 主函数
 # =====================
-
 def main(keyword):
-
+    if not keyword.strip():
+        return "请输入有效的关键词！", [], "### !!! 请输入关键词"
+        
     article = generate_article(keyword)
-
     images = search_images(keyword)
-
     videos = search_videos(keyword)
-
     return article, images, videos
 
-
 # =====================
-# Gradio界面
+# Gradio界面布局
 # =====================
+with gr.Blocks(title="头条 AI Agent") as demo:
 
-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="相关视频"
-    )
+    gr.Markdown("# 头条 AI Agent (HF 免费版)")
+    gr.Markdown("输入关键词，依托 Hugging Face 免费云端算力一键生成爆款文章、匹配图片和视频资源。")
 
+    # 布局：上方输入栏
+    with gr.Row():
+        keyword_input = gr.Textbox(
+            label="关键词",
+            placeholder="例如：人工智能新趋势",
+            scale=4
+        )
+        btn = gr.Button("立即生成", variant="primary", scale=1)
+
+    # 布局：下方左右分栏
+    with gr.Row():
+        # 左栏：文章输出
+        with gr.Column(scale=2):
+            article_output = gr.Textbox(
+                label="生成文章",
+                lines=22,
+                show_copy_button=True  # 开启一键复制功能，方便搬运
+            )
+        
+        # 右栏：素材匹配
+        with gr.Column(scale=1):
+            gallery_output = gr.Gallery(
+                label="匹配素材（Pexels 免费商用图）",
+                columns=2,
+                rows=3,
+                height=350
+            )
+            
+            # 视频链接包裹在一个基础小框里，保持视觉整洁
+            with gr.Box():
+                video_output = gr.Markdown(
+                    label="素材视频链接"
+                )
+
+    # 绑定点击事件
     btn.click(
         fn=main,
-        inputs=keyword,
+        inputs=keyword_input,
         outputs=[
             article_output,
             gallery_output,