ginipick commited on
Commit
1b0e9b2
ยท
verified ยท
1 Parent(s): 79f2daf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -616
app.py CHANGED
@@ -604,622 +604,6 @@ def chatbot_interface():
604
  def main():
605
  chatbot_interface()
606
 
607
- if __name__ == "__main__":
608
- # requirements.txt ํŒŒ์ผ ์ƒ์„ฑ
609
- with open("requirements.txt", "w") as f:
610
- f.write("streamlit>=1.31.0\n")
611
- f.write("anthropic>=0.18.1\n")
612
- f.write("gradio-client>=1.8.0\n")
613
- f.write("requests>=2.32.3\n")
614
- f.write("markdown>=3.5.1\n")
615
- f.write("pillow>=10.1.0\n")
616
-
617
- main()import os
618
- import streamlit as st
619
- import json
620
- import anthropic
621
- import requests
622
- import logging
623
- from gradio_client import Client
624
- import markdown
625
- import tempfile
626
- import base64
627
- from datetime import datetime
628
- import re
629
-
630
- # ๋กœ๊น… ์„ค์ •
631
- logging.basicConfig(
632
- level=logging.INFO,
633
- format='%(asctime)s - %(levelname)s - %(message)s')
634
-
635
- # API ์„ค์ •
636
- api_key = os.environ.get("API_KEY")
637
- client = anthropic.Anthropic(api_key=api_key)
638
-
639
- # ์ด๋ฏธ์ง€ ์ƒ์„ฑ API URL
640
- IMAGE_API_URL = "http://211.233.58.201:7896"
641
-
642
- # ์ตœ๋Œ€ ํ† ํฐ ์ˆ˜ ์„ค์ • (Claude-3 Sonnet์˜ ์ตœ๋Œ€ ํ† ํฐ ์ˆ˜)
643
- MAX_TOKENS = 7999
644
-
645
- # SerpHouse API Key ์„ค์ •
646
- SERPHOUSE_API_KEY = os.environ.get("SERPHOUSE_API_KEY", "")
647
-
648
- def get_system_prompt():
649
- return """
650
- ๋‹น์‹ ์€ ์ „๋ฌธ ๋ธ”๋กœ๊ทธ ์ž‘์„ฑ ์ „๋ฌธ๊ฐ€์ž…๋‹ˆ๋‹ค. ๋ชจ๋“  ๋ธ”๋กœ๊ทธ ๊ธ€ ์ž‘์„ฑ ์š”์ฒญ์— ๋Œ€ํ•ด ๋‹ค์Œ์˜ 8๋‹จ๊ณ„ ํ”„๋ ˆ์ž„์›Œํฌ๋ฅผ ์ฒ ์ €ํžˆ ๋”ฐ๋ฅด๋˜, ์ž์—ฐ์Šค๋Ÿฝ๊ณ  ๋งค๋ ฅ์ ์ธ ๊ธ€์ด ๋˜๋„๋ก ์ž‘์„ฑํ•ด์•ผ ํ•ฉ๋‹ˆ๋‹ค:
651
-
652
- ๋…์ž ์—ฐ๊ฒฐ ๋‹จ๊ณ„ 1.1. ๊ณต๊ฐ๋Œ€ ํ˜•์„ฑ์„ ์œ„ํ•œ ์นœ๊ทผํ•œ ์ธ์‚ฌ 1.2. ๋…์ž์˜ ์‹ค์ œ ๊ณ ๋ฏผ์„ ๋ฐ˜์˜ํ•œ ๋„์ž… ์งˆ๋ฌธ 1.3. ์ฃผ์ œ์— ๋Œ€ํ•œ ์ฆ‰๊ฐ์  ๊ด€์‹ฌ ์œ ๋„
653
-
654
- ๋ฌธ์ œ ์ •์˜ ๋‹จ๊ณ„ 2.1. ๋…์ž์˜ ํŽ˜์ธํฌ์ธํŠธ ๊ตฌ์ฒดํ™” 2.2. ๋ฌธ์ œ์˜ ์‹œ๊ธ‰์„ฑ๊ณผ ์˜ํ–ฅ๋„ ๋ถ„์„ 2.3. ํ•ด๊ฒฐ ํ•„์š”์„ฑ์— ๋Œ€ํ•œ ๊ณต๊ฐ๋Œ€ ํ˜•์„ฑ
655
-
656
- ์ „๋ฌธ์„ฑ ์ž…์ฆ ๋‹จ๊ณ„ 3.1. ๊ฐ๊ด€์  ๋ฐ์ดํ„ฐ ๊ธฐ๋ฐ˜ ๋ถ„์„ 3.2. ์ „๋ฌธ๊ฐ€ ๊ฒฌํ•ด์™€ ์—ฐ๊ตฌ ๊ฒฐ๊ณผ ์ธ์šฉ 3.3. ์‹ค์ œ ์‚ฌ๋ก€๋ฅผ ํ†ตํ•œ ๋ฌธ์ œ ๊ตฌ์ฒดํ™”
657
-
658
- ์†”๋ฃจ์…˜ ์ œ๊ณต ๋‹จ๊ณ„ 4.1. ๋‹จ๊ณ„๋ณ„ ์‹ค์ฒœ ๊ฐ€์ด๋“œ๋ผ์ธ ์ œ์‹œ 4.2. ์ฆ‰์‹œ ์ ์šฉ ๊ฐ€๋Šฅํ•œ ๊ตฌ์ฒด์  ํŒ 4.3. ์˜ˆ์ƒ ์žฅ์• ๋ฌผ๊ณผ ๊ทน๋ณต ๋ฐฉ์•ˆ ํฌํ•จ
659
-
660
- ์‹ ๋ขฐ๋„ ๊ฐ•ํ™” ๋‹จ๊ณ„ 5.1. ์‹ค์ œ ์„ฑ๊ณต ์‚ฌ๋ก€ ์ œ์‹œ 5.2. ๊ตฌ์ฒด์  ์‚ฌ์šฉ์ž ํ›„๊ธฐ ์ธ์šฉ 5.3. ๊ฐ๊ด€์  ๋ฐ์ดํ„ฐ๋กœ ํšจ๊ณผ ์ž…์ฆ
661
-
662
- ํ–‰๋™ ์œ ๋„ ๋‹จ๊ณ„ 6.1. ๋ช…ํ™•ํ•œ ์ฒซ ์‹ค์ฒœ ๋‹จ๊ณ„ ์ œ์‹œ 6.2. ์‹œ๊ธ‰์„ฑ์„ ๊ฐ•์กฐํ•œ ํ–‰๋™ ์ด‰๊ตฌ 6.3. ์‹ค์ฒœ ๋™๊ธฐ ๋ถ€์—ฌ ์š”์†Œ ํฌํ•จ
663
-
664
- ์ง„์ •์„ฑ ๊ฐ•ํ™” ๋‹จ๊ณ„ 7.1. ์†”๋ฃจ์…˜์˜ ํ•œ๊ณ„ ํˆฌ๋ช…ํ•˜๊ฒŒ ๊ณต๊ฐœ 7.2. ๊ฐœ์ธ๋ณ„ ์ฐจ์ด ์กด์žฌ ์ธ์ • 7.3. ํ•„์š” ์กฐ๊ฑด๊ณผ ์ฃผ์˜์‚ฌํ•ญ ๋ช…์‹œ
665
-
666
- ๊ด€๊ณ„ ์ง€์† ๋‹จ๊ณ„ 8.1. ์ง„์ •์„ฑ ์žˆ๋Š” ๊ฐ์‚ฌ ์ธ์‚ฌ 8.2. ๋‹ค์Œ ์ปจํ…์ธ  ์˜ˆ๊ณ ๋กœ ๊ธฐ๋Œ€๊ฐ ์กฐ์„ฑ 8.3. ์†Œํ†ต ์ฑ„๋„ ์•ˆ๋‚ด
667
-
668
- ์ž‘์„ฑ ์‹œ ์ค€์ˆ˜์‚ฌํ•ญ 9.1. ๊ธ€์ž ์ˆ˜: 1500-2000์ž ๋‚ด์™ธ 9.2. ๋ฌธ๋‹จ ๊ธธ์ด: 3-4๋ฌธ์žฅ ์ด๋‚ด 9.3. ์‹œ๊ฐ์  ๊ตฌ๋ถ„: ์†Œ์ œ๋ชฉ, ๊ตฌ๋ถ„์„ , ๋ฒˆํ˜ธ ๋ชฉ๋ก ํ™œ์šฉ 9.4. ํ†ค์•ค๋งค๋„ˆ: ์นœ๊ทผํ•˜๊ณ  ์ „๋ฌธ์ ์ธ ๋Œ€ํ™”์ฒด 9.5. ๋ฐ์ดํ„ฐ: ๋ชจ๋“  ์ •๋ณด์˜ ์ถœ์ฒ˜ ๋ช…์‹œ 9.6. ๊ฐ€๋…์„ฑ: ๋ช…ํ™•ํ•œ ๋‹จ๋ฝ ๊ตฌ๋ถ„๊ณผ ๊ฐ•์กฐ์  ์‚ฌ์šฉ
669
-
670
- ์ด๋Ÿฌํ•œ ํ”„๋ ˆ์ž„์›Œํฌ๋ฅผ ๋ฐ”ํƒ•์œผ๋กœ, ์š”์ฒญ๋ฐ›์€ ์ฃผ์ œ์— ๋Œ€ํ•ด ์ฒด๊ณ„์ ์ด๊ณ  ๋งค๋ ฅ์ ์ธ ๋ธ”๋กœ๊ทธ ํฌ์ŠคํŠธ๋ฅผ ์ž‘์„ฑํ•˜๊ฒ ์Šต๋‹ˆ๋‹ค.
671
- """
672
-
673
- def test_image_api_connection():
674
- """์ด๋ฏธ์ง€ API ์„œ๋ฒ„ ์—ฐ๊ฒฐ ํ…Œ์ŠคํŠธ"""
675
- try:
676
- client = Client(IMAGE_API_URL)
677
- return "์ด๋ฏธ์ง€ API ์—ฐ๊ฒฐ ์„ฑ๊ณต: ์ •์ƒ ์ž‘๋™ ์ค‘"
678
- except Exception as e:
679
- logging.error(f"์ด๋ฏธ์ง€ API ์—ฐ๊ฒฐ ํ…Œ์ŠคํŠธ ์‹คํŒจ: {e}")
680
- return f"์ด๋ฏธ์ง€ API ์—ฐ๊ฒฐ ์‹คํŒจ: {e}"
681
-
682
- def generate_image(prompt, width=768, height=768, guidance=3.5, inference_steps=30, seed=3):
683
- """์ด๋ฏธ์ง€ ์ƒ์„ฑ ํ•จ์ˆ˜"""
684
- if not prompt:
685
- return None, "์˜ค๋ฅ˜: ํ”„๋กฌํ”„ํŠธ๋ฅผ ์ž…๋ ฅํ•ด์ฃผ์„ธ์š”"
686
-
687
- try:
688
- client = Client(IMAGE_API_URL)
689
- result = client.predict(
690
- prompt=prompt,
691
- width=int(width),
692
- height=int(height),
693
- guidance=float(guidance),
694
- inference_steps=int(inference_steps),
695
- seed=int(seed),
696
- do_img2img=False,
697
- init_image=None,
698
- image2image_strength=0.8,
699
- resize_img=True,
700
- api_name="/generate_image"
701
- )
702
- logging.info(f"์ด๋ฏธ์ง€ ์ƒ์„ฑ ์„ฑ๊ณต: {result[1]}")
703
- return result[0], f"์‚ฌ์šฉ๋œ ์‹œ๋“œ: {result[1]}"
704
- except Exception as e:
705
- logging.error(f"์ด๋ฏธ์ง€ ์ƒ์„ฑ ์‹คํŒจ: {str(e)}")
706
- return None, f"์˜ค๋ฅ˜: {str(e)}"
707
-
708
- def extract_image_prompt(blog_content, blog_topic):
709
- """๋ธ”๋กœ๊ทธ ๋‚ด์šฉ์—์„œ ์ด๋ฏธ์ง€ ์ƒ์„ฑ์„ ์œ„ํ•œ ํ”„๋กฌํ”„ํŠธ ์ถ”์ถœ"""
710
- image_prompt_system = f"""
711
- ๋‹ค์Œ์€ '{blog_topic}'์— ๊ด€ํ•œ ๋ธ”๋กœ๊ทธ ๊ธ€์ž…๋‹ˆ๋‹ค. ์ด ๋ธ”๋กœ๊ทธ ๊ธ€์˜ ๋‚ด์šฉ์„ ๊ธฐ๋ฐ˜์œผ๋กœ ์ ์ ˆํ•œ ์ด๋ฏธ์ง€๋ฅผ ์ƒ์„ฑํ•˜๊ธฐ ์œ„ํ•œ
712
- ํ”„๋กฌํ”„ํŠธ๋ฅผ ์ž‘์„ฑํ•ด์ฃผ์„ธ์š”. ํ”„๋กฌํ”„ํŠธ๋Š” ์˜์–ด๋กœ ์ž‘์„ฑํ•˜๊ณ , ๊ตฌ์ฒด์ ์ธ ์‹œ๊ฐ์  ์š”์†Œ๋ฅผ ๋‹ด์•„์•ผ ํ•ฉ๋‹ˆ๋‹ค.
713
- ํ”„๋กฌํ”„ํŠธ๋งŒ ๋ฐ˜ํ™˜ํ•˜์„ธ์š”(๋‹ค๋ฅธ ์„ค๋ช… ์—†์ด).
714
-
715
- ์˜ˆ์‹œ ํ˜•์‹:
716
- "A professional photo of [subject], [specific details], [atmosphere], [lighting], [perspective], high quality, detailed"
717
- """
718
-
719
- try:
720
- response = client.messages.create(
721
- model="claude-3-7-sonnet-20250219",
722
- max_tokens=150,
723
- system=image_prompt_system,
724
- messages=[{"role": "user", "content": blog_content}]
725
- )
726
-
727
- # ์‘๋‹ต์—์„œ ํ”„๋กฌํ”„ํŠธ ์ถ”์ถœ
728
- image_prompt = response.content[0].text.strip()
729
- logging.info(f"์ƒ์„ฑ๋œ ์ด๋ฏธ์ง€ ํ”„๋กฌํ”„ํŠธ: {image_prompt}")
730
- return image_prompt
731
- except Exception as e:
732
- logging.error(f"์ด๋ฏธ์ง€ ํ”„๋กฌํ”„ํŠธ ์ƒ์„ฑ ์˜ค๋ฅ˜: {e}")
733
- return f"A professional photo related to {blog_topic}, detailed, high quality"
734
-
735
- # ๋งˆํฌ๋‹ค์šด์„ HTML๋กœ ๋ณ€ํ™˜ํ•˜๋Š” ํ•จ์ˆ˜
736
- def convert_md_to_html(md_text, title="Ginigen Blog"):
737
- html_content = markdown.markdown(md_text)
738
- html_doc = f"""
739
- <!DOCTYPE html>
740
- <html>
741
- <head>
742
- <title>{title}</title>
743
- <meta charset="utf-8">
744
- <style>
745
- body {{ font-family: Arial, sans-serif; line-height: 1.6; max-width: 800px; margin: 0 auto; padding: 20px; }}
746
- h1 {{ color: #2c3e50; font-size: 2.5em; margin-bottom: 20px; }}
747
- h2 {{ color: #3498db; margin-top: 25px; font-size: 1.8em; }}
748
- h3 {{ color: #2980b9; font-size: 1.5em; }}
749
- p {{ margin-bottom: 15px; font-size: 1.1em; }}
750
- blockquote {{ background: #f9f9f9; border-left: 10px solid #ccc; margin: 1.5em 10px; padding: 1em 10px; }}
751
- ul, ol {{ margin-bottom: 15px; }}
752
- li {{ margin-bottom: 5px; }}
753
- hr {{ border: 0; height: 1px; background: #ddd; margin: 20px 0; }}
754
- img {{ max-width: 100%; height: auto; display: block; margin: 20px auto; }}
755
- </style>
756
- </head>
757
- <body>
758
- {html_content}
759
- </body>
760
- </html>
761
- """
762
- return html_doc
763
-
764
- # ์›น ๊ฒ€์ƒ‰ ํ‚ค์›Œ๋“œ ์ถ”์ถœ ํ•จ์ˆ˜
765
- def extract_keywords(text: str, top_k: int = 5) -> str:
766
- """
767
- 1) ํ•œ๊ธ€(๊ฐ€-ํžฃ), ์˜์–ด(a-zA-Z), ์ˆซ์ž(0-9), ๊ณต๋ฐฑ๋งŒ ๋‚จ๊น€
768
- 2) ๊ณต๋ฐฑ ๊ธฐ์ค€ ํ† ํฐ ๋ถ„๋ฆฌ
769
- 3) ์ตœ๋Œ€ top_k๊ฐœ๋งŒ
770
- """
771
- text = re.sub(r"[^a-zA-Z0-9๊ฐ€-ํžฃ\s]", "", text)
772
- tokens = text.split()
773
- key_tokens = tokens[:top_k]
774
- return " ".join(key_tokens)
775
-
776
- # Mock ๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ ์ƒ์„ฑ ํ•จ์ˆ˜
777
- def generate_mock_search_results(query):
778
- """API ์—ฐ๊ฒฐ์ด ์•ˆ๋  ๋•Œ ์‚ฌ์šฉํ•  ๊ฐ€์ƒ ๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ ์ƒ์„ฑ"""
779
- current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
780
- mock_results = [
781
- {
782
- "title": f"{query}์— ๊ด€ํ•œ ์ตœ์‹  ์ •๋ณด",
783
- "link": "https://example.com/article1",
784
- "snippet": f"{query}์— ๊ด€ํ•œ ๊ฐ€์ƒ ๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ์ž…๋‹ˆ๋‹ค. ์ด ๊ฒฐ๊ณผ๋Š” API ์—ฐ๊ฒฐ ๋ฌธ์ œ๋กœ ์ธํ•ด ์ƒ์„ฑ๋œ ๊ฐ€์ƒ ๋ฐ์ดํ„ฐ์ž…๋‹ˆ๋‹ค. ์‹ค์ œ ๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ๊ฐ€ ์•„๋‹˜์„ ์ฐธ๊ณ ํ•˜์„ธ์š”. ์ƒ์„ฑ ์‹œ๊ฐ„: {current_time}",
785
- "displayed_link": "example.com/article1"
786
- },
787
- {
788
- "title": f"{query} ๊ด€๋ จ ์—ฐ๊ตฌ ๋™ํ–ฅ",
789
- "link": "https://example.org/research",
790
- "snippet": "์ด๊ฒƒ์€ API ์—ฐ๊ฒฐ ๋ฌธ์ œ๋กœ ์ธํ•œ ๊ฐ€์ƒ ๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ์ž…๋‹ˆ๋‹ค. ์‹ค์ œ ๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ๋ฅผ ๋ณด์—ฌ๋“œ๋ฆฌ์ง€ ๋ชปํ•ด ์ฃ„์†กํ•ฉ๋‹ˆ๋‹ค. ๋Œ€์‹  AI์˜ ๊ธฐ์กด ์ง€์‹์„ ํ™œ์šฉํ•˜์—ฌ ๋‹ต๋ณ€๋“œ๋ฆฌ๊ฒ ์Šต๋‹ˆ๋‹ค.",
791
- "displayed_link": "example.org/research"
792
- },
793
- {
794
- "title": f"{query}์˜ ์—ญ์‚ฌ์  ๋ฐฐ๊ฒฝ",
795
- "link": "https://example.net/history",
796
- "snippet": "์ด ๊ฐ€์ƒ ๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ๋Š” API ์—ฐ๊ฒฐ ๋ฌธ์ œ๋กœ ์ธํ•ด ์ƒ์„ฑ๋˜์—ˆ์Šต๋‹ˆ๋‹ค. ์ฐธ๊ณ ์šฉ์œผ๋กœ๋งŒ ์‚ฌ์šฉํ•ด์ฃผ์„ธ์š”.",
797
- "displayed_link": "example.net/history"
798
- }
799
- ]
800
-
801
- summary_lines = []
802
- for idx, item in enumerate(mock_results, start=1):
803
- title = item.get("title", "No title")
804
- link = item.get("link", "#")
805
- snippet = item.get("snippet", "No description")
806
- displayed_link = item.get("displayed_link", link)
807
-
808
- summary_lines.append(
809
- f"### Result {idx}: {title}\n\n"
810
- f"{snippet}\n\n"
811
- f"**์ถœ์ฒ˜**: [{displayed_link}]({link})\n\n"
812
- f"---\n"
813
- )
814
-
815
- notice = """
816
- # ๊ฐ€์ƒ ๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ (API ์—ฐ๊ฒฐ ๋ฌธ์ œ๋กœ ์ธํ•ด ์ƒ์„ฑ๋จ)
817
- ์•„๋ž˜๋Š” API ์—ฐ๊ฒฐ ๋ฌธ์ œ๋กœ ์ธํ•ด ์ƒ์„ฑ๋œ ๊ฐ€์ƒ ๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ์ž…๋‹ˆ๋‹ค. ์‹ค์ œ ๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ๊ฐ€ ์•„๋‹˜์„ ์ฐธ๊ณ ํ•˜์„ธ์š”.
818
- ๋Œ€์‹  AI์˜ ๊ธฐ์กด ์ง€์‹์„ ํ™œ์šฉํ•˜์—ฌ ์ตœ๋Œ€ํ•œ ์ •ํ™•ํ•œ ๋‹ต๋ณ€์„ ๋“œ๋ฆฌ๊ฒ ์Šต๋‹ˆ๋‹ค.
819
- """
820
-
821
- return notice + "\n".join(summary_lines)
822
-
823
- # Google ๊ฒ€์ƒ‰ ํ•จ์ˆ˜ (SerpAPI ๋Œ€์‹  ์ง์ ‘ ๊ฒ€์ƒ‰)
824
- def do_google_search(query, num_results=5):
825
- try:
826
- # ๊ธฐ๋ณธ ํ—ค๋” ์„ค์ • (๋ธŒ๋ผ์šฐ์ €์ฒ˜๋Ÿผ ๋ณด์ด๊ฒŒ)
827
- headers = {
828
- 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
829
- }
830
-
831
- # ๊ฒ€์ƒ‰ URL ๋งŒ๋“ค๊ธฐ
832
- search_url = f"https://www.google.com/search?q={query}&num={num_results}"
833
- logging.info(f"๊ตฌ๊ธ€ ๊ฒ€์ƒ‰ URL: {search_url}")
834
-
835
- # ์š”์ฒญ ๋ณด๋‚ด๊ธฐ (์งง์€ ํƒ€์ž„์•„์›ƒ ์„ค์ •)
836
- response = requests.get(search_url, headers=headers, timeout=10)
837
-
838
- # ์‘๋‹ต์ด ์„ฑ๊ณต์ ์ธ์ง€ ํ™•์ธ
839
- if response.status_code != 200:
840
- logging.error(f"Google ๊ฒ€์ƒ‰ ์‘๋‹ต ์ƒํƒœ ์ฝ”๋“œ: {response.status_code}")
841
- return generate_mock_search_results(query)
842
-
843
- # ์—ฌ๊ธฐ์„œ๋Š” HTML ๋ถ„์„์ด ๋ณต์žกํ•˜๋ฏ€๋กœ, ์‹ค์ œ๋กœ๋Š” BeautifulSoup ๊ฐ™์€ ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ๋ฅผ ์‚ฌ์šฉํ•˜๋Š” ๊ฒƒ์ด ์ข‹์Œ
844
- # ๊ฐ„๋‹จํ•œ ๋ฐ๋ชจ ๋ชฉ์ ์œผ๋กœ ๋Œ€์‹  Mock ๋ฐ์ดํ„ฐ ๋ฐ˜ํ™˜
845
- logging.info("Google ๊ฒ€์ƒ‰ ์„ฑ๊ณตํ–ˆ์œผ๋‚˜ ํŒŒ์‹ฑ ์ œํ•œ์œผ๋กœ Mock ๋ฐ์ดํ„ฐ ๋ฐ˜ํ™˜")
846
- return generate_mock_search_results(query)
847
-
848
- except Exception as e:
849
- logging.error(f"Google ๊ฒ€์ƒ‰ ์‹คํŒจ: {e}")
850
- return generate_mock_search_results(query)
851
-
852
- # ์›น ๊ฒ€์ƒ‰ ํ•จ์ˆ˜
853
- def do_web_search(query: str) -> str:
854
- """
855
- ์›น ๊ฒ€์ƒ‰์„ ์ˆ˜ํ–‰ํ•˜๋Š” ํ•จ์ˆ˜ - SerpHouse API ๋˜๋Š” ์ง์ ‘ ๊ตฌ๊ธ€ ๊ฒ€์ƒ‰
856
- """
857
- try:
858
- # API ํ‚ค๊ฐ€ ์—†๊ฑฐ๋‚˜ 'mock'์ธ ๊ฒฝ์šฐ
859
- if not SERPHOUSE_API_KEY or "mock" in SERPHOUSE_API_KEY.lower():
860
- logging.warning("API ํ‚ค๊ฐ€ ์—†๊ฑฐ๋‚˜ Mock ๋ชจ๋“œ์ž…๋‹ˆ๋‹ค. ๊ฐ€์ƒ ๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ๋ฅผ ๋ฐ˜ํ™˜ํ•ฉ๋‹ˆ๋‹ค.")
861
- return generate_mock_search_results(query)
862
-
863
- # SerpHouse API ์‚ฌ์šฉ
864
- url = "https://api.serphouse.com/serp/live"
865
- params = {
866
- "q": query,
867
- "domain": "google.com",
868
- "serp_type": "web",
869
- "device": "desktop",
870
- "lang": "ko", # ํ•œ๊ตญ์–ด ๊ฒฐ๊ณผ
871
- "num": "5" # ๊ฒฐ๊ณผ ์ˆ˜ ์ค„์ž„
872
- }
873
-
874
- headers = {
875
- "Authorization": f"Bearer {SERPHOUSE_API_KEY}"
876
- }
877
-
878
- logging.info(f"SerpHouse API ํ˜ธ์ถœ ์ค‘... ๊ฒ€์ƒ‰์–ด: {query}")
879
-
880
- # ์งง์€ ํƒ€์ž„์•„์›ƒ์œผ๋กœ ์š”์ฒญ ์‹œ๋„
881
- response = requests.get(url, headers=headers, params=params, timeout=15)
882
- response.raise_for_status()
883
-
884
- logging.info(f"SerpHouse API ์‘๋‹ต ์ƒํƒœ ์ฝ”๋“œ: {response.status_code}")
885
- data = response.json()
886
-
887
- # ๋‹ค์–‘ํ•œ ์‘๋‹ต ๊ตฌ์กฐ ์ฒ˜๋ฆฌ
888
- results = data.get("results", {})
889
- organic = None
890
-
891
- # ๊ฐ€๋Šฅํ•œ ์‘๋‹ต ๊ตฌ์กฐ 1
892
- if isinstance(results, dict) and "organic" in results:
893
- organic = results["organic"]
894
- # ๊ฐ€๋Šฅํ•œ ์‘๋‹ต ๊ตฌ์กฐ 2
895
- elif isinstance(results, dict) and "results" in results:
896
- if isinstance(results["results"], dict) and "organic" in results["results"]:
897
- organic = results["results"]["organic"]
898
- # ๊ฐ€๋Šฅํ•œ ์‘๋‹ต ๊ตฌ์กฐ 3
899
- elif "organic" in data:
900
- organic = data["organic"]
901
-
902
- if not organic:
903
- logging.warning("์‘๋‹ต์—์„œ organic ๊ฒฐ๊ณผ๋ฅผ ์ฐพ์„ ์ˆ˜ ์—†์Šต๋‹ˆ๋‹ค. ๊ตฌ๊ธ€ ์ง์ ‘ ๊ฒ€์ƒ‰์œผ๋กœ ์ „ํ™˜ํ•ฉ๋‹ˆ๋‹ค.")
904
- return do_google_search(query)
905
-
906
- # ๊ฒฐ๊ณผ ์ˆ˜ ์ œํ•œ ๋ฐ ์ปจํ…์ŠคํŠธ ๊ธธ์ด ์ตœ์ ํ™”
907
- max_results = min(5, len(organic))
908
- limited_organic = organic[:max_results]
909
-
910
- # ๊ฒฐ๊ณผ ํ˜•์‹ ๊ฐœ์„ 
911
- summary_lines = []
912
- for idx, item in enumerate(limited_organic, start=1):
913
- title = item.get("title", "No title")
914
- link = item.get("link", "#")
915
- snippet = item.get("snippet", "No description")
916
- displayed_link = item.get("displayed_link", link)
917
-
918
- summary_lines.append(
919
- f"### Result {idx}: {title}\n\n"
920
- f"{snippet}\n\n"
921
- f"**์ถœ์ฒ˜**: [{displayed_link}]({link})\n\n"
922
- f"---\n"
923
- )
924
-
925
- # ๋ชจ๋ธ์—๊ฒŒ ๋ช…ํ™•ํ•œ ์ง€์นจ ์ถ”๊ฐ€
926
- instructions = """
927
- # ์›น ๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ
928
- ์•„๋ž˜๋Š” ๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ์ž…๋‹ˆ๋‹ค. ์งˆ๋ฌธ์— ๋‹ต๋ณ€ํ•  ๋•Œ ์ด ์ •๋ณด๋ฅผ ํ™œ์šฉํ•˜์„ธ์š”:
929
- 1. ๊ฐ ๊ฒฐ๊ณผ์˜ ์ œ๋ชฉ, ๋‚ด์šฉ, ์ถœ์ฒ˜ ๋งํฌ๋ฅผ ์ฐธ๊ณ ํ•˜์„ธ์š”
930
- 2. ๋‹ต๋ณ€์— ๊ด€๋ จ ์ •๋ณด์˜ ์ถœ์ฒ˜๋ฅผ ๋ช…์‹œ์ ์œผ๋กœ ์ธ์šฉํ•˜์„ธ์š” (์˜ˆ: "X ์ถœ์ฒ˜์— ๋”ฐ๋ฅด๋ฉด...")
931
- 3. ์‘๋‹ต์— ์‹ค์ œ ์ถœ์ฒ˜ ๋งํฌ๋ฅผ ํฌํ•จํ•˜์„ธ์š”
932
- 4. ์—ฌ๋Ÿฌ ์ถœ์ฒ˜์˜ ์ •๋ณด๋ฅผ ์ข…ํ•ฉํ•˜์—ฌ ๋‹ต๋ณ€ํ•˜์„ธ์š”
933
- """
934
-
935
- search_results = instructions + "\n".join(summary_lines)
936
- logging.info(f"๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ {len(limited_organic)}๊ฐœ ์ฒ˜๋ฆฌ ์™„๋ฃŒ")
937
- return search_results
938
-
939
- except requests.exceptions.Timeout:
940
- logging.error("Web search timed out, ์ง์ ‘ ๊ตฌ๊ธ€ ๊ฒ€์ƒ‰์œผ๋กœ ์ „ํ™˜ํ•ฉ๋‹ˆ๋‹ค.")
941
- return do_google_search(query)
942
- except Exception as e:
943
- logging.error(f"Web search failed: {e}, ์ง์ ‘ ๊ตฌ๊ธ€ ๊ฒ€์ƒ‰์œผ๋กœ ์ „ํ™˜ํ•ฉ๋‹ˆ๋‹ค.")
944
- return do_google_search(query)
945
-
946
- def chatbot_interface():
947
- st.title("Ginigen Blog")
948
-
949
- # ๋ชจ๋ธ ๊ณ ์ • ์„ค์ •
950
- if "ai_model" not in st.session_state:
951
- st.session_state["ai_model"] = "claude-3-7-sonnet-20250219"
952
-
953
- # ์„ธ์…˜ ์ƒํƒœ ์ดˆ๊ธฐํ™”
954
- if "messages" not in st.session_state:
955
- st.session_state.messages = []
956
-
957
- # ์ž๋™ ์ €์žฅ ๊ธฐ๋Šฅ
958
- if "auto_save" not in st.session_state:
959
- st.session_state.auto_save = True
960
-
961
- # ์ด๋ฏธ์ง€ ์ƒ์„ฑ ํ† ๊ธ€
962
- if "generate_image" not in st.session_state:
963
- st.session_state.generate_image = False
964
-
965
- # ์›น ๊ฒ€์ƒ‰ ํ† ๊ธ€
966
- if "use_web_search" not in st.session_state:
967
- st.session_state.use_web_search = False
968
-
969
- # ์ด๋ฏธ์ง€ API ์ƒํƒœ
970
- if "image_api_status" not in st.session_state:
971
- st.session_state.image_api_status = test_image_api_connection()
972
-
973
- # ๋Œ€ํ™” ๊ธฐ๋ก ๊ด€๋ฆฌ (์‚ฌ์ด๋“œ๋ฐ”)
974
- st.sidebar.title("๋Œ€ํ™” ๊ธฐ๋ก ๊ด€๋ฆฌ")
975
-
976
- # ์ž๋™ ์ €์žฅ ํ† ๊ธ€
977
- st.session_state.auto_save = st.sidebar.toggle("์ž๋™ ์ €์žฅ", value=st.session_state.auto_save)
978
-
979
- # ์ด๋ฏธ์ง€ ์ƒ์„ฑ ํ† ๊ธ€
980
- st.session_state.generate_image = st.sidebar.toggle("๋ธ”๋กœ๊ทธ ๊ธ€ ์ž‘์„ฑ ํ›„ ์ด๋ฏธ์ง€ ์ž๋™ ์ƒ์„ฑ", value=st.session_state.generate_image)
981
-
982
- # ์›น ๊ฒ€์ƒ‰ ํ† ๊ธ€
983
- st.session_state.use_web_search = st.sidebar.toggle("์ฃผ์ œ ์›น ๊ฒ€์ƒ‰ ๋ฐ ๋ถ„์„", value=st.session_state.use_web_search)
984
-
985
- # ์ด๋ฏธ์ง€ API ์ƒํƒœ ํ‘œ์‹œ
986
- st.sidebar.text(st.session_state.image_api_status)
987
-
988
- # ์ด๋ฏธ์ง€ ์ƒ์„ฑ ์„ค์ • (ํ† ๊ธ€์ด ์ผœ์ ธ ์žˆ์„ ๋•Œ๋งŒ ํ‘œ์‹œ)
989
- if st.session_state.generate_image:
990
- st.sidebar.subheader("์ด๋ฏธ์ง€ ์ƒ์„ฑ ์„ค์ •")
991
- width = st.sidebar.slider("๋„ˆ๋น„", 256, 1024, 768, 64)
992
- height = st.sidebar.slider("๋†’์ด", 256, 1024, 768, 64)
993
- guidance = st.sidebar.slider("๊ฐ€์ด๋˜์Šค ์Šค์ผ€์ผ", 1.0, 20.0, 3.5, 0.1)
994
- inference_steps = st.sidebar.slider("์ธํผ๋Ÿฐ์Šค ์Šคํ…", 1, 50, 30, 1)
995
- seed = st.sidebar.number_input("์‹œ๋“œ", value=3, min_value=0, step=1)
996
- else:
997
- # ๊ธฐ๋ณธ๊ฐ’ ์„ค์ •
998
- width, height, guidance, inference_steps, seed = 768, 768, 3.5, 30, 3
999
-
1000
- # ๋ธ”๋กœ๊ทธ ๋‚ด์šฉ ๋‹ค์šด๋กœ๋“œ ์„น์…˜
1001
- st.sidebar.title("๋ธ”๋กœ๊ทธ ๋‹ค์šด๋กœ๋“œ")
1002
-
1003
- # ์ตœ์‹  ๋ธ”๋กœ๊ทธ ๋‚ด์šฉ ๊ฐ€์ ธ์˜ค๊ธฐ
1004
- latest_blog = None
1005
- latest_blog_title = "๋ธ”๋กœ๊ทธ ๊ธ€"
1006
-
1007
- if len(st.session_state.messages) > 0:
1008
- # ๊ฐ€์žฅ ์ตœ๊ทผ assistant ๋ฉ”์‹œ์ง€ ์ฐพ๊ธฐ
1009
- for msg in reversed(st.session_state.messages):
1010
- if msg["role"] == "assistant" and msg["content"].strip():
1011
- latest_blog = msg["content"]
1012
-
1013
- # ํƒ€์ดํ‹€ ์ถ”์ถœ ์‹œ๋„ (์ฒซ ๋ฒˆ์งธ ์ œ๋ชฉ ํƒœ๊ทธ ์‚ฌ์šฉ)
1014
- title_match = re.search(r'# (.*?)(\n|$)', latest_blog)
1015
- if title_match:
1016
- latest_blog_title = title_match.group(1).strip()
1017
- # ์‚ฌ์šฉ์ž ์ž…๋ ฅ์„ ํƒ€์ดํ‹€๋กœ ์‚ฌ์šฉ
1018
- elif len(st.session_state.messages) >= 2:
1019
- for i in range(len(st.session_state.messages)-1, -1, -1):
1020
- if st.session_state.messages[i]["role"] == "user":
1021
- latest_blog_title = st.session_state.messages[i]["content"][:30].strip()
1022
- if len(st.session_state.messages[i]["content"]) > 30:
1023
- latest_blog_title += "..."
1024
- break
1025
- break
1026
-
1027
- # ๋‹ค์šด๋กœ๋“œ ๋ฒ„ํŠผ ๊ทธ๋ฃน
1028
- if latest_blog:
1029
- st.sidebar.subheader("์ตœ๊ทผ ๋ธ”๋กœ๊ทธ ๋‹ค์šด๋กœ๋“œ")
1030
-
1031
- col1, col2 = st.sidebar.columns(2)
1032
-
1033
- # ๋งˆํฌ๋‹ค์šด์œผ๋กœ ๋‹ค์šด๋กœ๋“œ
1034
- with col1:
1035
- st.download_button(
1036
- label="๋งˆํฌ๋‹ค์šด",
1037
- data=latest_blog,
1038
- file_name=f"{latest_blog_title}.md",
1039
- mime="text/markdown"
1040
- )
1041
-
1042
- # HTML๋กœ ๋‹ค์šด๋กœ๋“œ
1043
- with col2:
1044
- html_content = convert_md_to_html(latest_blog, latest_blog_title)
1045
- st.download_button(
1046
- label="HTML",
1047
- data=html_content,
1048
- file_name=f"{latest_blog_title}.html",
1049
- mime="text/html"
1050
- )
1051
-
1052
- # ๋Œ€ํ™” ๊ธฐ๋ก ๋ถˆ๋Ÿฌ์˜ค๊ธฐ
1053
- uploaded_file = st.sidebar.file_uploader("๋Œ€ํ™” ๊ธฐ๋ก ๋ถˆ๋Ÿฌ์˜ค๊ธฐ", type=['json'])
1054
- if uploaded_file is not None:
1055
- try:
1056
- content = uploaded_file.getvalue().decode()
1057
- if content.strip():
1058
- st.session_state.messages = json.loads(content)
1059
- st.sidebar.success("๋Œ€ํ™” ๊ธฐ๋ก์„ ์„ฑ๊ณต์ ์œผ๋กœ ๋ถˆ๋Ÿฌ์™”์Šต๋‹ˆ๋‹ค!")
1060
- else:
1061
- st.sidebar.warning("์—…๋กœ๋“œ๋œ ํŒŒ์ผ์ด ๋น„์–ด ์žˆ์Šต๋‹ˆ๋‹ค.")
1062
- except json.JSONDecodeError:
1063
- st.sidebar.error("์˜ฌ๋ฐ”๋ฅธ JSON ํ˜•์‹์˜ ํŒŒ์ผ์ด ์•„๋‹™๋‹ˆ๋‹ค.")
1064
- except Exception as e:
1065
- st.sidebar.error(f"ํŒŒ์ผ ์ฒ˜๋ฆฌ ์ค‘ ์˜ค๋ฅ˜๊ฐ€ ๋ฐœ์ƒํ–ˆ์Šต๋‹ˆ๋‹ค: {str(e)}")
1066
-
1067
- # ๋Œ€ํ™” ๊ธฐ๋ก ์ดˆ๊ธฐํ™” ๋ฒ„ํŠผ
1068
- if st.sidebar.button("๋Œ€ํ™” ๊ธฐ๋ก ์ดˆ๊ธฐํ™”"):
1069
- st.session_state.messages = []
1070
- st.sidebar.success("๋Œ€ํ™” ๊ธฐ๋ก์ด ์ดˆ๊ธฐํ™”๋˜์—ˆ์Šต๋‹ˆ๋‹ค.")
1071
-
1072
- # ๋ฉ”์‹œ์ง€ ํ‘œ์‹œ
1073
- for message in st.session_state.messages:
1074
- with st.chat_message(message["role"]):
1075
- st.markdown(message["content"])
1076
- # ์ด๋ฏธ์ง€๊ฐ€ ์žˆ๋Š” ๊ฒฝ์šฐ ํ‘œ์‹œ
1077
- if "image" in message:
1078
- st.image(message["image"], caption=message.get("image_caption", "์ƒ์„ฑ๋œ ์ด๋ฏธ์ง€"))
1079
-
1080
- # ์‚ฌ์šฉ์ž ์ž…๋ ฅ
1081
- if prompt := st.chat_input("๋ฌด์—‡์„ ๋„์™€๋“œ๋ฆด๊นŒ์š”?"):
1082
- st.session_state.messages.append({"role": "user", "content": prompt})
1083
- with st.chat_message("user"):
1084
- st.markdown(prompt)
1085
-
1086
- # AI ์‘๋‹ต ์ƒ์„ฑ
1087
- with st.chat_message("assistant"):
1088
- message_placeholder = st.empty()
1089
- full_response = ""
1090
-
1091
- # ์›น ๊ฒ€์ƒ‰ ์ˆ˜ํ–‰ (์›น ๊ฒ€์ƒ‰ ์˜ต์…˜์ด ์ผœ์ ธ ์žˆ์„ ๊ฒฝ์šฐ)
1092
- system_prompt = get_system_prompt()
1093
- if st.session_state.use_web_search:
1094
- with st.spinner("์›น์—์„œ ๊ด€๋ จ ์ •๋ณด๋ฅผ ๊ฒ€์ƒ‰ ์ค‘..."):
1095
- try:
1096
- search_query = extract_keywords(prompt, top_k=5)
1097
- st.info(f"๊ฒ€์ƒ‰์–ด: {search_query}")
1098
-
1099
- # ๋‘ ๊ฐ€์ง€ ๋ฐฉ๋ฒ• ๋ชจ๋‘ ์‹œ๋„ (SerpHouse API์™€ ์ง์ ‘ ๊ฒ€์ƒ‰)
1100
- search_results = do_web_search(search_query)
1101
-
1102
- if "๊ฐ€์ƒ ๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ" in search_results:
1103
- st.warning("์‹ค์ œ ๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ๋ฅผ ๊ฐ€์ ธ์˜ฌ ์ˆ˜ ์—†์–ด ๊ธฐ์กด ์ง€์‹์„ ํ™œ์šฉํ•ฉ๋‹ˆ๋‹ค.")
1104
- else:
1105
- st.success(f"๊ฒ€์ƒ‰ ์™„๋ฃŒ: '{search_query}'์— ๋Œ€ํ•œ ์ •๋ณด๋ฅผ ์ˆ˜์ง‘ํ–ˆ์Šต๋‹ˆ๋‹ค.")
1106
-
1107
- # ์‹œ์Šคํ…œ ํ”„๋กฌํ”„ํŠธ์— ๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ ์ถ”๊ฐ€
1108
- system_prompt += f"\n\n๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ:\n{search_results}\n"
1109
- except Exception as e:
1110
- st.error(f"์›น ๊ฒ€์ƒ‰ ์ค‘ ์˜ค๋ฅ˜๊ฐ€ ๋ฐœ์ƒํ–ˆ์Šต๋‹ˆ๋‹ค: {str(e)}")
1111
- logging.error(f"์›น ๊ฒ€์ƒ‰ ์˜ค๋ฅ˜: {str(e)}")
1112
- system_prompt += "\n\n์›น ๊ฒ€์ƒ‰์ด ์‹คํŒจํ–ˆ์Šต๋‹ˆ๋‹ค. ๊ธฐ์กด ์ง€์‹์„ ๋ฐ”ํƒ•์œผ๋กœ ๋‹ต๋ณ€ํ•˜์„ธ์š”."
1113
-
1114
- # API ํ˜ธ์ถœ
1115
- with client.messages.stream(
1116
- max_tokens=MAX_TOKENS,
1117
- system=system_prompt,
1118
- messages=[{"role": m["role"], "content": m["content"]} for m in st.session_state.messages],
1119
- model=st.session_state["ai_model"]
1120
- ) as stream:
1121
- for text in stream.text_stream:
1122
- full_response += str(text) if text is not None else ""
1123
- message_placeholder.markdown(full_response + "โ–Œ")
1124
-
1125
- message_placeholder.markdown(full_response)
1126
-
1127
- # ์ด๋ฏธ์ง€ ์ƒ์„ฑ ์˜ต์…˜์ด ์ผœ์ ธ ์žˆ๋Š” ๊ฒฝ์šฐ
1128
- if st.session_state.generate_image:
1129
- with st.spinner("๋ธ”๋กœ๊ทธ์— ๋งž๋Š” ์ด๋ฏธ์ง€ ์ƒ์„ฑ ์ค‘..."):
1130
- # ์ด๋ฏธ์ง€ ํ”„๋กฌํ”„ํŠธ ์ƒ์„ฑ
1131
- image_prompt = extract_image_prompt(full_response, prompt)
1132
-
1133
- # ์ด๋ฏธ์ง€ ์ƒ์„ฑ
1134
- image, image_caption = generate_image(
1135
- image_prompt,
1136
- width=width,
1137
- height=height,
1138
- guidance=guidance,
1139
- inference_steps=inference_steps,
1140
- seed=seed
1141
- )
1142
-
1143
- if image:
1144
- st.image(image, caption=image_caption)
1145
- # ์ด๋ฏธ์ง€ ์ •๋ณด๋ฅผ ์‘๋‹ต์— ํฌํ•จ
1146
- st.session_state.messages.append({
1147
- "role": "assistant",
1148
- "content": full_response,
1149
- "image": image,
1150
- "image_caption": image_caption
1151
- })
1152
- else:
1153
- st.error(f"์ด๋ฏธ์ง€ ์ƒ์„ฑ ์‹คํŒจ: {image_caption}")
1154
- st.session_state.messages.append({
1155
- "role": "assistant",
1156
- "content": full_response
1157
- })
1158
- else:
1159
- # ์ด๋ฏธ์ง€ ์ƒ์„ฑ ์—†์ด ์‘๋‹ต๋งŒ ์ €์žฅ
1160
- st.session_state.messages.append({
1161
- "role": "assistant",
1162
- "content": full_response
1163
- })
1164
-
1165
- # ๋ธ”๋กœ๊ทธ ๋‹ค์šด๋กœ๋“œ ๋ฒ„ํŠผ ํ‘œ์‹œ (์‘๋‹ต ๋ฐ”๋กœ ์•„๋ž˜์—)
1166
- st.subheader("์ด ๋ธ”๋กœ๊ทธ ๋‹ค์šด๋กœ๋“œ:")
1167
- col1, col2 = st.columns(2)
1168
-
1169
- with col1:
1170
- st.download_button(
1171
- label="๋งˆํฌ๋‹ค์šด์œผ๋กœ ์ €์žฅ",
1172
- data=full_response,
1173
- file_name=f"{prompt[:30]}.md",
1174
- mime="text/markdown"
1175
- )
1176
-
1177
- with col2:
1178
- html_content = convert_md_to_html(full_response, prompt[:30])
1179
- st.download_button(
1180
- label="HTML๋กœ ์ €์žฅ",
1181
- data=html_content,
1182
- file_name=f"{prompt[:30]}.html",
1183
- mime="text/html"
1184
- )
1185
-
1186
- # ์ž๋™ ์ €์žฅ ๊ธฐ๋Šฅ
1187
- if st.session_state.auto_save:
1188
- try:
1189
- # ์ด๋ฏธ์ง€ ์ •๋ณด๋Š” ์ €์žฅํ•˜์ง€ ์•Š์Œ (JSON์—๋Š” ๋ฐ”์ด๋„ˆ๋ฆฌ ๋ฐ์ดํ„ฐ๋ฅผ ์ง์ ‘ ์ €์žฅํ•  ์ˆ˜ ์—†์Œ)
1190
- save_messages = []
1191
- for msg in st.session_state.messages:
1192
- save_msg = {"role": msg["role"], "content": msg["content"]}
1193
- save_messages.append(save_msg)
1194
-
1195
- # ํ˜„์žฌ ์‹œ๊ฐ„์„ ํฌํ•จํ•œ ํŒŒ์ผ๋ช… ์ƒ์„ฑ
1196
- current_time = datetime.now().strftime("%Y%m%d_%H%M%S")
1197
- filename = f'chat_history_auto_save_{current_time}.json'
1198
-
1199
- with open(filename, 'w', encoding='utf-8') as f:
1200
- json.dump(save_messages, f, ensure_ascii=False, indent=4)
1201
- except Exception as e:
1202
- st.sidebar.error(f"์ž๋™ ์ €์žฅ ์ค‘ ์˜ค๋ฅ˜ ๋ฐœ์ƒ: {str(e)}")
1203
-
1204
- # ๋Œ€ํ™” ๊ธฐ๋ก ๋‹ค์šด๋กœ๋“œ
1205
- if st.sidebar.button("๋Œ€ํ™” ๊ธฐ๋ก ๋‹ค์šด๋กœ๋“œ"):
1206
- # ์ด๋ฏธ์ง€ ์ •๋ณด๋Š” ์ €์žฅํ•˜์ง€ ์•Š์Œ
1207
- save_messages = []
1208
- for msg in st.session_state.messages:
1209
- save_msg = {"role": msg["role"], "content": msg["content"]}
1210
- save_messages.append(save_msg)
1211
-
1212
- json_history = json.dumps(save_messages, indent=4, ensure_ascii=False)
1213
- st.sidebar.download_button(
1214
- label="๋Œ€ํ™” ๊ธฐ๋ก ์ €์žฅํ•˜๊ธฐ",
1215
- data=json_history,
1216
- file_name="chat_history.json",
1217
- mime="application/json"
1218
- )
1219
-
1220
- def main():
1221
- chatbot_interface()
1222
-
1223
  if __name__ == "__main__":
1224
  # requirements.txt ํŒŒ์ผ ์ƒ์„ฑ
1225
  with open("requirements.txt", "w") as f:
 
604
  def main():
605
  chatbot_interface()
606
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
607
  if __name__ == "__main__":
608
  # requirements.txt ํŒŒ์ผ ์ƒ์„ฑ
609
  with open("requirements.txt", "w") as f: