Delete app.py

app.py

@@ -1,660 +0,0 @@
-"""
-EmpathemeBot - Hugging Face Spaces用統合版Streamlitアプリ（セキュア版v2）
-APIキーをセッション単位で管理し、ユーザー間で共有されないようにする
-「新しいチャット」ボタンでAPIキーを維持する改良版
-"""
-
-import html
-import logging
-import re
-import time
-import uuid
-from datetime import datetime
-from typing import List, Dict, Optional
-from pathlib import Path
-import sys
-
-import streamlit as st
-from dotenv import load_dotenv
-import os
-
-# 環境変数の読み込み
-load_dotenv()
-
-# srcディレクトリをパスに追加
-sys.path.append(str(Path(__file__).parent))
-
-# ロギング設定
-logging.basicConfig(
-    level=logging.INFO,
-    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
-)
-logger = logging.getLogger(__name__)
-
-# 設定
-st.set_page_config(
-    page_title="EmpathemeBot QA System",
-    page_icon="",
-    layout="wide",
-    initial_sidebar_state="collapsed",
-    menu_items={}
-)
-
-class SecureQAChain:
-    """APIキーを直接渡すQAChainラッパー"""
-    
-    def __init__(self, api_key: str, persist_dir: str = "data/vector_store", **kwargs):
-        """
-        APIキーを直接受け取ってQAChainを初期化
-        
-        Args:
-            api_key: OpenAI APIキー
-            persist_dir: ベクトルストアのパス
-            **kwargs: その他のQAChain用パラメータ
-        """
-        self.api_key = api_key
-        self.persist_dir = persist_dir
-        self.conversation_history = []
-        self.qa_chain_initialized = False
-        
-        # QAChainの遅延初期化用のパラメータを保存
-        self.chain_params = kwargs
-        
-        # ベクトルストアの存在確認
-        self.vector_store_exists = Path(persist_dir).exists()
-        
-        if not self.vector_store_exists:
-            logger.warning("ベクトルストアが見つかりません。デモモードで動作します。")
-    
-    def _init_qa_chain(self):
-        """実際のQAChainを初期化（遅延初期化）"""
-        if not self.qa_chain_initialized:
-            try:
-                from langchain_openai import ChatOpenAI, OpenAIEmbeddings
-                from langchain_chroma import Chroma
-                
-                if self.vector_store_exists:
-                    # ベクトルストアがある場合
-                    embeddings = OpenAIEmbeddings(
-                        model="text-embedding-3-small",
-                        openai_api_key=self.api_key
-                    )
-                    self.db = Chroma(
-                        persist_directory=self.persist_dir,
-                        embedding_function=embeddings
-                    )
-                    self.retriever = self.db.as_retriever(
-                        search_type='similarity',
-                        search_kwargs={'k': 5}
-                    )
-                
-                # LLMを初期化（APIキーを直接渡す）
-                self.llm = ChatOpenAI(
-                    model_name="gpt-4o-mini",
-                    temperature=0.7,
-                    openai_api_key=self.api_key
-                )
-                
-                self.qa_chain_initialized = True
-                logger.info("QAChain初期化完了")
-                
-            except Exception as e:
-                logger.error(f"QAChain初期化エラー: {e}")
-                raise
-    
-    def ask_with_history(self, question: str):
-        """質問に回答（会話履歴を考慮）"""
-        self._init_qa_chain()  # 遅延初期化
-        
-        try:
-            from langchain.schema import HumanMessage, SystemMessage
-            
-            # 会話履歴を追加
-            self.conversation_history.append(f"Q: {question}")
-            
-            # システムメッセージ
-            messages = [
-                SystemMessage(content="あなたは英語学習をサポートするKurageSan®という親切なアシスタントです。"),
-            ]
-            
-            # ベクトルストアが使える場合はコンテキストを追加
-            if self.vector_store_exists and hasattr(self, 'retriever'):
-                try:
-                    # 関連文書を検索
-                    docs = self.retriever.invoke(question)
-                    if docs:
-                        context = "\n".join([doc.page_content[:500] for doc in docs[:3]])  # 上位3件を使用
-                        messages.append(SystemMessage(content=f"以下の情報を参考にして回答してください：\n{context}"))
-                except Exception as e:
-                    logger.warning(f"文書検索エラー: {e}")
-            
-            # 質問を追加
-            messages.append(HumanMessage(content=question))
-            
-            # 回答を生成
-            response = self.llm.invoke(messages)
-            answer = response.content
-            
-            # 会話履歴���追加
-            self.conversation_history.append(f"A: {answer}")
-            
-            # 文書情報（ダミー）
-            source_docs = []
-            if self.vector_store_exists and hasattr(self, 'retriever'):
-                try:
-                    source_docs = self.retriever.invoke(question)[:3]
-                except:
-                    pass
-            
-            return answer, source_docs
-            
-        except Exception as e:
-            logger.error(f"質問処理エラー: {e}")
-            # エラー時のフォールバック
-            answer = f"申し訳ございません。エラーが発生しました: {str(e)}"
-            self.conversation_history.append(f"A: {answer}")
-            return answer, []
-    
-    def clear_history(self):
-        """会話履歴をクリア"""
-        self.conversation_history = []
-        logger.info("会話履歴をクリアしました")
-    
-    def get_history(self):
-        """会話履歴を取得"""
-        return "\n".join(self.conversation_history)
-
-
-class EmpathemeBotUI:
-    """Hugging Face Spaces用セキュア版EmpathemeBot UIクラス"""
-    
-    def __init__(self):
-        # セッション状態の初期化（APIキーは保存しない）
-        if 'session_id' not in st.session_state:
-            st.session_state.session_id = str(uuid.uuid4())
-        if 'messages' not in st.session_state:
-            st.session_state.messages = []
-        if 'qa_chain' not in st.session_state:
-            st.session_state.qa_chain = None
-        if 'last_activity' not in st.session_state:
-            st.session_state.last_activity = datetime.now()
-        if 'vector_store_initialized' not in st.session_state:
-            st.session_state.vector_store_initialized = False
-        # APIキー関連の状態（セッション単位）
-        if 'current_api_key' not in st.session_state:
-            st.session_state.current_api_key = ""  # 現在のセッションのAPIキー
-            
-    def initialize_qa_chain(self, api_key: str) -> bool:
-        """
-        QAChainを初期化（APIキーを直接渡す）
-        
-        Args:
-            api_key: OpenAI APIキー
-            
-        Returns:
-            初期化成功の場合True
-        """
-        try:
-            # ベクトルストアのパスを確認
-            vector_store_path = Path("data/vector_store")
-            
-            # QAChainの初期化（APIキーを直接渡す）
-            logger.info("QAChainを初期化中...")
-            
-            # SecureQAChainを使用（APIキーを直接渡す）
-            st.session_state.qa_chain = SecureQAChain(
-                api_key=api_key,
-                persist_dir=str(vector_store_path),
-                verbose=False,
-                max_history_turns=10,
-                max_history_chars=10000
-            )
-            
-            st.session_state.vector_store_initialized = True
-            st.session_state.current_api_key = api_key  # 現在のセッションのAPIキーを記録
-            logger.info("QAChain初期化完了")
-            return True
-                
-        except Exception as e:
-            logger.error(f"QAChain初期化エラー: {e}")
-            st.error(f"初期化エラー: {str(e)}")
-            return False
-    
-    def ask_question(self, question: str) -> Optional[Dict]:
-        """
-        質問を処理して回答を取得
-        
-        Args:
-            question: ユーザーの質問
-            
-        Returns:
-            回答データ
-        """
-        try:
-            if st.session_state.qa_chain is None:
-                st.error("システムが初期化されていません。APIキーを入力してください。")
-                return None
-            
-            # 質問処理
-            logger.info(f"質問処理開始: {question[:100]}...")
-            answer, source_docs = st.session_state.qa_chain.ask_with_history(question)
-            
-            # ソースURLを抽出（重複除去）
-            source_urls = []
-            for doc in source_docs:
-                if hasattr(doc, 'metadata'):
-                    url = doc.metadata.get('source_url', '')
-                    if url and url not in source_urls:
-                        source_urls.append(url)
-            
-            result = {
-                "answer": answer,
-                "source_count": len(source_docs),
-                "source_urls": source_urls
-            }
-            
-            logger.info(f"回答生成成功: {len(source_docs)}件のソース参照")
-            return result
-                
-        except Exception as e:
-            logger.error(f"エラー発生: {e}")
-            st.error(f"予期しないエラーが発生しました: {str(e)}")
-            return None
-    
-    def clear_history(self):
-        """会話履歴をクリア"""
-        try:
-            if st.session_state.qa_chain:
-                st.session_state.qa_chain.clear_history()
-            st.session_state.messages = []
-            st.success("会話履歴をクリアしました")
-            logger.info("履歴クリア成功")
-        except Exception as e:
-            logger.error(f"履歴クリアエラー: {e}")
-            st.error("エラーが発生しました")
-    
-    def create_new_session(self):
-        """新しいセッションIDを生成（APIキーは維持）"""
-        st.session_state.session_id = str(uuid.uuid4())
-        st.session_state.messages = []  # メッセージのみクリア
-        
-        # APIキーが設定済みの場合は維持
-        if st.session_state.current_api_key:
-            # 既存のQAChainの会話履歴のみクリア
-            if st.session_state.qa_chain:
-                st.session_state.qa_chain.clear_history()
-            # APIキーとvector_store_initializedは維持
-            logger.info(f"新しいチャット開始（APIキー維持）: {st.session_state.session_id}")
-        else:
-            # APIキーが未設定の場合のみリセット
-            st.session_state.qa_chain = None
-            st.session_state.vector_store_initialized = False
-            logger.info(f"新しいセッション作成: {st.session_state.session_id}")
-
-def main():
-    """メイン関数"""
-    
-    # カスタムCSS
-    st.markdown("""
-        <style>
-        /* メインコンテナのスタイル */
-        .main {
-            padding-top: 1rem;
-            max-width: 1000px;
-            margin: 0 auto;
-        }
-        
-        .block-container {
-            padding: 1rem 2rem;
-            max-width: 100%;
-        }
-        
-        /* チャット入力のスタイル */
-        .stChatInput {
-            border: none !important;
-            box-shadow: none !important;
-            background: transparent !important;
-            position: fixed;
-            bottom: 0;
-            padding-bottom: 1rem;
-            background: white !important;
-            z-index: 999;
-        }
-        
-        /* チャット入力のテキストエリア */
-        .stChatInput textarea {
-            font-size: 14px;
-            border: 1px solid #E5E7EB !important;
-            border-radius: 8px !important;
-            padding: 0.6rem 1rem !important;
-            background: #FAFAFA !important;
-            transition: all 0.2s ease;
-        }
-        
-        .stChatInput textarea:focus {
-            background: white !important;
-            border-color: #4F46E5 !important;
-            outline: none !important;
-            box-shadow: 0 0 0 2px rgba(79, 70, 229, 0.1) !important;
-        }
-        
-        /* ボタンのスタイル */
-        .stButton > button {
-            background: #4F46E5;
-            color: white;
-            border: none;
-            border-radius: 6px;
-            padding: 0.5rem 1rem;
-            font-weight: 500;
-            font-size: 13px;
-            transition: all 0.15s ease;
-        }
-        
-        .stButton > button:hover {
-            background: #4338CA;
-        }
-        
-        /* タイトルのスタイル */
-        h1 {
-            color: #111827;
-            font-weight: 600;
-            text-align: center;
-            font-size: 1.75rem;
-            margin-bottom: 0.5rem;
-        }
-        
-        /* サイドバーのスタイル */
-        section[data-testid="stSidebar"] {
-            background: #FAFAFB;
-        }
-        
-        /* 吹き出し内のコンテンツスタイル */
-        .bubble-content {
-            font-family: inherit;
-            font-size: inherit;
-            white-space: pre-wrap;
-            word-wrap: break-word;
-            margin: 0;
-            padding: 0;
-            color: inherit;
-        }
-        </style>
-    """, unsafe_allow_html=True)
-    
-    # UIインスタンス作成
-    bot = EmpathemeBotUI()
-    
-    # サイドバー設定
-    with st.sidebar:
-        st.markdown("## 設定")
-        
-        # APIキー入力欄（セッションごとに必要）
-        st.markdown("### OpenAI API キー")
-        
-        # 現在のAPIキー状態を表示
-        if st.session_state.current_api_key:
-            st.success("APIキー設定済み")
-            # APIキーの一部を表示（セキュリティのため）
-            masked_key = st.session_state.current_api_key[:7] + "..." + st.session_state.current_api_key[-4:]
-            st.caption(f"現在のキー: {masked_key}")
-        else:
-            st.info("APIキーを入力してください")
-        
-        # APIキー入力フォーム
-        with st.form("api_key_form"):
-            api_key_input = st.text_input(
-                "APIキーを入力",
-                type="password",
-                placeholder="sk-...",
-                help="このセッション専用のAPIキーです。ブラウザを閉じると消去されます。",
-                key="api_key_input_field"
-            )
-            
-            submit_button = st.form_submit_button("APIキーを設定", use_container_width=True)
-            
-            if submit_button and api_key_input:
-                if api_key_input.startswith("sk-"):
-                    # QAChainを初期化
-                    if bot.initialize_qa_chain(api_key_input):
-                        st.success("APIキーが設定されました")
-                        st.rerun()
-                    else:
-                        st.error("初期化に失敗しました")
-                else:
-                    st.error("有効なAPIキー（sk-で始まる）を入力してください")
-        
-        st.markdown("---")
-        
-        # コントロールボタン
-        st.markdown("### コントロール")
-        if st.button("新しいチャット", use_container_width=True):
-            bot.create_new_session()
-            st.rerun()
-        
-        if st.button("履歴クリア", use_container_width=True):
-            bot.clear_history()
-            st.rerun()
-        
-        st.markdown("---")
-        
-        # ステータス
-        st.markdown("### ステータス")
-        if st.session_state.vector_store_initialized:
-            st.success("システム準備完了")
-        else:
-            st.info("システム待機中")
-        
-        # セッション情報
-        st.markdown("---")
-        st.caption(f"セッションID: {st.session_state.session_id[:8]}...")
-        if st.session_state.current_api_key:
-            st.caption("APIキー設定済み（新しいチャットでも維持）")
-    
-    # メインヘッダー
-    st.markdown(
-        """
-        <div style='text-align: center; margin-bottom: 2rem;'>
-            <h1 style='margin-bottom: 0.25rem;'>EmpathemeBot</h1>
-            <p style='color: #6B7280; font-size: 0.9rem;'>Potionベースの質問応答システム</p>
-        </div>
-        """,
-        unsafe_allow_html=True
-    )
-    
-    # APIキー未入力時の警告メッセージ
-    if not st.session_state.current_api_key:
-        st.markdown(
-            """
-            <div style="background: #FEF3C7; border: 2px solid #F59E0B; border-radius: 12px; padding: 1.5rem; margin: 2rem 0;">
-                <h3 style="color: #92400E; margin-top: 0;">APIキーの入力が必要です</h3>
-                <p style="color: #78350F; margin-bottom: 1rem;">
-                    EmpathemeBotを使用するには、OpenAI APIキーが必要です。
-                </p>
-                <ol style="color: #78350F; margin-left: 1.5rem;">
-                    <li>左上の「>」ボタンをクリックしてサイドバーを開く</li>
-                    <li>「OpenAI API キー」セクションにAPIキー（sk-...）を入力</li>
-                    <li>「APIキーを設定」ボタンをクリック</li>
-                </ol>
-                <p style="color: #78350F; font-size: 0.9rem; margin-top: 1rem;">
-                    APIキーは <a href="https://platform.openai.com/api-keys" target="__blank" style="color: #F59E0B;">OpenAIのダッシュボード</a> から取得できます。
-                </p>
-            </div>
-            """,
-            unsafe_allow_html=True
-        )
-        st.stop()
-    
-    # ウェルカムメッセージ（初回のみ）
-    if len(st.session_state.messages) == 0:
-        st.markdown(
-            """
-            <div style="text-align: center; padding: 3rem 0; color: #6B7280;">
-                <p style="font-size: 0.95rem;">こんにちは、KurageSan®だよ！何か英語学習に関して困っていることはありますか？</p>
-            </div>
-            """,
-            unsafe_allow_html=True
-        )
-    
-    # チャット履歴の表示
-    for message in st.session_state.messages:
-        if message["role"] == "user":
-            # ユーザーメッセージ（右側）
-            st.markdown(
-                f"""
-                <div style="display: flex; justify-content: flex-end; margin: 1rem 0; padding-right: 1rem;">
-                    <div style="background: linear-gradient(135deg, #4F46E5 0%, #6366F1 100%); 
-                                color: white; 
-                                padding: 0.75rem 1.25rem; 
-                                border-radius: 18px 18px 4px 18px; 
-                                max-width: 60%; 
-                                box-shadow: 0 2px 10px rgba(79, 70, 229, 0.2);
-                                word-wrap: break-word;">
-                        <pre class="bubble-content">{html.escape(message['content'])}</pre>
-                        <div style="font-size: 0.7rem; opacity: 0.8; margin-top: 0.3rem; text-align: right;">
-                            {message.get('timestamp', '')}
-                        </div>
-                    </div>
-                </div>
-                """,
-                unsafe_allow_html=True
-            )
-        else:
-            # アシスタントメッセージ（左側）
-            st.markdown(
-                f"""
-                <div style="display: flex; justify-content: flex-start; margin: 1rem 0; padding-left: 1rem;">
-                    <div style="background: #F3F4F6; 
-                                color: #111827; 
-                                padding: 0.75rem 1.25rem; 
-                                border-radius: 18px 18px 18px 4px; 
-                                max-width: 60%; 
-                                box-shadow: 0 2px 10px rgba(0, 0, 0, 0.08);
-                                word-wrap: break-word;">
-                        <pre class="bubble-content">{html.escape(message['content'])}</pre>
-                        <div style="font-size: 0.7rem; opacity: 0.6; margin-top: 0.3rem;">
-                            {message.get('timestamp', '')}
-                        </div>
-                    </div>
-                </div>
-                """,
-                unsafe_allow_html=True
-            )
-    
-    # チャット入力
-    if prompt := st.chat_input("質問を入力してください...", key="chat_input"):
-        # タイムスタンプを追加
-        timestamp = datetime.now().strftime("%H:%M")
-        
-        # ユーザーメッセージを追加
-        st.session_state.messages.append({
-            "role": "user",
-            "content": prompt,
-            "timestamp": timestamp
-        })
-        
-        # ユーザーメッセージを表示
-        st.markdown(
-            f"""
-            <div style="display: flex; justify-content: flex-end; margin: 1rem 0; padding-right: 1rem;">
-                <div style="background: linear-gradient(135deg, #4F46E5 0%, #6366F1 100%); 
-                            color: white; 
-                            padding: 0.75rem 1.25rem; 
-                            border-radius: 18px 18px 4px 18px; 
-                            max-width: 60%; 
-                            box-shadow: 0 2px 10px rgba(79, 70, 229, 0.2);
-                            word-wrap: break-word;">
-                    <pre class="bubble-content">{html.escape(prompt)}</pre>
-                    <div style="font-size: 0.7rem; opacity: 0.8; margin-top: 0.3rem; text-align: right;">
-                        {timestamp}
-                    </div>
-                </div>
-            </div>
-            """,
-            unsafe_allow_html=True
-        )
-        
-        # アシスタントの応答を生成
-        with st.spinner("考えています..."):
-            response_timestamp = datetime.now().strftime("%H:%M")
-            response_data = bot.ask_question(prompt)
-        
-        if response_data:
-            answer = response_data["answer"]
-            
-            # メッセージ履歴に追加
-            st.session_state.messages.append({
-                "role": "assistant",
-                "content": answer,
-                "timestamp": response_timestamp,
-                "metadata": {
-                    "source_count": response_data.get("source_count", 0)
-                }
-            })
-            
-            # アシスタントメッセージを表示
-            st.markdown(
-                f"""
-                <div style="display: flex; justify-content: flex-start; margin: 1rem 0; padding-left: 1rem;">
-                    <div style="background: #F3F4F6; 
-                                color: #111827; 
-                                padding: 0.75rem 1.25rem; 
-                                border-radius: 18px 18px 18px 4px; 
-                                max-width: 60%; 
-                                box-shadow: 0 2px 10px rgba(0, 0, 0, 0.08);
-                                word-wrap: break-word;">
-                        <pre class="bubble-content">{html.escape(answer)}</pre>
-                        <div style="font-size: 0.7rem; opacity: 0.6; margin-top: 0.3rem;">
-                            {response_timestamp}
-                        </div>
-                    </div>
-                </div>
-                """,
-                unsafe_allow_html=True
-            )
-        else:
-            # エラーの場合
-            error_message = "申し訳ございません。回答の生成に失敗しました。もう一度お試しください。"
-            
-            st.session_state.messages.append({
-                "role": "assistant",
-                "content": error_message,
-                "timestamp": response_timestamp
-            })
-            
-            # エラーメッセージを表示
-            st.markdown(
-                f"""
-                <div style="display: flex; justify-content: flex-start; margin: 1rem 0; padding-left: 1rem;">
-                    <div style="background: #F3F4F6; 
-                                color: #111827; 
-                                padding: 0.75rem 1.25rem; 
-                                border-radius: 18px 18px 18px 4px; 
-                                max-width: 60%; 
-                                box-shadow: 0 2px 10px rgba(0, 0, 0, 0.08);
-                                word-wrap: break-word;">
-                        <pre class="bubble-content">{html.escape(error_message)}</pre>
-                        <div style="font-size: 0.7rem; opacity: 0.6; margin-top: 0.3rem;">
-                            {response_timestamp}
-                        </div>
-                    </div>
-                </div>
-                """,
-                unsafe_allow_html=True
-            )
-        
-        # アクティビティを更新
-        st.session_state.last_activity = datetime.now()
-    
-    # フッター
-    st.markdown(
-        f"""
-        <div style="text-align: center; margin-top: 3rem; padding: 1rem 0; 
-                    border-top: 1px solid #E5E7EB; color: #9CA3AF; font-size: 0.8rem;">
-            EmpathemeBot · セッション: {st.session_state.session_id[:8]}
-        </div>
-        """,
-        unsafe_allow_html=True
-    )
-
-if __name__ == "__main__":
-    main()