datasaur-dev commited on
Commit
b6954d6
·
verified ·
1 Parent(s): 1f2d6db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -5,7 +5,7 @@ import re
5
 
6
  # WARNING: It is not recommended to hardcode sensitive data like API tokens in code.
7
  # Consider using environment variables or other secure methods for production applications.
8
- API_URL = "https://deployment.datasaur.ai/api/deployment/8/2717/chat/completions"
9
  API_TOKEN = os.environ["DATASAUR_API_KEY"]
10
 
11
  def magic_function(input_text):
@@ -17,7 +17,7 @@ def magic_function(input_text):
17
  "Authorization": f"Bearer {API_TOKEN}",
18
  }
19
  data = {
20
- "messages": [{"role": "user", "content": f"User Input: `{input_text}`"}]
21
  }
22
 
23
  try:
@@ -31,8 +31,8 @@ def magic_function(input_text):
31
  content = response_json.get("choices", [{}])[0].get("message", {}).get("content", "Error: Could not parse response.")
32
 
33
  # Remove unwanted prefixes and characters from the response.
34
- # This regex removes "Your Suggestion: " or "**Your Suggestion**: " from the beginning of the string, case-insensitively.
35
- content = re.sub(r'^\s*(\*\*Your Suggestion\*\*:\s*|Your Suggestion:\s*)', '', content, flags=re.IGNORECASE)
36
  content = content.replace("`", "")
37
 
38
  return content.strip()
 
5
 
6
  # WARNING: It is not recommended to hardcode sensitive data like API tokens in code.
7
  # Consider using environment variables or other secure methods for production applications.
8
+ API_URL = "https://deployment.datasaur.ai/api/deployment/8/2723/chat/completions"
9
  API_TOKEN = os.environ["DATASAUR_API_KEY"]
10
 
11
  def magic_function(input_text):
 
17
  "Authorization": f"Bearer {API_TOKEN}",
18
  }
19
  data = {
20
+ "messages": [{"role": "user", "content": f"**Input:** `{input_text}`"}]
21
  }
22
 
23
  try:
 
31
  content = response_json.get("choices", [{}])[0].get("message", {}).get("content", "Error: Could not parse response.")
32
 
33
  # Remove unwanted prefixes and characters from the response.
34
+ # This regex removes "Suggestion: " or "**Suggestion**: " from the beginning of the string, case-insensitively.
35
+ content = re.sub(r'^\s*(\*\*Suggestion\*\*:\s*|Suggestion:\s*)', '', content, flags=re.IGNORECASE)
36
  content = content.replace("`", "")
37
 
38
  return content.strip()