Update app.py

app.py

@@ -14,25 +14,8 @@ openai.api_key = os.getenv("OPENAI_API_KEY")
 
 # 문장 구분 함수
 def split_sentences(text):
-    sentences = re.split(r"(니다|에요|구나|해요|군요|겠어요|시오|해라|예요|아요|데요|대요|세요|어요|게요|구요|고요|나요|하죠)(?![\w])", text)
-    combined_sentences = []
-    current_sentence = ""
-    for i in range(0, len(sentences), 2):
-        if i + 1 < len(sentences):
-            sentence = sentences[i] + sentences[i + 1]
-        else:
-            sentence = sentences[i]
-        if len(current_sentence) + len(sentence) > 100:  # 100자를 초과할 경우
-            combined_sentences.append(current_sentence.strip())
-            current_sentence = sentence.strip()
-        else:
-            current_sentence += sentence
-        if sentence.endswith(('.', '?', '!')):
-            combined_sentences.append(current_sentence.strip())
-            current_sentence = ""
-    if current_sentence:
-        combined_sentences.append(current_sentence.strip())
-    return combined_sentences
+    sentences = re.split(r'([.!?]+)\s*', text)
+    return ''.join(sentences).split()
 
 def parse_api_response(response):
     try:
@@ -116,17 +99,18 @@ def analyze(url, progress=gr.Progress()):
         
         progress(33, desc="원문 스크립트 처리 중...")
         script_sentences = split_sentences(script)
-        script_content = "\n".join(script_sentences)
+        script_content = f"# {title}\n\n" + "\n".join(script_sentences)
         
         progress(66, desc="요약 생성 중...")
         summary = summarize_text(title, description, script)
+        summary_content = f"# {title}\n\n{summary}"
         
         progress(100, desc="완료")
-        return title, script_content, summary
+        return script_content, summary_content
     except Exception as e:
         error_msg = f"처리 중 오류 발생: {str(e)}"
         logging.exception(error_msg)
-        return "오류 발생", error_msg, error_msg
+        return error_msg, error_msg
 
 # Gradio 인터페이스
 with gr.Blocks() as demo:
@@ -134,8 +118,6 @@ with gr.Blocks() as demo:
     youtube_url_input = gr.Textbox(label="YouTube URL 입력")
     analyze_button = gr.Button("분석하기")
     
-    title_output = gr.Textbox(label="영상 제목")
-    
     with gr.Tabs():
         with gr.TabItem("원문 스크립트"):
             script_output = gr.Markdown()
@@ -145,7 +127,7 @@ with gr.Blocks() as demo:
     analyze_button.click(
         analyze,
         inputs=[youtube_url_input],
-        outputs=[title_output, script_output, summary_output]
+        outputs=[script_output, summary_output]
     )
 
 if __name__ == "__main__":