Upload 9 files

app.py

@@ -5,73 +5,41 @@ import json
 from dotenv import load_dotenv
 from transformers import pipeline
 import re
-# llama-cpp-pythonのインポート（様々な可能性を試す）
+# llama-cpp-pythonのインポート（シンプルに）
 Llama = None
 import sys
 import importlib
 
-# インストールされているパッケージを確認
-print("インストールされているパッケージ:")
-try:
-    import pkg_resources
-    for pkg in pkg_resources.working_set:
-        if "llama" in pkg.key.lower():
-            print(f"- {pkg.key} ({pkg.version})")
-except Exception as e:
-    print(f"パッケージ一覧の取得に失敗: {e}")
+# ログファイルを設定
+import logging
+logging.basicConfig(filename='app.log', level=logging.INFO)
+logger = logging.getLogger(__name__)
 
-# 可能性のあるモジュール名のリスト
+# 可能性のあるモジュール名のリスト（最小限に）
 possible_modules = [
     "llama_cpp",
-    "llama_cpp_python",
-    "llama",
-    "llamacpp",
-    "cpp_llama",
-    "llama.cpp",
-    "llama_cpp.llama",
-    "llama_cpp_python.llama_cpp",
-    "llama_cpp_python.llama"
+    "llama_cpp_python"
 ]
 
-# 各モジュールを試す
+# 各モジュールを試す（最小限の出力）
 for module_name in possible_modules:
     try:
-        print(f"{module_name}のインポートを試みます...")
+        logger.info(f"{module_name}のインポートを試みます...")
         module = importlib.import_module(module_name)
-        print(f"成功: {module_name}をインポートしました")
         
         # モジュール内のLlamaクラスを探す
         if hasattr(module, "Llama"):
             Llama = module.Llama
-            print(f"Llamaクラスを{module_name}から取得しました")
+            logger.info(f"Llamaクラスを{module_name}から取得しました")
             break
-        else:
-            print(f"{module_name}にLlamaクラスが見つかりません")
-            
-            # モジュールの属性を調べる
-            print(f"{module_name}の属性: {dir(module)}")
-            
-            # 'llama'という名前の属性があるか確認
-            if hasattr(module, "llama"):
-                llama_submodule = getattr(module, "llama")
-                if hasattr(llama_submodule, "Llama"):
-                    Llama = llama_submodule.Llama
-                    print(f"Llamaクラスを{module_name}.llamaから取得しました")
-                    break
     except ImportError:
-        print(f"{module_name}のインポートに失敗しました")
+        logger.info(f"{module_name}のインポートに失敗しました")
     except Exception as e:
-        print(f"{module_name}のインポート中にエラーが発生しました: {e}")
+        logger.error(f"{module_name}のインポート中にエラーが発生しました: {e}")
 
 # Llamaクラスが見つからなかった場合
 if Llama is None:
-    print("警告: Llamaクラスが見つかりませんでした。")
-    print("Pythonパス:")
-    for path in sys.path:
-        print(f"- {path}")
-    
-    # フォールバックとして、transformersのパイプラインを使用
-    print("フォールバックとして、transformersのパイプラインを使用します。")
+    logger.warning("Llamaクラスが見つかりませんでした。フォールバックを使用します。")
 from huggingface_hub import hf_hub_download
 from generate_dialogue_with_swallow import generate_dialogue_with_swallow
 
@@ -558,6 +526,49 @@ except FileNotFoundError:
         padding: 15px !important;
         box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1) !important;
         margin-bottom: 20px !important;
