Update app.py

app.py

@@ -8,14 +8,14 @@ def process_video(url):
     API_KEY = os.getenv('GEMINI_API_KEY')
     
     if not API_KEY:
-        return "ERROR: GEMINI_API_KEY가 설정되지 않았습니다.", None
+        return "ERROR: GEMINI_API_KEY가 설정되지 않았습니다. .env 파일을 확인해주세요."
     
     worksheet = YouTubeWorksheet(API_KEY)
     
     # 자막 추출
     transcript = worksheet.get_transcript(url)
     if not transcript:
-        return "자막을 추출할 수 없습니다.", None
+        return "자막을 추출할 수 없습니다."
     
     # 워크시트 생성
     content = worksheet.create_worksheet(transcript)
@@ -25,18 +25,17 @@ def process_video(url):
     
     return f"워크시트가 생성되었습니다. 파일명: {output_file}", output_file
 
-# Gradio 인터페이스 생성
-load_dotenv()
-iface = gr.Interface(
-    fn=process_video,
-    inputs=[gr.Textbox(label="YouTube URL을 입력하세요", placeholder="https://www.youtube.com/watch?v=...")],
-    outputs=[
-        gr.Textbox(label="처리 결과"),
-        gr.File(label="생성된 워크시트")
-    ],
-    title="YouTube 학습 워크시트 생성기",
-    description="YouTube 영상의 자막을 이용하여 학습 워크시트를 생성합니다."
-)
+def main():
+    # Gradio 인터페이스 실행
+    iface = gr.Interface(
+        fn=process_video,
+        inputs=[gr.Textbox(label="YouTube URL을 입력하세요")],
+        outputs=[gr.Textbox(label="처리 결과"), gr.File(label="생성된 워크시트")],
+        title="YouTube 학습 워크시트 생성기",
+        description="YouTube 영상의 자막을 이용하여 학습 워크시트를 생성합니다."
+    )
+    iface.launch(share=True)
 
-if __name__ == "__main__":
-    iface.launch() 
+if __name__ == '__main__':
+    load_dotenv()
+    main()