Spaces:
Sleeping
Sleeping
Update api.py
Browse files
api.py
CHANGED
@@ -3,12 +3,9 @@ from duckai import DuckAI
|
|
3 |
from flask_cors import CORS
|
4 |
|
5 |
app = Flask(__name__)
|
6 |
-
CORS(app
|
7 |
-
|
8 |
-
|
9 |
-
"methods": ["GET", "POST", "PUT", "DELETE", "OPTIONS"],
|
10 |
-
"supports_credentials": True
|
11 |
-
}})
|
12 |
|
13 |
@app.route("/chat/", methods=["GET"])
|
14 |
def chat():
|
@@ -16,7 +13,6 @@ def chat():
|
|
16 |
if not query:
|
17 |
return jsonify({"error": "Query parameter is required"}), 400
|
18 |
|
19 |
-
duck = DuckAI() # create one DuckAI instance
|
20 |
try:
|
21 |
results = duck.chat(query, model='gpt-4o-mini')
|
22 |
return jsonify({"results": results})
|
@@ -33,13 +29,5 @@ def chat():
|
|
33 |
"fallback_error": str(e2)
|
34 |
}), 500
|
35 |
|
36 |
-
def chat_with_model(query: str, model: str):
|
37 |
-
try:
|
38 |
-
duck = DuckAI()
|
39 |
-
results = duck.chat(query, model=model)
|
40 |
-
return jsonify({"results": results})
|
41 |
-
except Exception as e:
|
42 |
-
return jsonify({"error": str(e)}), 500
|
43 |
-
|
44 |
if __name__ == "__main__":
|
45 |
-
app.run(host="
|
|
|
3 |
from flask_cors import CORS
|
4 |
|
5 |
app = Flask(__name__)
|
6 |
+
CORS(app) # Simpler
|
7 |
+
|
8 |
+
duck = DuckAI() # Create once if DuckAI is heavy
|
|
|
|
|
|
|
9 |
|
10 |
@app.route("/chat/", methods=["GET"])
|
11 |
def chat():
|
|
|
13 |
if not query:
|
14 |
return jsonify({"error": "Query parameter is required"}), 400
|
15 |
|
|
|
16 |
try:
|
17 |
results = duck.chat(query, model='gpt-4o-mini')
|
18 |
return jsonify({"results": results})
|
|
|
29 |
"fallback_error": str(e2)
|
30 |
}), 500
|
31 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
32 |
if __name__ == "__main__":
|
33 |
+
app.run(host="0.0.0.0", port=3000, debug=True)
|