deepak191z commited on
Commit
fc8fe81
·
verified ·
1 Parent(s): 3d9ca49

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +4 -16
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, resources={r"/*": {
7
- "origins": "*",
8
- "allow_headers": "*",
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="localhost", port=3000, debug=True)
 
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)