Update app.py

app.py

@@ -16,15 +16,15 @@ def prepare_model():
             env["HF_TOKEN"] = token
         
         try:
-            # スクショにある正確なモデル名。ミスは許されない。
+            # スクショにある正確なモデル名。
             subprocess.run(
                 ["mergekit-yaml", "config.yaml", model_path, "--allow-crimes"], 
                 check=True, 
                 env=env
             )
-            print("✨ マージ成功。のっぽ、今度は本物だ。")
+            print("✨ マージ成功だぜ！")
         except subprocess.CalledProcessError as e:
-            print(f"❌ マージエラー。モデル名かゲート設定を確認。")
+            print(f"❌ マージエラー。")
             raise e
     
     print("🧠 NI-v1 ロード中...")
@@ -37,7 +37,7 @@ def prepare_model():
     )
     return pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-# 初期化
+# グローバルでパイプラインを初期化
 try:
     pipe = prepare_model()
 except Exception as e:
@@ -45,25 +45,33 @@ except Exception as e:
     pipe = None
 
 def chat_fn(message, history):
-    if pipe is None: return "知能ユニット起動失敗。ログを見てくれ。"
+    if pipe is None: return "知能ユニットが起動してないぜ。"
+    
+    # プロンプト構築
     prompt = f"<|im_start|>user\n{message}<|im_end|>\n<|im_start|>assistant\n"
     outputs = pipe(prompt, max_new_tokens=512, do_sample=True, temperature=0.7)
-    return outputs[0]['generated_text'].split("assistant\n")[-1].replace("<|im_end|>", "")
+    
+    # 応答を抽出
+    response = outputs[0]['generated_text'].split("assistant\n")[-1].replace("<|im_end|>", "")
+    return response
 
+# --- UI設定 ---
 with gr.Blocks(title="NI-v1.0") as demo:
     gr.Markdown("# 🤖 Noppo-Intelligence v1.0")
-    gr.Markdown("Maverick-17B と Gemma-4-E4B を融合した、のっぽ指定の真のマージAIだ。")
     
     with gr.Tab("チャット"):
-        chatbot = gr.Chatbot()
+        # Docker環境で最新のGradioがインストールされることを想定し、辞書型(messages)を強制
+        chatbot = gr.Chatbot(type="messages")
         msg = gr.Textbox(placeholder="最新モデルの力を見せてくれ。")
         with gr.Row():
             send = gr.Button("送信", variant="primary")
             clear = gr.Button("クリア")
         
         def respond(message, chat_history):
+            # 辞書形式で履歴を管理
+            chat_history.append({"role": "user", "content": message})
             bot_message = chat_fn(message, chat_history)
-            chat_history.append((message, bot_message))
+            chat_history.append({"role": "assistant", "content": bot_message})
             return "", chat_history
 
         msg.submit(respond, [msg, chatbot], [msg, chatbot])
@@ -71,7 +79,7 @@ with gr.Blocks(title="NI-v1.0") as demo:
         clear.click(lambda: None, None, chatbot, queue=False)
     
     with gr.Tab("公開"):
-        repo_id = gr.Textbox(label="Repo ID", placeholder="noppodev/NoppoIntelligence")
+        repo_id = gr.Textbox(label="Repo ID")
         user_token = gr.Textbox(label="Write Token", type="password")
         pub_btn = gr.Button("アップロード")
         status = gr.Textbox(label="Status")
@@ -86,4 +94,6 @@ with gr.Blocks(title="NI-v1.0") as demo:
         
         pub_btn.click(upload, [repo_id, user_token], status)
 
-demo.launch(server_name="0.0.0.0", server_port=7860)
+if __name__ == "__main__":
+    # Dockerコンテナ内で外部からアクセスできるように設定
+    demo.launch(server_name="0.0.0.0", server_port=7860)