Update app.py

app.py

@@ -10,18 +10,18 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 from huggingface_hub import snapshot_download
 
 # ==================================================
-# 設定（DBの保存先を/tmpに変更してエラーを回避）
+# 設定
 # ==================================================
 
 PAPER_DIR = "papers"
-DB_DIR = "/tmp/chroma_db"  # HF Spaces等での書き込みエラー対策
+DB_DIR = "/tmp/chroma_db"  # Read-onlyエラー回避のため、一時ディレクトリを指定
 NOTE_FILE = "research_notes.txt"
 
 os.makedirs(PAPER_DIR, exist_ok=True)
 
 
 # ==================================================
-# Gemini / 埋め込みモデルの設定
+# Gemini / Embedding モデル設定
 # ==================================================
 
 API_KEY = os.getenv("GEMINI_API_KEY")
@@ -31,17 +31,19 @@ if API_KEY is None:
         "GEMINI_API_KEY が設定されていません。環境変数を確認してください。"
     )
 
+# 埋め込みモデルの初期化
 embedding_model = HuggingFaceEmbeddings(
     model="sentence-transformers/all-MiniLM-L6-v2"
 )
 
+# Geminiモデルの初期化 (安定性の高い2.5-flashを採用)
 llm = ChatGoogleGenerativeAI(
     model="gemini-2.5-flash",
     temperature=0.3
 )
 
 # ==================================================
-# PDF → DB 登録処理
+# PDF → DB 登録・再構築
 # ==================================================
 
 def rebuild_database():
@@ -65,8 +67,9 @@ def rebuild_database():
             print(f"PDF読み込みエラー ({pdf}): {e}")
 
     if not all_docs:
-        return "読み込み可能なPDFテキストがありませんでした。"
+        return "読み込み可能なPDFテキストがありませんでした"
 
