Update app.py

app.py

@@ -1,209 +1,208 @@
 import gradio as gr
-import os
+from groq import Groq
 from twilio.rest import Client
-import requests
-import json
+import base64
+import io
+from PIL import Image
 
-class APIManager:
-    def __init__(self):
-        self.twilio_client = None
-        self.groq_api_key = None
-        self.twilio_from_number = None
-        
-    def update_credentials(self, twilio_sid, twilio_token, twilio_from, groq_key):
-        """更新 API 憑證"""
-        try:
-            # 驗證 Twilio 憑證
-            if twilio_sid and twilio_token:
-                test_client = Client(twilio_sid, twilio_token)
-                # 簡單測試連接
-                test_client.api.accounts(twilio_sid).fetch()
-                self.twilio_client = test_client
-                self.twilio_from_number = twilio_from
-                
-            # 更新 GROQ API Key
-            if groq_key:
-                self.groq_api_key = groq_key
-                
-            return "✅ API 憑證更新成功！", gr.update(visible=True)
-        except Exception as e:
-            return f"❌ 憑證驗證失敗: {str(e)}", gr.update(visible=False)
-
-# 創建 API 管理器實例
-api_manager = APIManager()
+def encode_image(image):
+    """將PIL圖片編碼為base64字符串"""
+    buffered = io.BytesIO()
+    image.save(buffered, format="JPEG")
+    return base64.b64encode(buffered.getvalue()).decode("utf-8")
 
-def send_sms(to_number, message):
-    """發送 SMS 訊息"""
-    if not api_manager.twilio_client:
-        return "❌ 請先設定 Twilio 憑證"
+def analyze_and_send_sms(
+    image, 
+    groq_api_key, 
+    account_sid, 
+    auth_token, 
+    from_number, 
+    to_number, 
+    custom_message=""
+):
+    """
+    分析圖片並發送SMS訊息
     
-    try:
-        message_obj = api_manager.twilio_client.messages.create(
-            body=message,
-            from_=api_manager.twilio_from_number,
-            to=to_number
-        )
-        return f"✅ SMS 發送成功！訊息 SID: {message_obj.sid}"
-    except Exception as e:
-        return f"❌ SMS 發送失敗: {str(e)}"
-
-def call_groq_api(prompt, model="mixtral-8x7b-32768"):
-    """調用 GROQ API"""
-    if not api_manager.groq_api_key:
-        return "❌ 請先設定 GROQ API Key"
+    Args:
+        image: 上傳的圖片
+        groq_api_key: Groq API金鑰
+        account_sid: Twilio帳戶SID
+        auth_token: Twilio認證令牌
+        from_number: Twilio虛擬手機號碼
+        to_number: 接收SMS的手機號碼
+        custom_message: 自定義訊息內容
+    
+    Returns:
+        tuple: (圖片分析結果, SMS發送狀態)
+    """
     
     try:
-        headers = {
-            "Authorization": f"Bearer {api_manager.groq_api_key}",
-            "Content-Type": "application/json"
-        }
+        # 步驟1: 使用Groq分析圖片
+        if image is None:
+            return "錯誤：請上傳圖片", "未發送SMS"
         
