Update app.py

app.py

@@ -1,12 +1,51 @@
+import os
 import gradio as gr
-from transformers import pipeline
 
-# 創建語音識別的 pipeline
-asr = pipeline("automatic-speech-recognition")
+# 確保 groq 套件可用
+try:
+    from groq import Groq
+except ImportError:
+    os.system('pip install groq')
+    from groq import Groq
 
-def recognize_speech(audio):
-    result = asr(audio)
-    return result['text']
+# 初始化 Groq 客戶端
+client = Groq()
 
-iface = gr.Interface(fn=recognize_speech, inputs=gr.Audio(source="microphone", type="filepath"), outputs="text", title="語音轉文字應用", description="請說話，然後按下停止按鈕進行文字轉換。")
+# 定義 chatbot 回應的函數
+def chat_with_groq(user_input):
+    completion = client.chat.completions.create(
+        model="llama-3.1-70b-versatile",
+        messages=[
+            {
+                "role": "system",
+                "content": "我是國文老師，專門批改國小作文"
+            },
+            {
+                "role": "user",
+                "content": user_input
+            }
+        ],
+        temperature=1,
+        max_tokens=1024,
+        top_p=1,
+        stream=True,
+        stop=None,
+    )
+
+    response = ""
+    for chunk in completion:
+        response += chunk.choices[0].delta.content or ""
+    
+    return response
+
+# 使用 Gradio 創建 chatbot
+iface = gr.Interface(
+    fn=chat_with_groq,
+    inputs=gr.inputs.Textbox(label="請輸入您的問題："),
+    outputs=gr.outputs.Textbox(label="回應："),
+    title="國文老師 Chatbot",
+    description="這是一個專門批改國小作文的 Chatbot"
+)
+
+# 啟動應用
 iface.launch()