Update app.py

app.py

@@ -3,200 +3,111 @@ import requests
 from PIL import Image
 import io
 import os
-import time
+import random
 
-# 使用环境变量获取 Token，确保安全
+# 使用环境变量获取 Token
 HF_TOKEN = os.environ.get('HF_TOKEN')
 API_URL = "https://api-inference.huggingface.co/models/rupeshs/LCM-runwayml-stable-diffusion-v1-5"
 headers = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
 
-def query_huggingface(payload):
-    """调用Hugging Face API生成图像"""
-    if not HF_TOKEN:
-        return None
+def generate_image(prompt, style):
+    """专注于图像生成，简化界面"""
     
-    try:
-        response = requests.post(API_URL, headers=headers, json=payload)
-        response.raise_for_status()
-        return response.content
-    except Exception as e:
-        print(f"API调用失败: {e}")
-        return None
-
-def analyze_copyright_risk(prompt, style_option):
-    """分析版权风险"""
-    risky_artists = ["梵高", "毕加索", "莫奈", "达利", "达芬奇", "当代艺术家", "现代艺术家"]
-    high_risk_indicators = []
-    medium_risk_indicators = []
-    
-    # 检测高风险词汇
-    for artist in risky_artists:
-        if artist in prompt:
-            high_risk_indicators.append(f"涉及特定艺术家: {artist}")
-    
-    # 检测风格模仿
-    if "风格" in prompt and style_option != "原创风格":
-        medium_risk_indicators.append("风格模仿可能涉及版权问题")
-    
-    # 生成风险报告
-    if high_risk_indicators:
-        risk_level = "🔴 高风险"
-        suggestions = [
-            "建议使用更通用的风格描述",
-            "避免直接提及在世艺术家",
-            "考虑使用已进入公共领域的古典大师风格"
-        ]
-    elif medium_risk_indicators:
-        risk_level = "🟡 中风险"
-        suggestions = [
-            "建议注明'风格参考'",
-            "避免商业用途",
-            "考虑原创风格创作"
-        ]
-    else:
-        risk_level = "🟢 低风险"
-        suggestions = [
-            "原创性较高，适合个人使用",
-            "可考虑添加创作说明"
-        ]
-    
-    return risk_level, high_risk_indicators + medium_risk_indicators, suggestions
-
-def generate_art_with_analysis(prompt, style_option):
-    """生成艺术作品并分析伦理风险"""
+    # 构建风格化提示词
+    style_prompts = {
+        "梵高风格": f"{prompt}, 梵高风格, 油画笔触, 浓烈色彩",
+        "水墨风格": f"{prompt}, 中国传统水墨风格, 黑白灰调, 留白意境", 
+        "数字艺术": f"{prompt}, 数字艺术, 未来科技感, 赛博朋克",
+        "写实风格": f"{prompt}, 摄影风格, 真实感, 高细节"
+    }
     
-    # 构建完整的提示词
-    if style_option == "梵高风格":
-        full_prompt = f"{prompt}, 梵高风格, 油画笔触, 星空效果"
-    elif style_option == "水墨风格":
-        full_prompt = f"{prompt}, 中国传统水墨风格, 黑白灰调, 留白"
-    elif style_option == "数字艺术":
-        full_prompt = f"{prompt}, 数字艺术, 未来感, 科技感"
-    else:  # 原创风格
-        full_prompt = f"{prompt}, 原创数字艺术"
+    full_prompt = style_prompts.get(style, prompt)
     
-    # 分析版权风险
-    risk_level, risk_indicators, suggestions = analyze_copyright_risk(prompt, style_option)
-    
-    # 尝试生成图像（仅在配置了Token时）
+    # 生成图像
     image = None
-    generation_status = "演示模式"
+    status = "等待生成..."
     
-    # 如果有Token，尝试真实生成
     if HF_TOKEN:
-        image_bytes = query_huggingface({"inputs": full_prompt})
-        if image_bytes:
-            image = Image.open(io.BytesIO(image_bytes))
-            generation_status = "🟢 使用AI生成"
-        else:
-            generation_status = "⚠️ AI生成失败 - 使用演示模式"
-    
-    # 构建详细报告
-    analysis_report = f"""
-## 📊 伦理分析报告
-
-**风险等级**: {risk_level}
-
-### 🔍 检测到的风险因素:
-{chr(10).join(['• ' + item for item in risk_indicators]) if risk_indicators else '• 未检测到明显风险因素'}
-
-### 💡 建议措施:
-{chr(10).join(['• ' + item for item in suggestions])}
-
-### 🎨 生成信息:
-- 提示词: {full_prompt}
-- 生成状态: {generation_status}
-- 风格选项: {style_option}
-- API状态: {'已配置' if HF_TOKEN else '未配置 - 使用演示模式'}
-
-> 💡 提示: {'已配置Hugging Face Token，可真实生成图像' if HF_TOKEN else '如需真实生成图像，请在Space设置中添加HF_TOKEN环境变量'}
-"""
+        try:
+            response = requests.post(API_URL, headers=headers, json={"inputs": full_prompt})
+            if response.status_code == 200:
+                image = Image.open(io.BytesIO(response.content))
+                status = "✅ 生成成功！"
+            else:
+                status = f"❌ 生成失败: {response.status_code}"
+        except Exception as e:
+            status = f"❌ 错误: {str(e)}"
+    else:
+        status = "❌ 未配置API Token"
     