+        min-height: 400px !important;
+        max-height: 600px !important;
+        overflow-y: auto !important;
+    }
+    
+    /* 送信ボタンのスタイル */
+    button.primary {
+        background-color: #ff6b6b !important;
+        color: white !important;
+        border-radius: 8px !important;
+        padding: 10px 20px !important;
+        font-weight: bold !important;
+        margin-top: 10px !important;
+        width: 100% !important;
+        max-width: 200px !important;
+    }
+    
+    /* 入力欄のスタイル */
+    .input-box textarea {
+        border: 2px solid #ff6b6b !important;
+        border-radius: 8px !important;
+        padding: 10px !important;
+        font-size: 16px !important;
+        width: 100% !important;
+    }
+    
+    /* チャットメッセージのスタイル */
+    .message {
+        margin-bottom: 10px !important;
+        padding: 10px !important;
+        border-radius: 8px !important;
+    }
+    
+    /* ユーザーメッセージ */
+    .user-message {
+        background-color: rgba(230, 230, 255, 0.95) !important;
+        border-left: 4px solid #6c5ce7 !important;
+    }
+    
+    /* ボットメッセージ */
+    .bot-message {
+        background-color: rgba(255, 230, 230, 0.95) !important;
+        border-left: 4px solid #ff6b6b !important;
     }
     """
 
@@ -583,7 +594,7 @@ except Exception as e:
     print(f"テーマカスタマイズエラー: {e}")
     # エラーが発生した場合はデフォルトのテーマを使用
 
-with gr.Blocks(css=custom_css, theme=custom_theme) as demo:
+with gr.Blocks(css=custom_css, theme=custom_theme, title="麻理チャット") as demo:
     scene_state = gr.State(DEFAULT_SCENE_PARAMS)
     affection_state = gr.State(30)
     history_state = gr.State([])
@@ -647,22 +658,30 @@ with gr.Blocks(css=custom_css, theme=custom_theme) as demo:
         with gr.Column(scale=2):
             # チャットコンテナ（背景と分離）
             with gr.Group(elem_id="chat_container", elem_classes="chat-box"):
-                # Gradio 5.x用のChatbot設定
+                # Gradio 5.x用のChatbot設定（シンプルな設定に変更）
                 chatbot = gr.Chatbot(
                     label="麻理との会話", 
                     elem_id="chat_area", 
-                    show_label=False,
-                    height=400,
-                    # Gradio 5.0では type="messages" を使用すると形式が変わるため、
-                    # 従来の形式を使用
-                    avatar_images=[
+                    show_label=True,  # ラベルを表示
+                    height=500,       # 高さを増やす
+                    bubble=True,      # バブルスタイルを使用
+                    # アバター画像を設定
+                    avatar_images=(
                         "https://cdn.pixabay.com/photo/2016/04/01/10/04/amusing-1299756_1280.png", 
                         "https://cdn.pixabay.com/photo/2016/03/31/21/40/bot-1296595_1280.png"
-                    ]
+                    )
                 )
             # 入力欄（背景と分離）
             with gr.Group(elem_classes="input-box"):
-                msg_input = gr.Textbox(label="あなたのメッセージ", placeholder="「水族館はどう？」と聞いた後、「いいね、行こう！」のように返してみてください", show_label=False)
+                msg_input = gr.Textbox(
+                    label="あなたのメッセージ", 
+                    placeholder="「水族館はどう？」と聞いた後、「いいね、行こう！」のように返してみてください", 
+                    show_label=False,
+                    lines=2,
+                    max_lines=5
+                )
+                # 送信ボタンを明示的に追加
+                submit_btn = gr.Button("送信", variant="primary")
         
         # ステータス部分（右側、背景と分離）
         with gr.Column(scale=1):
@@ -678,11 +697,19 @@ with gr.Blocks(css=custom_css, theme=custom_theme) as demo:
         アイコン: <a href="https://pixabay.com" target="_blank">Pixabay</a>
         </div>
         """)
+    # 入力欄のEnterキーと送信ボタンの両方にイベントハンドラを設定
     msg_input.submit(
         respond,
         [msg_input, chatbot, affection_state, history_state, scene_state],
         [msg_input, chatbot, affection_gauge, stage_display, affection_state, history_state, scene_state, background_display]
     )
+    
+    # 送信ボタンにも同じイベントハンドラを設定
+    submit_btn.click(
+        respond,
+        [msg_input, chatbot, affection_state, history_state, scene_state],
+        [msg_input, chatbot, affection_gauge, stage_display, affection_state, history_state, scene_state, background_display]
+    )
     # 通常の関数として定義
     def load_stage(affection):
         return get_relationship_stage(affection)
@@ -693,10 +720,23 @@ if __name__ == "__main__":
     # Hugging Face Spacesでの実行時の設定
     if os.getenv("SPACE_ID"):
         # Hugging Face Spacesでは、サーバーの設定を自動的に行う
-        demo.launch()
+        demo.launch(
+            server_name="0.0.0.0",
+            share=False,
+            debug=False,
+            show_api=False,
+            show_error=False,
+            quiet=True,
+            favicon_path="https://cdn.pixabay.com/photo/2016/03/31/21/40/bot-1296595_1280.png"
+        )
     else:
         # ローカル環境での実行時の設定
         demo.launch(
-            show_error=True,           # エラーを表示
-            quiet=False                # 詳細なログを出力
+            server_name="0.0.0.0",
+            share=False,
+            debug=True,
+            show_api=False,
+            show_error=True,
+            quiet=False,
+            favicon_path="https://cdn.pixabay.com/photo/2016/03/31/21/40/bot-1296595_1280.png"
         )