Utkarsh Verma commited on
Commit
e5b71df
·
1 Parent(s): 470bb28

Adding Docker File

Browse files
Files changed (1) hide show
  1. app.py +9 -6
app.py CHANGED
@@ -4,8 +4,8 @@ from flask import Flask, request, jsonify, render_template
4
 
5
  app = Flask(__name__)
6
 
7
- # Set your OpenAI API key (Replace 'your-api-key' with your actual key)
8
- openai.api_key = "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"
9
 
10
  @app.route('/')
11
  def home():
@@ -19,14 +19,17 @@ def chat():
19
  return jsonify({"error": "Empty message received"})
20
 
21
  try:
22
- response = openai.ChatCompletion.create(
 
23
  model="gpt-4",
24
  messages=[{"role": "user", "content": user_message}]
25
  )
26
- reply = response["choices"][0]["message"]["content"]
27
  return jsonify({"reply": reply})
28
- except Exception as e:
29
- return jsonify({"error": str(e)})
 
 
30
 
31
  if __name__ == '__main__':
32
  app.run(host='0.0.0.0', port=7860)
 
4
 
5
  app = Flask(__name__)
6
 
7
+ # Set your OpenAI API key
8
+ openai.api_key = "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" # Use environment variables for security
9
 
10
  @app.route('/')
11
  def home():
 
19
  return jsonify({"error": "Empty message received"})
20
 
21
  try:
22
+ client = openai.OpenAI() # New API usage
23
+ response = client.chat.completions.create(
24
  model="gpt-4",
25
  messages=[{"role": "user", "content": user_message}]
26
  )
27
+ reply = response.choices[0].message.content
28
  return jsonify({"reply": reply})
29
+
30
+ except openai.OpenAIError as e:
31
+ print(f"OpenAI API Error: {e}") # Print error for debugging
32
+ return jsonify({"reply": f"Error in response: {str(e)}"}) # Show error in UI
33
 
34
  if __name__ == '__main__':
35
  app.run(host='0.0.0.0', port=7860)