-        data = {
-            "messages": [
-                {"role": "user", "content": prompt}
-            ],
-            "model": model,
-            "max_tokens": 1000
+        if not groq_api_key:
+            return "錯誤：請輸入Groq API金鑰", "未發送SMS"
+        
+        # 編碼圖片
+        base64_image = encode_image(image)
+        image_content = {
+            "type": "image_url",
+            "image_url": {"url": f"data:image/jpeg;base64,{base64_image}"}
         }
         
-        response = requests.post(
-            "https://api.groq.com/openai/v1/chat/completions",
-            headers=headers,
-            json=data
+        # 設定提示詞
+        prompt = "解釋圖片的內容，10個字說明，使用繁體中文。"
+        
+        # 建立Groq客戶端
+        groq_client = Groq(api_key=groq_api_key)
+        
+        # 呼叫Groq API
+        completion = groq_client.chat.completions.create(
+            model="meta-llama/llama-4-scout-17b-16e-instruct",
+            messages=[{
+                "role": "user",
+                "content": [
+                    {"type": "text", "text": prompt},
+                    image_content
+                ]
+            }],
+            temperature=1,
+            max_completion_tokens=512,
+            top_p=1,
+            stream=False,
+            stop=None,
         )
         
-        if response.status_code == 200:
-            result = response.json()
-            return result["choices"][0]["message"]["content"]
-        else:
-            return f"❌ GROQ API 錯誤: {response.status_code} - {response.text}"
-            
-    except Exception as e:
-        return f"❌ GROQ API 調用失敗: {str(e)}"
-
-def create_gradio_interface():
-    with gr.Blocks(title="Dynamic API Configuration Demo", theme=gr.themes.Soft()) as demo:
+        # 取得圖片分析結果
+        analysis_result = completion.choices[0].message.content
         
-        gr.Markdown("# 🔧 Dynamic API Configuration Demo")
-        gr.Markdown("### 動態設定 API 憑證，無需預先寫在程式中")
+        # 步驟2: 發送SMS
+        if not all([account_sid, auth_token, from_number, to_number]):
+            return analysis_result, "錯誤：請填入所有Twilio設定"
         
-        with gr.Row():
-            with gr.Column(scale=1):
-                gr.Markdown("## 📱 API 設定")
-                
-                # API 憑證輸入區域
-                with gr.Group():
-                    gr.Markdown("### Twilio 設定")
-                    twilio_sid = gr.Textbox(
-                        label="TWILIO_ACCOUNT_SID",
-                        placeholder="輸入您的 Twilio Account SID",
-                        type="password"
-                    )
-                    twilio_token = gr.Textbox(
-                        label="TWILIO_AUTH_TOKEN", 
-                        placeholder="輸入您的 Twilio Auth Token",
-                        type="password"
-                    )
-                    twilio_from = gr.Textbox(
-                        label="TWILIO_FROM_NUMBER",
-                        placeholder="+1234567890",
-                        value=""
-                    )
-                
-                with gr.Group():
-                    gr.Markdown("### GROQ 設定")
-                    groq_key = gr.Textbox(
-                        label="GROQ_API_KEY",
-                        placeholder="輸入您的 GROQ API Key", 
-                        type="password"
-                    )
-                
-                # 更新憑證按鈕
-                update_btn = gr.Button("🔄 更新 API 憑證", variant="primary")
-                status_msg = gr.Textbox(label="狀態", interactive=False)
-            
-            with gr.Column(scale=2):
-                gr.Markdown("## 🚀 功能測試")
-                
-                # SMS 功能區域
-                with gr.Group(visible=False) as sms_group:
-                    gr.Markdown("### 📱 SMS 發送測試")
-                    sms_to = gr.Textbox(label="收件人號碼", placeholder="+886912345678")
-                    sms_message = gr.Textbox(
-                        label="訊息內容", 
-                        placeholder="輸入要發送的訊息",
-                        lines=3
-                    )
-                    sms_btn = gr.Button("發送 SMS", variant="secondary")
-                    sms_result = gr.Textbox(label="SMS 結果", interactive=False)
-                
-                # GROQ 功能區域  
-                with gr.Group(visible=False) as groq_group:
-                    gr.Markdown("### 🤖 GROQ AI 測試")
-                    groq_prompt = gr.Textbox(
-                        label="提示詞",
-                        placeholder="輸入您想要詢問 AI 的問題",
-                        lines=3
-                    )
-                    groq_model = gr.Dropdown(
-                        choices=[
-                            "mixtral-8x7b-32768",
-                            "llama2-70b-4096", 
-                            "gemma-7b-it"
-                        ],
-                        value="mixtral-8x7b-32768",
-                        label="選擇模型"
-                    )
-                    groq_btn = gr.Button("調用 GROQ API", variant="secondary")
-                    groq_result = gr.Textbox(
-                        label="AI 回應", 
-                        lines=10,
-                        interactive=False
-                    )
+        # 準備SMS內容
+        if custom_message:
+            sms_body = custom_message
+        else:
+            sms_body = f"圖片分析結果：{analysis_result}"
         
-        # 事件處理
-        update_btn.click(
-            fn=api_manager.update_credentials,
-            inputs=[twilio_sid, twilio_token, twilio_from, groq_key],
-            outputs=[status_msg, sms_group, groq_group]
-        )
+        # 建立Twilio客戶端
+        twilio_client = Client(account_sid, auth_token)
         
-        sms_btn.click(
-            fn=send_sms,
-            inputs=[sms_to, sms_message],
-            outputs=[sms_result]
+        # 發送SMS
+        message = twilio_client.messages.create(
+            from_=from_number,
+            to=to_number,
+            body=sms_body
         )
         
-        groq_btn.click(
-            fn=call_groq_api,
-            inputs=[groq_prompt, groq_model],
-            outputs=[groq_result]
-        )
+        sms_status = f"SMS發送成功！訊息SID: {message.sid}"
         
-        # 示例說明
-        with gr.Row():
-            gr.Markdown("""
-            ## 📝 使用說明
+        return analysis_result, sms_status
+        
+    except Exception as e:
+        error_msg = f"發生錯誤: {str(e)}"
+        return error_msg, "SMS發送失敗"
+
+# 建立Gradio介面
+with gr.Blocks(title="圖片分析與SMS發送應用") as demo:
+    gr.Markdown("# 圖片分析與SMS發送應用")
+    gr.Markdown("上傳圖片進行AI分析，並可選擇發送分析結果到手機")
+    
+    with gr.Row():
+        with gr.Column():
+            # 圖��上傳
+            image_input = gr.Image(
+                type="pil", 
+                label="上傳圖片",
+                height=300
+            )
             
-            1. **設定 API 憑證**: 在左側輸入您的 API 憑證
-            2. **點擊更新**: 系統會驗證憑證的有效性
-            3. **測試功能**: 憑證驗證通過後，右側功能區域會顯示
-            4. **安全提醒**: 所有憑證僅在當前會話中使用，不會被儲存
+            # Groq API設定
+            groq_key = gr.Textbox(
+                label="Groq API金鑰",
+                type="password",
+                placeholder="輸入您的Groq API金鑰"
+            )
             
-            ### 🔒 安全特性
-            - 密碼欄位隱藏輸入內容
-            - 憑證僅存於記憶體中
-            - 不會記錄或儲存敏感資訊
-            - 每次重新載入需重新輸入
-            """)
+        with gr.Column():
+            # Twilio設定
+            account_sid = gr.Textbox(
+                label="Twilio Account SID",
+                placeholder="ACxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
+            )
+            auth_token = gr.Textbox(
+                label="Twilio Auth Token",
+                type="password",
+                placeholder="輸入您的Twilio認證令牌"
+            )
+            from_number = gr.Textbox(
+                label="Twilio虛擬手機號碼",
+                placeholder="+1xxxxxxxxxx"
+            )
+            to_number = gr.Textbox(
+                label="接收SMS的手機號碼",
+                placeholder="+886xxxxxxxxx"
+            )
+            custom_message = gr.Textbox(
+                label="自定義訊息內容 (選填)",
+                placeholder="留空則自動使用圖片分析結果",
+                lines=3
+            )
+    
+    # 執行按鈕
+    analyze_button = gr.Button("分析圖片並發送SMS", variant="primary")
+    
+    # 輸出結果
+    with gr.Row():
+        analysis_output = gr.Textbox(
+            label="圖片分析結果",
+            interactive=False
+        )
+        sms_output = gr.Textbox(
+            label="SMS發送狀態",
+            interactive=False
+        )
+    
+    # 設定按鈕點擊事件
+    analyze_button.click(
+        fn=analyze_and_send_sms,
+        inputs=[
+            image_input,
+            groq_key,
+            account_sid,
+            auth_token,
+            from_number,
+            to_number,
+            custom_message
+        ],
+        outputs=[analysis_output, sms_output]
+    )
+    
+    # 使用範例
+    gr.Markdown("""
+    ## 使用說明：
+    1. **上傳圖片**：選擇要分析的圖片檔案
+    2. **輸入Groq API金鑰**：從Groq官網取得您的API金鑰
+    3. **設定Twilio參數**：
+       - Account SID：您的Twilio帳戶識別碼
+       - Auth Token：您的Twilio認證令牌
+       - 虛擬手機號碼：Twilio提供的發送號碼
+       - 接收手機號碼：要接收SMS的號碼
+    4. **自定義訊息**（選填）：如果留空，會自動發送圖片分析結果
+    5. **點擊執行**：系統會分析圖片並發送SMS
     
-    return demo
+    ## 注意事項：
+    - 請確保Twilio帳戶有足夠餘額
+    - 手機號碼需包含國碼（如台灣：+886）
+    - API金鑰請妥善保管，勿洩露給他人
+    """)
 
 # 啟動應用程式
 if __name__ == "__main__":
-    app = create_gradio_interface()
-    app.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=False
+    demo.launch(
+        share=False,  # 設為True可產生公開連結
+        debug=True,
+        # 移除固定端口設定，讓Gradio自動尋找可用端口
     )