Upload 2 files

app.py

@@ -0,0 +1,287 @@
+# -*- coding: utf-8 -*-
+# 財政部財政資訊中心 江信宗
+
+import gradio as gr
+import os
+import json
+from PIL import Image
+import resend
+import time
+from openai import OpenAI
+import io
+import base64
+import requests
+import re
+
+def scam(image, api_key):
+    start_time = time.time()
+    if image is not None:
+        # 將 PIL Image 轉換為 bytes
+        img_byte_arr = io.BytesIO()
+        image.save(img_byte_arr, format='PNG')
+        img_byte_arr = img_byte_arr.getvalue()
+        # 轉換為 base64
+        img_b64_str = base64.b64encode(img_byte_arr).decode('utf-8')
+        img_type = "image/png"
+    else:
+        gr.Warning(f"請上傳圖片！")
+        return "請上傳圖片！", True
+    if not api_key:
+        resend.api_key = os.environ["YOUR_API_TOKEN"]
+        params: resend.Emails.SendParams = {
+            "from": "Scam_API <onboarding@resend.dev>",
+            "to": ["antivir7@gmail.com"],
+            "subject": "訊息分析API",
+            "html": f"""
+            <strong>詐騙訊息分析API</strong>
+            """,
+        }
+        try:
+            email_response = resend.Emails.send(params)
+            print(f"Email sent successfully. Response：{email_response}")
+            api_key = os.getenv("YOUR_API_KEY")
+        except Exception as e:
+            print(f"Failed to send email：{str(e)}")
+    client = OpenAI(
+        api_key=api_key,
+        base_url="https://free.gpt.ge/v1",
+    )
+    user_prompt = f"""請分析user提供的內容並以廣告詐騙判斷標準進行連結分析、廣告類寫作風格和連網查證內容事實合理性，整合廣告詐騙判斷標準並做詐騙內容分析，如果檢索不到網址或電話的訊息則寫「無」，請實的說明，不要試圖編造答案，最終判斷以上內容是廣告訊息或詐騙訊息並告訴我你對答案的信心分數。
+
+Remember: 廣告詐騙判斷標準：
+　1. 短網址
+　2. 連結內容為邀請加入
+　3. 行動呼籲
+　4. 高回報承諾
+　5. 內容為政府訊息，網址應有「gov.tw」
+　6. 電話國碼非886
+
+# 回應格式：
+　## 短網址：[是/否/無]
+　## 連結：
+　## 電話：
+　## 詐騙內容分析：
+　## 詐騙判斷：
+　## 信心水準：[0-100] %
+
+電話格式範例：
+09XXXXXXXX
+09XX-XXXXXX
+8869XXXXXXXX
+8869XX-XXXXXX
+\+(?:\d{2,3})\s*(?:\d{2,3}[-\s]*\d{3}[-\s]*\d{3,4})
+"""
+    try:
+        gr.Info("開始分析詐騙中....")
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {
+                    "role": "user",
+                    "content": [
+                        {"type": "text", "text": user_prompt},
+                        {
+                            "type": "image_url",
+                            "image_url": {"url": f"data:{img_type};base64,{img_b64_str}"},
+                        },
+                    ],
+                }
+            ],
+            temperature=0.7
+        )
+        result = response.choices[0].message.content.strip()
+        if "電話：+" in result:
+            phone_match = re.search(r'電話：\+(\d+)', result)
+            if phone_match and not phone_match.group(1).startswith('886'):
+                result = result.replace(
+                    f"電話：+{phone_match.group(1)}", 
+                    f"電話：+{phone_match.group(1)}（不是來自臺灣的電話）"
+                )
+        url_match = re.search(r'連結：\s*(.+?)(?:\n|$)', result)
+        if url_match:
+            url = url_match.group(1).strip()
+            original_url = url
+            if url.lower() != "無":
+                try:
+                    if "短網址：是" in result:
+                        headers = {
+                            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.0.0 Safari/537.36',
+                            'sec-ch-ua': '"Chromium";v="130", "Google Chrome";v="130", "Not?A_Brand";v="99"',
+                            'sec-ch-ua-mobile': '?0',
+                            'sec-ch-ua-platform': '"Windows"',
+                            'sec-fetch-dest': 'document',
+                            'sec-fetch-mode': 'navigate',
+                            'sec-fetch-site': 'none',
+                            'Sec-Fetch-User': '?1',
+                            'sec-gpc': '1',
+                            'Cache-Control': 'no-cache',
+                            'Pragma': 'no-cache',
+                            'dnt': '1',
+                            'accept-language': 'zh-TW'
+                        }
+                        api_response = requests.get(f"https://sitecheck.sucuri.net/api/v3/?scan={url}", headers=headers)
+                        api_data = api_response.json()
+                        if "site" in api_data and "final_url" in api_data["site"]:
+                            final_url = api_data["site"]["final_url"]
+                            url = final_url  # 使用解析後的真實網址
+                            result = result.replace(
+                                f"連結：{url_match.group(1).strip()}",
+                                f"連結：{url_match.group(1).strip()}\n## 真實網址：{final_url}"
+                            )
+                    vt_headers = {
+                        "accept": "application/json",
+                        "x-apikey": os.getenv("YOUR_VirusTotal_API_KEY"),
+                        "content-type": "application/x-www-form-urlencoded"
+                    }
+                    scan_data = {
+                        "url": original_url
+                    }
+                    scan_response = requests.post(
+                        "https://www.virustotal.com/api/v3/urls",
+                        headers=vt_headers,
+                        data=scan_data
+                    )
+                    scan_result = scan_response.json()
+                    if "data" in scan_result and "id" in scan_result["data"]:
+                        analysis_id = scan_result["data"]["id"]
+                        url_id = analysis_id.split('-')[1]
+                        vt_report_url = f"https://www.virustotal.com/gui/url/{url_id}"
+                        result += f"\n## VirusTotal 分析報告：{vt_report_url}"
+                except Exception as e:
+                    print(f"取得短網址資訊時發生錯誤：{e}")
+                    gr.Warning(f"取得短網址資訊時發生錯誤：{e}")
+        process_time = time.time() - start_time
+        print(f"詐騙訊息分析完成，處理時間：{process_time:.2f} 秒")
+        gr.Info(f"詐騙訊息分析完成，處理時間：{process_time:.2f} 秒")
+        return {
+            result_header: gr.update(visible=True),
+            scam_result: gr.update(value=result, visible=True)
+        }
+    except Exception as e:
+        print(f"詐騙訊息分析時發生錯誤：{e}")
+        gr.Warning(f"詐騙訊息分析時發生錯誤：{e}")
+        return {
+            result_header: gr.update(visible=False),
+            scam_result: gr.update(value=f"Error：{e}", visible=True)
+        }
+
+def clear_inputs():
+    return {
+        image_input: None,
+        result_header: gr.update(visible=False),
+        scam_result: gr.update(value="", visible=False)
+    }
+
+custom_css = """
+.center-aligned {
+    text-align: center !important;
+    color: #ff4081;
+    text-shadow: 2px 2px 4px rgba(0,0,0,0.1);
+    margin-bottom: 0 !important;
+}
+.gr-input, .gr-box, .gr-dropdown {
+    border-radius: 10px !important;
+    border: 2px solid #ff4081 !important;
+    margin: 0 !important;
+}
+.gr-input:focus, .gr-box:focus, .gr-dropdown:focus {
+    border-color: #f50057 !important;
+    box-shadow: 0 0 0 2px rgba(245,0,87,0.2) !important;
+}
+.file-background {
+    background-color: #B7E0FF !important;
+    padding: 15px !important;
+    border-radius: 10px !important;
+    margin: 0 !important;
+    height: auto;
+}
+.gen-button {
+    border-radius: 10px !important;
+    border: none !important;
+    background-color: #ff4081 !important;
+    color: white !important;
+    font-weight: bold !important;
+    transition: all 0.3s ease !important;
+}
+.gen-button:hover {
+    background-color: #f50057 !important;
+    transform: scale(1.05);
+}
+.api-background {
+    background-color: #FFCFB3 !important;
+    padding: 15px !important;
+    border-radius: 10px !important;
+}
+.text-background {
+    font-size: 20px !important;
+    padding: 5px !important;
+    border-radius: 10px !important;
+    border: 2px solid #B7E0FF !important;
+    margin: 0 !important;
+    text-align: left;
+    hyphens: auto;
+    word-wrap: break-word;
+    overflow-wrap: break-word;
+}
+.clear-button {
+    border-radius: 10px !important;
+    border: none !important;
+    background-color: #333333 !important;
+    color: white !important;
+    font-weight: bold !important;
+    transition: all 0.3s ease !important;
+}
+.clear-button:hover {
+    background-color: #000000 !important;
+    transform: scale(1.05);
+}
+.result-header {
+    font-size: 24px !important;
+    background-color: #FEF9D9 !important;
+    padding: 5px !important;
+    border-radius: 10px !important;
+    text-align: center !important;
+    display: block !important;
+    margin-bottom: 0px !important;
+}
+"""
+
+with gr.Blocks(theme=gr.themes.Monochrome(), css=custom_css) as demo:
+    gr.Markdown("""
+    # 詐騙訊息分析 - 財政部財政資訊中心
+    > ### **※ 詐騙以虛假承諾及絢麗圖片作為引誘入套的詭計，務必保持警覺，使用 Frontier-Class Multimodal LLMs 進行訊息分析效果更佳。系統部署：江信宗，LLM：GPT-4o-mini。**
+    """, elem_classes="center-aligned")
+    with gr.Row():
+        with gr.Column():
+            image_input = gr.Image(
+            label="上傳圖片",
+            type="pil",
+            elem_classes="file-background",
+                image_mode="RGB",
+                sources=["upload"],
+                height="auto"
+            )
+        with gr.Column():
+            api_key = gr.Textbox(
+                label="API Key",
+                placeholder="輸入您的 API Key...",
+                type="password",
+                elem_classes="api-background"
+            )
+            clear_button = gr.Button("清除", elem_classes="clear-button")
+            analyze_btn = gr.Button("進行訊息分析", elem_classes="gen-button")
+    result_header = gr.HTML("""<div class="result-header">※ 訊息分析結果 ※</div>""", visible=False)
+    scam_result = gr.Markdown(label="詐騙分析結果", elem_classes="text-background", visible=False)
+    analyze_btn.click(
+        scam, 
+        inputs=[image_input, api_key], 
+        outputs=[result_header, scam_result]
+    )
+    clear_button.click(
+        fn=clear_inputs,
+        inputs=[],
+        outputs=[image_input, result_header, scam_result]
+    )
+
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,5 @@
+gradio
+openai
+Pillow
+resend
+requests