Initial commit with OpenRouter/Tavily updates and app.py path fix

Dockerfile

@@ -0,0 +1,27 @@
+FROM python:3.11-slim
+
+WORKDIR /app
+
+# システムの依存関係をインストール
+RUN apt-get update && apt-get install -y \
+    git \
+    build-essential \
+    curl \
+    && rm -rf /var/lib/apt/lists/*
+
+# Pythonの依存関係をコピーしてインストール
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# プロジェクトファイルをコピー
+COPY . .
+
+# Streamlitのポート設定
+EXPOSE 7860
+
+# ヘルスチェック
+HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:7860/_stcore/health || exit 1
+
+# アプリケーション起動
+CMD ["streamlit", "run", "app.py", "--server.port=7860", "--server.address=0.0.0.0"]

app.py

@@ -0,0 +1,134 @@
+# app.py
+import os
+import sys
+import streamlit as st
+import subprocess
+from pathlib import Path
+
+# プロジェクトルートをパスに追加
+project_root = Path(__file__).parent
+sys.path.insert(0, str(project_root))
+sys.path.insert(0, str(project_root / "src")) # src ディレクトリをPythonのパスに追加
+
+def check_environment():
+    """環境設定をチェック"""
+    st.sidebar.markdown("### 🔧 システム状態")
+    
+    # API キーの確認
+    openai_key = os.getenv('OPENAI_API_KEY')
+    tavily_key = os.getenv('TAVILY_API_KEY')
+    
+    st.sidebar.markdown(f"**OpenRouter API**: {'✅' if openai_key else '❌'}")
+    st.sidebar.markdown(f"**Tavily API**: {'✅' if tavily_key else '❌'}")
+    
+    # Python バージョン
+    python_version = sys.version.split()[0]
+    st.sidebar.markdown(f"**Python**: {python_version}")
+    
+    # パッケージの確認
+    try:
+        import streamlit
+        st.sidebar.markdown(f"**Streamlit**: {streamlit.__version__}")
+    except ImportError:
+        st.sidebar.markdown("**Streamlit**: ❌")
+    
+    return openai_key, tavily_key
+
+def main():
+    st.set_page_config(
+        page_title="Lawsy - 法令Deep Research",
+        page_icon="⚖️",
+        layout="wide"
+    )
+    
+    # 環境チェック
+    openai_key, tavily_key = check_environment()
+    
+    # 必要な環境変数のチェック
+    if not openai_key:
+        st.error("⚠️ OPENAI_API_KEY（OpenRouter）が設定されていません")
+        st.info("Hugging Face Spaces の Settings → Variables and secrets で設定してください")
+        st.stop()
+    
+    if not tavily_key:
+        st.error("⚠️ TAVILY_API_KEY が設定されていません")
+        st.info("Hugging Face Spaces の Settings → Variables and secrets で設定してください")
+        st.stop()
+    
+    # 環境変数を設定
+    os.environ.setdefault('LAWSY_LM_PROVIDER', 'openrouter')
+    os.environ.setdefault('LAWSY_WEB_SEARCH_ENGINE', 'Tavily')
+    
+    # メインアプリケーションを起動
+    try:
+        # まず、lawsyモジュールが存在するかチェック
+        if (Path(__file__).parent / "src" / "lawsy").exists():
+            st.success("✅ lawsyプロジェクトが見つかりました")
+            
+            # 利用可能なモジュールを確認
+            lawsy_path = Path(__file__).parent / "lawsy"
+            python_files = list(lawsy_path.glob("**/*.py"))
+            
+            st.sidebar.markdown("### 📁 プロジェクト構造")
+            for py_file in python_files[:10]:  # 最初の10ファイルのみ表示
+                relative_path = py_file.relative_to(lawsy_path)
+                st.sidebar.markdown(f"- {relative_path}")
+            
+            if len(python_files) > 10:
+                st.sidebar.markdown(f"... 他 {len(python_files) - 10} ファイル")
+            
+            # デモインターフェース
+            st.title("⚖️ Lawsy - 法令Deep Research")
+            st.markdown("OpenRouter (shisa-ai/shisa-v2-llama3.3-70b:free) + Tavily API を使用")
+            
+            # 簡単なテストインターフェース
+            query = st.text_input("法令に関する質問を入力してください：", "民法第１条について教えてください")
+            
+            if st.button("検索・分析実行"):
+                with st.spinner("処理中..."):
+                    try:
+                        # ここで実際のlawsy機能を呼び出す
+                        st.info("🔄 OpenRouterとTavily APIを使用してリサーチを実行しています...")
+                        
+                        # Tavily検索のテスト
+                        from lawsy.search.tavily_search import TavilySearch
+                        search_engine = TavilySearch()
+                        search_results = search_engine.search(query, max_results=3)
+                        
+                        st.subheader("🔍 検索結果")
+                        for i, result in enumerate(search_results, 1):
+                            with st.expander(f"{i}. {result['title']}"):
+                                st.write(f"**URL**: {result['url']}")
+                                st.write(f"**内容**: {result['content'][:300]}...")
+                                st.write(f"**スコア**: {result['score']}")
+                        
+                        # OpenRouter LLMのテスト
+                        from lawsy.providers.openrouter import OpenRouterClient
+                        llm_client = OpenRouterClient()
+                        
+                        # 検索結果を含むプロンプトを作成
+                        context = "\n".join([f"- {r['title']}: {r['content'][:200]}..." for r in search_results])
+                        messages = [
+                            {"role": "system", "content": "あなたは法令の専門家です。検索結果を参考に、詳細で正確な回答を提供してください。"},
+                            {"role": "user", "content": f"質問: {query}\n\n検索結果:\n{context}\n\n上記の情報を踏まえて回答してください。"}
+                        ]
+                        
+                        response = llm_client.chat_completion(messages)
+                        
+                        st.subheader("🤖 AI分析結果")
+                        st.write(response['choices'][0]['message']['content'])
+                        
+                    except Exception as e:
+                        st.error(f"❌ 処理中にエラーが発生しました: {str(e)}")
+                        st.exception(e)
+            
+        else:
+            st.error("❌ lawsyプロジェクトが見つかりません")
+            st.info("プロジェクトの構造を確認してください")
+            
+    except Exception as e:
+        st.error(f"❌ アプリケーションの初期化に失敗しました: {str(e)}")
+        st.exception(e)
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -0,0 +1,16 @@
+streamlit>=1.28.0
+streamlit-chat
+streamlit-aggrid
+requests>=2.31.0
+openai>=1.0.0
+python-dotenv
+pandas
+numpy
+matplotlib
+plotly
+altair
+beautifulsoup4
+lxml
+pydantic
+fastapi
+uvicorn

