Spaces:

ginipick
/

IDEA-DESIGN

Running

App Files Files Community

ginipick commited on Apr 23

Commit

7046eff

verified ·

1 Parent(s): eaf1509

Update app-BACKUP2.py

Browse files

Files changed (1) hide show

app-BACKUP2.py +163 -0

app-BACKUP2.py CHANGED Viewed

@@ -5,6 +5,10 @@ import anthropic
 import requests
 import logging
 from gradio_client import Client
 # 로깅 설정
 logging.basicConfig(
@@ -108,6 +112,61 @@ def extract_image_prompt(blog_content, blog_topic):
         logging.error(f"이미지 프롬프트 생성 오류: {e}")
         return f"A professional photo related to {blog_topic}, detailed, high quality"
 def chatbot_interface():
     st.title("Ginigen Blog")
@@ -155,6 +214,73 @@ def chatbot_interface():
         # 기본값 설정
         width, height, guidance, inference_steps, seed = 768, 768, 3.5, 30, 3
     # 대화 기록 불러오기
     uploaded_file = st.sidebar.file_uploader("대화 기록 불러오기", type=['json'])
     if uploaded_file is not None:
@@ -244,6 +370,40 @@ def chatbot_interface():
                     "role": "assistant",
                     "content": full_response
                 })
         # 자동 저장 기능
         if st.session_state.auto_save:
@@ -285,5 +445,8 @@ if __name__ == "__main__":
         f.write("anthropic>=0.18.1\n")
         f.write("gradio-client>=1.8.0\n")
         f.write("requests>=2.32.3\n")
     main()

 import requests
 import logging
 from gradio_client import Client
