Create app.py

app.py

@@ -0,0 +1,50 @@
+import os
+import openai
+import gradio as gr
+from gradio import ChatInterface
+from google.colab import userdata
+
+# 設定 OpenAI API key - 從環境變數獲取
+openai.api_key = os.environ['OPENAI_API_KEY']
+
+def predict(inputs, chatbot):
+    messages = []
+    
+    # 建立對話歷史
+    for conv in chatbot:
+        messages.append({"role": "user", "content": conv[0]})
+        messages.append({"role": "assistant", "content": conv[1]})
+    
+    # 加入新的用戶輸入
+    messages.append({"role": "user", "content": inputs})
+
+    try:
+        # 建立 ChatCompletion 請求
+        response = openai.ChatCompletion.create(
+            model='gpt-4',
+            messages=messages,
+            temperature=1.0,
+            stream=True
+        )
+
+        partial_message = ""
+        for chunk in response:
+            if 'choices' in chunk and len(chunk['choices']) > 0:
+                if 'delta' in chunk['choices'][0] and 'content' in chunk['choices'][0]['delta']:
+                    content = chunk['choices'][0]['delta']['content']
+                    partial_message += content
+                    yield partial_message
+
+    except Exception as e:
+        yield f"發生錯誤: {str(e)}"
+
+# 創建和啟動聊天界面
+chat_interface = gr.ChatInterface(
+    predict,
+    chatbot=gr.Chatbot(height=600),
+    title="AI 聊天助手",
+    description="請輸入您的問題",
+)
+
+if __name__ == "__main__":
+    chat_interface.launch()