Delete app.py

app.py

@@ -1,563 +0,0 @@
-"""
-EmpathemeBot - Hugging Face Spaces用統合版Streamlitアプリ
-"""
-
-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))
-
-# QAチェーンのインポート
-from src.qa.chain import QAChain
-
-# ロギング設定
-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 EmpathemeBotUI:
-    """Hugging Face Spaces用統合版EmpathemeBot UIクラス"""
-    
-    def __init__(self):
-        # セッション状態の初期化
-        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 'api_key' not in st.session_state:
-            # Hugging Face Secretsから取得を試みる
-            st.session_state.api_key = os.getenv("OPENAI_API_KEY", "")
-        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
-            
-    def initialize_qa_chain(self, api_key: str) -> bool:
-        """
-        QAChainを初期化
-        
-        Args:
-            api_key: OpenAI APIキー
-            
-        Returns:
-            初期化成功の場合True
-        """
-        try:
-            # 環境変数にAPIキーを設定
-            os.environ["OPENAI_API_KEY"] = api_key
-            
-            # ベクトルストアのパスを確認
-            vector_store_path = Path("data/vector_store")
-            
-            # QAChainの初期化
-            if st.session_state.qa_chain is None:
-                logger.info("QAChainを初期化中...")
-                
-                # ベクトルストアが存在しない場合の処理
-                if not vector_store_path.exists():
-                    st.warning("ベクトルストアが見つかりません。デモモードで実行します。")
-                    # デモモード用の簡易実装
-                    st.session_state.qa_chain = self.create_demo_chain()
-                else:
-                    st.session_state.qa_chain = QAChain(
-                        persist_dir=str(vector_store_path),
-                        verbose=False,
-                        max_history_turns=10,
-                        max_history_chars=10000
-                    )
-                st.session_state.vector_store_initialized = True
-                logger.info("QAChain初期化完了")
-                return True
-                
-        except Exception as e:
-            logger.error(f"QAChain初期化エラー: {e}")
-            st.error(f"初期化エラー: {str(e)}")
-            return False
-            
-        return True
-    
-    def create_demo_chain(self):
-        """
-        デモ用の簡易QAチェーンを作成（ベクトルストアなし）
-        """
-        class DemoQAChain:
-            def __init__(self):
-                self.conversation_history = []
-                
-            def ask_with_history(self, question: str):
-                # デモ用の簡単な応答
-                self.conversation_history.append(f"Q: {question}")
-                
-                # OpenAI APIを直接使用して応答を生成
-                try:
-                    from langchain_openai import ChatOpenAI
-                    from langchain.schema import HumanMessage, SystemMessage
-                    
-                    llm = ChatOpenAI(
-                        model_name="gpt-4o-mini",
-                        temperature=0.7
-                    )
-                    
-                    messages = [
-                        SystemMessage(content="あなたは英語学習をサポートするKurageSan®という親切なアシスタントです。"),
-                        HumanMessage(content=question)
-                    ]
-                    
-                    response = llm.invoke(messages)
-                    answer = response.content
-                    
-                except Exception as e:
-                    answer = f"申し訳ございません。現在デモモードで動作しており、詳細な回答ができません。エラー: {str(e)}"
-                
-                self.conversation_history.append(f"A: {answer}")
-                return answer, []
-            
-            def clear_history(self):
-                self.conversation_history = []
-                
-            def get_history(self):
-                return "\n".join(self.conversation_history)
-        
-        return DemoQAChain()
-    
-    def ask_question(self, question: str) -> Optional[Dict]:
-        """
-        質問を処理して回答を取得
-        
-        Args:
-            question: ユーザーの質問
-            
-        Returns:
-            回答データ
-        """
-        try:
-            if st.session_state.qa_chain is None:
-                st.error("システムが初期化されていません。")
-                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:
-                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を生成"""
-        st.session_state.session_id = str(uuid.uuid4())
-        st.session_state.messages = []
-        if st.session_state.qa_chain:
-            st.session_state.qa_chain.clear_history()
-        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_key = st.text_input(
-            "APIキーを入力（必須）",
-            value=st.session_state.api_key,
-            type="password",
-            placeholder="sk-...",
-            help="OpenAI APIキーを入力してください。このフィールドは必須です。"
-        )
-        
-        # APIキーが変更された場合、セッション状態を更新
-        if api_key != st.session_state.api_key:
-            st.session_state.api_key = api_key
-            if api_key:
-                # QAChainを初期化
-                if bot.initialize_qa_chain(api_key):
-                    st.success("✅ APIキーが設定されました")
-                else:
-                    st.error("❌ 初期化に失敗しました")
-            else:
-                st.warning("⚠️ APIキーが未入力です")
-        
-        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(
-        """
-        <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.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>Enterキーを押して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()
-    
-    # APIキーがあるがQAChainが初期化されていない場合
-    if st.session_state.api_key and st.session_state.qa_chain is None:
-        if bot.initialize_qa_chain(st.session_state.api_key):
-            st.rerun()
-    
-    # ウェルカムメッセージ（初回のみ）
-    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()