Update app.py

app.py

@@ -2,13 +2,9 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from flask import Flask, request, jsonify, render_template_string
 import time
-from flask_sse import sse
-import redis
 
 # Flaskアプリケーションの設定
 app = Flask(__name__)
-app.config["REDIS_URL"] = "redis://localhost:6379/0"
-app.register_blueprint(sse, url_prefix='/stream')
 
 # デバイスの設定
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -96,14 +92,6 @@ HTML_TEMPLATE = """
                 alert('エラーが発生しました。コンソールを確認してください。');
             });
         }
-
-        // SSEの設定
-        const eventSource = new EventSource("/stream");
-
-        eventSource.onmessage = function(event) {
-            const message = event.data;
-            addMessageToChat('assistant', message);
-        };
     </script>
 </body>
 </html>
@@ -172,12 +160,6 @@ def generate():
         return jsonify({"error": "role_instruction and new_conversation are required fields"}), 400
 
     formatted_output_all, response = generate_response(role_instruction, conversation_history, new_conversation)
-    
-    # ここでSSEを介してリアルタイムで応答をストリームします
-    for word in response.split():
-        sse.publish({"message": word}, type='message')
-        time.sleep(0.5)  # 送信間隔をシミュレート
-
     return jsonify({"response": response, "conversation_history": conversation_history})
 
 if __name__ == '__main__':