Update app.py

app.py

@@ -1,8 +1,11 @@
 import streamlit as st
-import openai
+import os
+from groq import Groq
 
-# Hugging Face Spacesの設定でAPIキーを設定していると仮定
-openai.api_key = st.secrets["GROQ_API_KEY"]
+# APIキーの設定
+client = Groq(
+    api_key=st.secrets["GROQ_API_KEY"],
+)
 
 # Streamlitアプリケーションのタイトルと説明
 st.title("Groq APIによるAIチャットボット")
@@ -17,8 +20,8 @@ user_input = st.text_input("質問を入力してください:")
 
 # Groq APIにリクエストを送信し、ストリーミング応答を取得する関数
 def get_groq_response(prompt):
-    response = openai.ChatCompletion.create(
-        model="gemma2-9b-it",  # 適切なモデル名に置き換えてください
+    response = client.chat.completions.create(
+        model="llama3-8b-8192",  # 適切なモデル名に置き換えてください
         messages=[{"role": "user", "content": prompt}],
         stream=True  # ストリーミングモードを有効にする
     )
@@ -39,8 +42,8 @@ if user_input:
     
     # ストリーミングで部分的な応答をリアルタイムで表示
     for chunk in response:
-        if chunk:
-            chunk_text = chunk.choices[0].delta.content
+        if "choices" in chunk:
+            chunk_text = chunk['choices'][0]['delta']['content']
             response_text += chunk_text
             st.write(response_text)  # 部分的な応答を表示