src/lawsy/providers/openrouter.py

@@ -0,0 +1,64 @@
+# lawsy/providers/openrouter.py
+import os
+import requests
+from typing import Dict, Any, Iterator
+
+class OpenRouterClient:
+    def __init__(self):
+        self.api_key = os.getenv('OPENAI_API_KEY')  # OpenRouterのキーを流用
+        self.base_url = "https://openrouter.ai/api/v1"
+        self.model = "shisa-ai/shisa-v2-llama3.3-70b:free"
+    
+    def chat_completion(self, messages: list, **kwargs) -> Dict[str, Any]:
+        headers = {
+            "Authorization": f"Bearer {self.api_key}",
+            "Content-Type": "application/json"
+        }
+        
+        data = {
+            "model": self.model,
+            "messages": messages,
+            **kwargs
+        }
+        
+        response = requests.post(
+            f"{self.base_url}/chat/completions",
+            headers=headers,
+            json=data
+        )
+        response.raise_for_status()
+        return response.json()
+    
+    def stream_completion(self, messages: list, **kwargs) -> Iterator[Dict[str, Any]]:
+        headers = {
+            "Authorization": f"Bearer {self.api_key}",
+            "Content-Type": "application/json"
+        }
+        
+        data = {
+            "model": self.model,
+            "messages": messages,
+            "stream": True,
+            **kwargs
+        }
+        
+        response = requests.post(
+            f"{self.base_url}/chat/completions",
+            headers=headers,
+            json=data,
+            stream=True
+        )
+        response.raise_for_status()
+        
+        for line in response.iter_lines():
+            if line:
+                line = line.decode('utf-8')
+                if line.startswith('data: '):
+                    data = line[6:]
+                    if data.strip() == '[DONE]':
+                        break
+                    try:
+                        import json
+                        yield json.loads(data)
+                    except json.JSONDecodeError:
+                        continue

src/lawsy/search/tavily_search.py

@@ -0,0 +1,47 @@
+# lawsy/search/tavily_search.py
+import os
+import requests
+from typing import List, Dict, Any
+
+class TavilySearch:
+    def __init__(self):
+        self.api_key = os.getenv('TAVILY_API_KEY')
+        self.base_url = "https://api.tavily.com"
+    
+    def search(self, query: str, max_results: int = 5) -> List[Dict[str, Any]]:
+        if not self.api_key:
+            raise ValueError("TAVILY_API_KEY environment variable is required")
+        
+        headers = {
+            "Content-Type": "application/json"
+        }
+        
+        data = {
+            "api_key": self.api_key,
+            "query": query,
+            "search_depth": "basic",
+            "include_answer": True,
+            "include_images": False,
+            "include_raw_content": False,
+            "max_results": max_results
+        }
+        
+        response = requests.post(
+            f"{self.base_url}/search",
+            headers=headers,
+            json=data
+        )
+        response.raise_for_status()
+        result = response.json()
+        
+        # 結果を標準化された形式に変換
+        formatted_results = []
+        for item in result.get('results', []):
+            formatted_results.append({
+                'title': item.get('title', ''),
+                'url': item.get('url', ''),
+                'content': item.get('content', ''),
+                'score': item.get('score', 0)
+            })
+        
+        return formatted_results