Create app.py

app.py

@@ -0,0 +1,49 @@
+import os
+from openai import OpenAI
+import gradio as gr
+from gradio import ChatInterface
+
+# 設定 OpenAI API key - 從環境變數獲取
+client = 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 請求
+        stream = client.chat.completions.create(
+            model='gpt-4',  # 確保您有權限使用 gpt-4
+            messages=messages,
+            temperature=1.0,
+            stream=True,
+        )
+
+        partial_message = ""
+        for chunk in stream:
+            if chunk.choices[0].delta.content is not None:
+                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(),
+    title="AI 聊天助手",
+    description="請輸入您的問題",
+)
+
+if __name__ == "__main__":
+    chat_interface.launch()