+    # テキストの最適化分割
     splitter = RecursiveCharacterTextSplitter(
         chunk_size=1000,
         chunk_overlap=200
@@ -74,14 +77,14 @@ def rebuild_database():
 
     chunks = splitter.split_documents(all_docs)
 
-    # 既存のDBディレクトリがあれば一度削除（初期化）
+    # 既存の古いDBディレクトリを安全に削除して再初期化
     if os.path.exists(DB_DIR):
         try:
             shutil.rmtree(DB_DIR)
         except Exception as e:
-            print(f"DB削除エラー: {e}")
+            print(f"既存DBの削除に失敗しました: {e}")
 
-    # 新しくベクトルDBを構築
+    # ベクトルDBの構築
     Chroma.from_documents(
         documents=chunks,
         embedding=embedding_model,
@@ -98,6 +101,7 @@ def rebuild_database():
 def get_db():
     if not os.path.exists(DB_DIR):
         return None
+
     return Chroma(
         persist_directory=DB_DIR,
         embedding_function=embedding_model
@@ -105,7 +109,7 @@ def get_db():
 
 
 # ==================================================
-# PDF追加（画面からのアップロード用）
+# PDF追加
 # ==================================================
 
 def upload_pdfs(files):
@@ -118,10 +122,15 @@ def upload_pdfs(files):
             PAPER_DIR,
             os.path.basename(file.name)
         )
-        shutil.copy(file.name, dst)
-        count += 1
+        try:
+            shutil.copy(file.name, dst)
+            count += 1
+        except Exception as e:
+            print(f"ファイルコピーエラー: {e}")
 
+    # 追加後にデータベースを再構築
     msg = rebuild_database()
+
     return f"{count}件アップロード\n{msg}"
 
 
@@ -131,9 +140,11 @@ def upload_pdfs(files):
 
 def search_method(question):
     db = get_db()
+
     if db is None:
         return "PDFを登録してください"
 
+    # 関連ドキュメントの検索
     docs = db.similarity_search(question, k=5)
     context = "\n\n".join([d.page_content for d in docs])
 
@@ -152,15 +163,16 @@ def search_method(question):
 
 【回答】
 """
+
     try:
         response = llm.invoke(prompt)
         return response.content
     except Exception as e:
-        return f"Geminiの呼び出し中にエラーが発生しました: {e}"
+        return f"Geminiによる生成エラーが発生しました: {e}"
 
 
 # ==================================================
-# データダウンロード & 初期DB構築
+# 初回データセット取得 & データベース初期化
 # ==================================================
 
 try:
@@ -169,9 +181,9 @@ try:
         repo_type="dataset",
         local_dir=PAPER_DIR
     )
-    print("データセットのダウンロードに成功しました:", os.listdir(PAPER_DIR))
+    print("データセットのダウンロードに成功しました")
 except Exception as e:
-    print(f"データセットのダウンロードに失敗しました (手動アップロードを試みてください): {e}")
+    print(f"データセットのダウンロードに失敗しました (手動での登録が必要です): {e}")
     
 if not os.path.exists(DB_DIR):
     print(rebuild_database())
@@ -183,6 +195,7 @@ if not os.path.exists(DB_DIR):
 
 def search_experiment(question):
     db = get_db()
+
     if db is None:
         return "PDFを登録してください"
 
@@ -198,7 +211,7 @@ def search_experiment(question):
 ・データセット
 ・学習率
 ・Epoch数
- * 評価指標
+・評価指標
 ・比較手法
 ・実験結果
 
@@ -210,15 +223,16 @@ def search_experiment(question):
 
 【回答】
 """
+
     try:
         response = llm.invoke(prompt)
         return response.content
     except Exception as e:
-        return f"Geminiの呼び出し中にエラーが発生しました: {e}"
+        return f"Geminiによる生成エラーが発生しました: {e}"
 
 
 # ==================================================
-# 保存処理
+# 保存
 # ==================================================
 
 def save_note(question, answer):
@@ -229,20 +243,24 @@ def save_note(question, answer):
 
     try:
         with open(NOTE_FILE, "a", encoding="utf-8") as f:
-            f.write("\n" + "=" * 60 + "\n")
+            f.write("\n")
+            f.write("=" * 60)
+            f.write("\n")
             f.write(f"日時: {now}\n\n")
             f.write(f"質問:\n{question}\n\n")
-            f.write(f"回答:\n{answer}\n\n")
+            f.write(f"回答:\n{answer}\n")
+            f.write("\n")
         return "保存完了"
     except Exception as e:
         return f"ファイル保存エラー: {e}"
 
 
 # ==================================================
-# UI (Gradio)
+# UI (Gradio) ※インターフェース構成は完全再現
 # ==================================================
 
 with gr.Blocks() as demo:
+
     gr.Markdown("# 研究支援RAGアプリ")
 
     # タブ1: 論文追加
@@ -252,7 +270,9 @@ with gr.Blocks() as demo:
             file_types=[".pdf"],
             label="論文PDF"
         )
+
         upload_btn = gr.Button("登録")
+
         upload_status = gr.Textbox(label="状態")
 
         upload_btn.click(
@@ -264,9 +284,16 @@ with gr.Blocks() as demo:
     # タブ2: 手法・専門用語検索
     with gr.Tab("手法・専門用語検索"):
         method_question = gr.Textbox(label="質問")
-        method_answer = gr.Textbox(label="回答", lines=15)
+
+        method_answer = gr.Textbox(
+            label="回答",
+            lines=15
+        )
+
         method_btn = gr.Button("検索")
-        method_save = gr.Button("ノートへ保存")
+
+        method_save = gr.Button("保存")
+
         method_status = gr.Textbox(label="保存状態")
 
         method_btn.click(
@@ -274,6 +301,7 @@ with gr.Blocks() as demo:
             inputs=method_question,
             outputs=method_answer
         )
+
         method_save.click(
             fn=save_note,
             inputs=[method_question, method_answer],
@@ -283,20 +311,19 @@ with gr.Blocks() as demo:
     # タブ3: 実験内容検索
     with gr.Tab("実験内容検索"):
         exp_question = gr.Textbox(label="質問")
-        exp_answer = gr.Textbox(label="回答", lines=15)
+
+        exp_answer = gr.Textbox(
+            label="回答",
+            lines=15
+        )
+
         exp_btn = gr.Button("検索")
-        exp_save = gr.Button("ノートへ保存")  # gr.DownloadButton から変更
+
+        exp_save = gr.Button("保存")  # 整合性を保つため gr.Button に修正
+
         exp_status = gr.Textbox(label="保存状態")
 
         exp_btn.click(
             fn=search_experiment,
             inputs=exp_question,
-            outputs=exp_answer
-        )
-        exp_save.click(
-            fn=save_note,
-            inputs=[exp_question, exp_answer],
-            outputs=exp_status
-        )
-
-demo.launch()
+            outputs=exp_answer