From 377855e34806da06275e7fb18599dd2aed295560 Mon Sep 17 00:00:00 2001
From: practicer2015 <practicer2015@devide>
Date: Thu, 17 Apr 2025 18:00:00 +0800
Subject: [PATCH] Thu Apr 17 18:00:00 CST 2025 inscode

---
 main.py | 59 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 58 insertions(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 4c0c135..8e6ee78 100644
--- a/main.py
+++ b/main.py
@@ -1 +1,58 @@
-print('欢迎来到 InsCode')
\ No newline at end of file
+import requests
+import json
+import os
+from flask import Flask, request, jsonify
+
+app = Flask(__name__)
+
+# 请替换为你的 DeepSeek API 地址和 API Key
+API_URL = "https://api.deepseek.com/v1"
+API_KEY = os.getenv('DeepSeek_API_KEY')
+
+def call_deepseek(prompt, max_new_tokens=150):
+    """
+    调用 DeepSeek API 生成文本Error: name 'os' is not defined
+
+ 
+
+    :param prompt: 输入的提示文本
+    :param max_new_tokens: 生成的最大新 token 数
+    :return: 生成的文本
+    """
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {API_KEY}"
+    }
+    data = {
+        "prompt": prompt,
+        "max_new_tokens": max_new_tokens
+    }
+    try:
+        response = requests.post(API_URL, headers=headers, data=json.dumps(data))
+        response.raise_for_status()
+        result = response.json().get("generated_text")
+        return result
+    except requests.RequestException as e:
+        print(f"调用 DeepSeek API 时请求出错: {e}")
+    except (KeyError, json.JSONDecodeError) as e:
+        print(f"解析 DeepSeek API 响应时出错: {e}")
+    return None
+
+@app.route('/generate', methods=['POST'])
+def generate_text():
+    data = request.get_json()
+    prompt = data.get('prompt')
+    max_new_tokens = data.get('max_new_tokens', 150)
+
+    if not prompt:
+        return jsonify({"error": "Prompt is required"}), 400
+
+    response = call_deepseek(prompt, max_new_tokens)
+    if response:
+        return jsonify({"generated_text": response})
+    else:
+        return jsonify({"error": "Failed to generate text"}), 500
+
+if __name__ == "__main__":
+    app.run(debug=True)
+    
\ No newline at end of file
-- 
GitLab