Update app.py

app.py

@@ -11,7 +11,6 @@ def search_youtube_videos(keyword):
     video_urls = [video['link'] for video in results['result']]
     return video_urls
 
-# Function to get the transcript for a given list of video URLs
 def get_transcript(urls):
     contents = ''
     for url in urls:
@@ -21,12 +20,12 @@ def get_transcript(urls):
             text += segment['text']
         contents += text
     return contents
-
-# Function to summarize text using OpenAI's API
+    
 def summarize_text(contents, OPENAI_API_KEY):
     
     response = openai.ChatCompletion.create(
-        model="gpt-4-turbo-preview",  # Corrected engine identifier
+        
+        model="gpt-4-turbo-preview",  
         messages=[
             {"role": "system", "content": "You are a helpful assistant."},
             {"role": "user", "content": f"Summarize this: {contents}"}
@@ -37,6 +36,7 @@ def summarize_text(contents, OPENAI_API_KEY):
         }
     )
     return response.choices[0].message['content'].strip()
+    
 # def summarize_text(contents):
 #     response = openai.ChatCompletion.create(
 #         engine="gpt-3.5-turbo",
@@ -46,12 +46,17 @@ def summarize_text(contents, OPENAI_API_KEY):
 #     return response.choices[0].text.strip()
 
 # Function to integrate all the functionalities
+    
 def summarize_youtube_videos(keyword, OPENAI_API_KEY):
     urls = search_youtube_videos(keyword)
     contents = get_transcript(urls)
     summary = summarize_text(contents,OPENAI_API_KEY)
     return summary
 
+def summarize_button_click(keyword, OPENAI_API_KEY):
+    summary = summarize_youtube_videos(keyword, OPENAI_API_KEY)
+    return summary
+    
 # Define Gradio interface
 iface = gr.Interface(
     inputs=[
@@ -64,9 +69,7 @@ iface = gr.Interface(
     fn=summarize_button_click  # 클릭 이벤트를 처리하는 함수를 여기에 정의합니다.
 )
 
-def summarize_button_click(keyword, OPENAI_API_KEY):
-    summary = summarize_youtube_videos(keyword, OPENAI_API_KEY)
-    return summary
+
 
 # Launch Gradio interface
 iface.launch()