test: 任意でYouTube API KEYを入力できる

.gitignore

@@ -0,0 +1,2 @@
+.env
+*.txt

app.py

@@ -1,4 +1,5 @@
 # 必要なライブラリをインポート
+import os
 import streamlit as st
 from googleapiclient.discovery import build
 from googleapiclient.errors import HttpError
@@ -10,6 +11,8 @@ from create_chatbot import chatbot
 YOUTUBE_API_SERVICE_NAME = 'youtube'
 YOUTUBE_API_VERSION = 'v3'
 
+YOUTUBE_API_KEY = os.environ.get('YOUTUBE_API_KEY')
+
 # Hugging Faceの設定
 classifier = pipeline('sentiment-analysis')
 
@@ -19,17 +22,20 @@ def main():
 
     # API Keyや質問の入力
     openai_api_key = st.text_input('Enter your OpenAI API key')
-    yt_api_key = st.text_input('Enter your YouTube Data API key')
     video_url = st.text_input('Enter a YouTube video URL')
     question = st.text_input('Enter a question')
     
     with st.expander('Options'):
         # OpenAIのモデルの選択などの設定
-        model_name = st.selectbox('Select a model', ['gpt-3.5-turbo', 'gpt-4'])
+        model_name = st.selectbox('Select a GPT model', ['gpt-3.5-turbo', 'gpt-4'])]
+        yt_api_key = st.text_input('Enter your YouTube Data API key')
 
-    if openai_api_key and yt_api_key and video_url and question:
+    if openai_api_key and video_url and question:
         video_id = video_url.split('v=')[1]
-        youtube = build(YOUTUBE_API_SERVICE_NAME, YOUTUBE_API_VERSION, developerKey=yt_api_key)
+        youtube = build(
+            YOUTUBE_API_SERVICE_NAME, 
+            YOUTUBE_API_VERSION, 
+            developerKey=yt_api_key if yt_api_key else YOUTUBE_API_KEY)
 
         try:
             get_comments(youtube, video_id)