-    return image, analysis_report
+    return image, status, full_prompt
 
-# 创建Gradio界面
-with gr.Blocks(theme=gr.themes.Soft(), title="AI艺术生成与版权伦理探索") as demo:
+# 创建简洁的Gradio界面
+with gr.Blocks(theme=gr.themes.Soft(), title="AI艺术生成演示") as demo:
     gr.Markdown("""
-    # 🎨 AI艺术生成与版权伦理探索
-    
+    # 🎨 AI艺术生成演示
     ## 技术之善 - AI项目制作与伦理探索
     
-    本演示展示AI艺术生成的能力，同时深入分析相关的版权和伦理问题。
+    **重点展示**: Stable Diffusion 图像生成能力
     """)
     
     with gr.Row():
-        with gr.Column(scale=1):
-            gr.Markdown("### 🎯 创作设置")
-            
+        with gr.Column():
             prompt_input = gr.Textbox(
                 label="创作描述",
-                placeholder="例如：星空下的城市、山水风景、未来科技...",
-                lines=3,
-                value="星空下的奇幻城市"
+                placeholder="描述你想要生成的画面...",
+                value="星空下的城市夜景",
+                lines=2
             )
             
-            style_dropdown = gr.Dropdown(
-                choices=["原创风格", "梵高风格", "水墨风格", "数字艺术"],
+            style_radio = gr.Radio(
+                choices=["梵高风格", "水墨风格", "数字艺术", "写实风格"],
                 label="艺术风格",
-                value="原创风格",
-                info="选择风格可能影响版权风险评估"
-            )
-            
-            generate_btn = gr.Button(
-                "🚀 生成并分析",
-                variant="primary",
-                size="lg"
+                value="梵高风格"
             )
             
-            gr.Markdown("""
-            ### 💡 使用提示
-            - 尽量使用原创描述而非特定艺术家名称
-            - 高风险内容会触发详细伦理分析
-            - 所有生成内容仅供教学演示使用
-            """)
+            generate_btn = gr.Button("生成图像", variant="primary", size="lg")
         
-        with gr.Column(scale=1):
-            gr.Markdown("### 🖼️ 生成结果")
+        with gr.Column():
             image_output = gr.Image(
-                label="AI生成的艺术作品",
+                label="生成的艺术作品",
                 height=400,
-                format="png"
+                show_download_button=True
             )
             
-            gr.Markdown("### 📋 伦理分析")
-            analysis_output = gr.Markdown(
-                label="版权风险评估",
-                value="请先生成艺术作品以查看分析报告..."
+            status_output = gr.Textbox(
+                label="生成状态",
+                interactive=False
+            )
+            
+            prompt_used = gr.Textbox(
+                label="实际使用的提示词",
+                interactive=False
             )
     
-    # 示例区
-    gr.Markdown("## 🎪 快速示例")
+    # 示例
+    gr.Markdown("### 💡 示例提示词")
     examples = gr.Examples(
         examples=[
             ["星空下的城市夜景", "梵高风格"],
-            ["山水间的亭台楼阁", "水墨风格"],
+            ["山水间的古代亭台", "水墨风格"],
             ["未来科技的都市景观", "数字艺术"],
-            ["花朵盛开的春日花园", "原创风格"]
+            ["森林中的小鹿", "写实风格"]
         ],
-        inputs=[prompt_input, style_dropdown],
-        label="点击示例快速填充"
+        inputs=[prompt_input, style_radio]
     )
     
-    # 页脚状态显示
-    status_text = "🔴 未配置API Token - 演示模式" if not HF_TOKEN else "🟢 API Token已配置 - 可真实生成图像"
-    gr.Markdown(f"""
-    ---
-    **系统状态**: {status_text}
-    
-    *本项目为工程伦理课程演示，重点探讨AI艺术生成的版权伦理问题*
-    *演示系统基于 Gradio + Hugging Face Spaces 构建*
-    """)
+    # 页脚
+    if HF_TOKEN:
+        gr.Markdown("---\n**系统状态**: 🟢 已配置API - 可真实生成图像")
+    else:
+        gr.Markdown("---\n**系统状态**: 🔴 演示模式 - 请在设置中配置HF_TOKEN环境变量")
     
     # 事件处理
     generate_btn.click(
-        fn=generate_art_with_analysis,
-        inputs=[prompt_input, style_dropdown],
-        outputs=[image_output, analysis_output]
+        fn=generate_image,
+        inputs=[prompt_input, style_radio],
+        outputs=[image_output, status_output, prompt_used]
     )
 
 if __name__ == "__main__":