Update app.py

app.py

@@ -4,7 +4,9 @@ from huggingface_hub import InferenceClient
 import gradio as gr
 from huggingface_hub import InferenceClient
 import os
-import zipfile  #  추가
+import zipfile  
+import nbformat
+from nbconvert.preprocessors import ExecutePreprocessor
 
 # zip 파일 자동 해제 코드
 zip_path = "solo_leveling_faiss_ko.zip"
@@ -22,6 +24,12 @@ For more information on `huggingface_hub` Inference API support, please check th
 """
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
+def run_notebook(path):
+    with open(path, encoding='utf-8') as f:
+        nb = nbformat.read(f, as_version=4)
+    ep = ExecutePreprocessor(timeout=600, kernel_name='python3')
+    ep.preprocess(nb, {'metadata': {'path': './'}})  # 현재 경로에서 실행
+    print(f"[INFO] Notebook {path} 실행 완료")
 
 def respond(
     message,
@@ -77,4 +85,5 @@ demo = gr.ChatInterface(
 
 
 if __name__ == "__main__":
+    run_notebook("r-story-test.ipynb")
     demo.launch()