Update app.py

app.py

@@ -1,57 +1,86 @@
-from flask import Flask, render_template, jsonify, request
-from movie_controller import MovieController
+import os
+from flask import Flask, request, abort
+import hashlib
+import xmltodict
+import openai
+from wechatpy import parse_message
+from wechatpy.utils import check_signature
+from wechatpy.exceptions import InvalidSignatureException
 
 app = Flask(__name__)
-controller = MovieController()
-
-@app.route('/')
-def index():
-    return render_template('index.html')
-
-@app.route('/api/movies')
-def get_movies():
-    movies = controller.get_all_movies()
-    return jsonify([movie.__dict__ for movie in movies])
-
-@app.route('/api/highest-rated')
-def get_highest_rated():
-    movie = controller.get_highest_rated_movie()
-    return jsonify(movie.__dict__)
-
-@app.route('/api/highest-grossing')
-def get_highest_grossing():
-    movie = controller.get_highest_grossing_movie()
-    return jsonify(movie.__dict__)
-
-@app.route('/api/movies-by-year')
-def get_movies_by_year():
-    movies = controller.get_movies_sorted_by_year()
-    return jsonify([movie.__dict__ for movie in movies])
-
-@app.route('/api/genres')
-def get_genres():
-    genres = controller.get_all_genres()
-    return jsonify(genres)
-
-@app.route('/api/languages')
-def get_languages():
-    languages = controller.get_all_languages()
-    return jsonify(languages)
-
-@app.route('/api/movies/filter')
-def filter_movies():
-    genre = request.args.get('genre', '')
-    year = request.args.get('year', '')
-    language = request.args.get('language', '')
-    movies = controller.filter_movies(genre, year, language)
-    return jsonify([movie.__dict__ for movie in movies])
-
-@app.route('/api/movies/sort')
-def sort_movies():
-    sort_by = request.args.get('sort_by', 'year')
-    order = request.args.get('order', 'desc')
-    movies = controller.sort_movies(sort_by, order)
-    return jsonify([movie.__dict__ for movie in movies])
+
+# 配置
+TOKEN = 'your_wechat_token'
+APPID = 'your_wechat_appid'
+APPSECRET = 'your_wechat_appsecret'
+OPENAI_API_KEY = 'your_openai_api_key'
+
+openai.api_key = OPENAI_API_KEY
+
+# 存储用户当前使用的模型
+user_models = {}
+
+def split_message(message, max_length=500):
+    """Split a message into chunks of max_length characters."""
+    return [message[i:i+max_length] for i in range(0, len(message), max_length)]
+
+def get_gpt_response(message, model="gpt-3.5-turbo"):
+    """Get response from GPT model."""
+    try:
+        response = openai.ChatCompletion.create(
+            model=model,
+            messages=[{"role": "user", "content": message}]
+        )
+        return response.choices[0].message.content
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+@app.route('/', methods=['GET', 'POST'])
+def wechat():
+    if request.method == 'GET':
+        token = TOKEN
+        signature = request.args.get('signature', '')
+        timestamp = request.args.get('timestamp', '')
+        nonce = request.args.get('nonce', '')
+        echostr = request.args.get('echostr', '')
+        try:
+            check_signature(token, signature, timestamp, nonce)
+        except InvalidSignatureException:
+            abort(403)
+        return echostr
+    else:
+        xml_data = request.data
+        msg = parse_message(xml_data)
+        if msg.type == 'text':
+            user_id = msg.source
+            content = msg.content
+
+            if content.startswith('/model'):
+                # 切换模型
+                model = content.split(' ')[1]
+                user_models[user_id] = model
+                return f'Model switched to {model}'
+
+            model = user_models.get(user_id, "gpt-3.5-turbo")
+            response = get_gpt_response(content, model)
+            
+            # 拆分长消息
+            response_parts = split_message(response)
+            
+            # 构建回复消息
+            reply = []
+            for part in response_parts:
+                reply.append(f"""
+                <xml>
+                <ToUserName><![CDATA[{msg.source}]]></ToUserName>
+                <FromUserName><![CDATA[{msg.target}]]></FromUserName>
+                <CreateTime>{int(time.time())}</CreateTime>
+                <MsgType><![CDATA[text]]></MsgType>
+                <Content><![CDATA[{part}]]></Content>
+                </xml>
+                """)
+            
+            return ''.join(reply)
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860, debug=True)
+    app.run(host='0.0.0.0', port=7860)