+import markdown
+import tempfile
+import base64
+from weasyprint import HTML
 # 로깅 설정
 logging.basicConfig(
         logging.error(f"이미지 프롬프트 생성 오류: {e}")
         return f"A professional photo related to {blog_topic}, detailed, high quality"
+# 마크다운을 HTML로 변환하는 함수
+def convert_md_to_html(md_text, title="Ginigen Blog"):
+    html_content = markdown.markdown(md_text)
+    html_doc = f"""
+    <!DOCTYPE html>
+    <html>
+    <head>
+        <title>{title}</title>
+        <meta charset="utf-8">
+        <style>
+            body {{ font-family: Arial, sans-serif; line-height: 1.6; max-width: 800px; margin: 0 auto; padding: 20px; }}
+            h1 {{ color: #2c3e50; font-size: 2.5em; margin-bottom: 20px; }}
+            h2 {{ color: #3498db; margin-top: 25px; font-size: 1.8em; }}
+            h3 {{ color: #2980b9; font-size: 1.5em; }}
+            p {{ margin-bottom: 15px; font-size: 1.1em; }}
+            blockquote {{ background: #f9f9f9; border-left: 10px solid #ccc; margin: 1.5em 10px; padding: 1em 10px; }}
+            ul, ol {{ margin-bottom: 15px; }}
+            li {{ margin-bottom: 5px; }}
+            hr {{ border: 0; height: 1px; background: #ddd; margin: 20px 0; }}
+            img {{ max-width: 100%; height: auto; display: block; margin: 20px auto; }}
+        </style>
+    </head>
+    <body>
+        {html_content}
+    </body>
+    </html>
+    """
+    return html_doc
+# 마크다운을 PDF로 변환하는 함수
+def convert_md_to_pdf(md_text, title="Ginigen Blog"):
+    html_content = convert_md_to_html(md_text, title)
+    # 임시 파일 생성
+    with tempfile.NamedTemporaryFile(suffix='.pdf', delete=False) as tmp:
+        tmp_path = tmp.name
+    # HTML을 PDF로 변환
+    HTML(string=html_content).write_pdf(tmp_path)
+    # 생성된 PDF 읽기
+    with open(tmp_path, 'rb') as f:
+        pdf_data = f.read()
+    # 임시 파일 삭제
+    os.unlink(tmp_path)
+    return pdf_data
+# 다운로드 버튼 생성 헬퍼 함수
+def create_download_link(bin_data, download_filename, link_text):
+    b64 = base64.b64encode(bin_data).decode()
+    href = f'<a href="data:application/octet-stream;base64,{b64}" download="{download_filename}">{link_text}</a>'
+    return href
 def chatbot_interface():
     st.title("Ginigen Blog")
         # 기본값 설정
         width, height, guidance, inference_steps, seed = 768, 768, 3.5, 30, 3
+    # 블로그 내용 다운로드 섹션
+    st.sidebar.title("블로그 다운로드")
+    # 최신 블로그 내용 가져오기
+    latest_blog = None
+    latest_blog_title = "블로그 글"
+    if len(st.session_state.messages) > 0:
+        # 가장 최근 assistant 메시지 찾기
+        for msg in reversed(st.session_state.messages):
+            if msg["role"] == "assistant" and msg["content"].strip():
+                latest_blog = msg["content"]
+                # 타이틀 추출 시도 (첫 번째 제목 태그 사용)
+                import re
+                title_match = re.search(r'# (.*?)(\n|$)', latest_blog)
+                if title_match:
+                    latest_blog_title = title_match.group(1).strip()
+                # 사용자 입력을 타이틀로 사용
+                elif len(st.session_state.messages) >= 2:
+                    for i in range(len(st.session_state.messages)-1, -1, -1):
+                        if st.session_state.messages[i]["role"] == "user":
+                            latest_blog_title = st.session_state.messages[i]["content"][:30].strip()
+                            if len(st.session_state.messages[i]["content"]) > 30:
+                                latest_blog_title += "..."
+                            break
+                break
+    # 다운로드 버튼 그룹
+    if latest_blog:
+        st.sidebar.subheader("최근 블로그 다운로드")
+        col1, col2, col3 = st.sidebar.columns(3)
+        # 마크다운으로 다운로드
+        with col1:
+            st.download_button(
+                label="마크다운",
+                data=latest_blog,
+                file_name=f"{latest_blog_title}.md",
+                mime="text/markdown"
+            )
+        # HTML로 다운로드
+        with col2:
+            html_content = convert_md_to_html(latest_blog, latest_blog_title)
+            st.download_button(
+                label="HTML",
+                data=html_content,
+                file_name=f"{latest_blog_title}.html",
+                mime="text/html"
+            )
+        # PDF로 다운로드
+        with col3:
+            try:
+                pdf_data = convert_md_to_pdf(latest_blog, latest_blog_title)
+                st.download_button(
+                    label="PDF",
+                    data=pdf_data,
+                    file_name=f"{latest_blog_title}.pdf",
+                    mime="application/pdf"
+                )
+            except Exception as e:
+                st.error(f"PDF 생성 오류: {e}")
+                logging.error(f"PDF 생성 오류: {e}")
     # 대화 기록 불러오기
     uploaded_file = st.sidebar.file_uploader("대화 기록 불러오기", type=['json'])
     if uploaded_file is not None:
                     "role": "assistant",
                     "content": full_response
                 })
+            # 블로그 다운로드 버튼 표시 (응답 바로 아래에)
+            st.subheader("이 블로그 다운로드:")
+            col1, col2, col3 = st.columns(3)
+            with col1:
+                st.download_button(
+                    label="마크다운으로 저장",
+                    data=full_response,
+                    file_name=f"{prompt[:30]}.md",
+                    mime="text/markdown"
+                )
+            with col2:
+                html_content = convert_md_to_html(full_response, prompt[:30])
+                st.download_button(
+                    label="HTML로 저장",
+                    data=html_content,
+                    file_name=f"{prompt[:30]}.html",
+                    mime="text/html"
+                )
+            with col3:
+                try:
+                    pdf_data = convert_md_to_pdf(full_response, prompt[:30])
+                    st.download_button(
+                        label="PDF로 저장",
+                        data=pdf_data,
+                        file_name=f"{prompt[:30]}.pdf",
+                        mime="application/pdf"
+                    )
+                except Exception as e:
+                    st.error(f"PDF 생성 오류: {e}")
+                    logging.error(f"PDF 생성 오류: {e}")
         # 자동 저장 기능
         if st.session_state.auto_save:
         f.write("anthropic>=0.18.1\n")
         f.write("gradio-client>=1.8.0\n")
         f.write("requests>=2.32.3\n")
+        f.write("markdown>=3.5.1\n")
+        f.write("weasyprint>=60.2\n")
+        f.write("pillow>=10.1.0\n")
     main()