hruday96 commited on
Commit
be75daa
·
verified ·
1 Parent(s): b92d835

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -40,8 +40,11 @@ if st.button("Enhance Prompt"):
40
  if user_prompt.strip() == "":
41
  st.warning("⚠️ Please enter a prompt before enhancing.")
42
  else:
43
- # Generate enhanced prompt using OpenAI API
44
  try:
 
 
 
 
45
  completion = client.chat.completions.create(
46
  model="gpt-4o",
47
  messages=[
@@ -57,7 +60,7 @@ if st.button("Enhance Prompt"):
57
  {"role": "user", "content": user_prompt}
58
  ],
59
  temperature=0.3, # Low randomness
60
- max_tokens=min(len(user_prompt) * 1.5, 500) # Limit expansion
61
  )
62
  enhanced_prompt = completion.choices[0].message.content.strip()
63
 
 
40
  if user_prompt.strip() == "":
41
  st.warning("⚠️ Please enter a prompt before enhancing.")
42
  else:
 
43
  try:
44
+ # Ensure max_tokens is an integer
45
+ max_tokens = int(min(len(user_prompt) * 1.5, 500)) # Convert to integer
46
+
47
+ # Generate enhanced prompt using OpenAI API
48
  completion = client.chat.completions.create(
49
  model="gpt-4o",
50
  messages=[
 
60
  {"role": "user", "content": user_prompt}
61
  ],
62
  temperature=0.3, # Low randomness
63
+ max_tokens=max_tokens # Ensure integer value
64
  )
65
  enhanced_prompt = completion.choices[0].message.content.strip()
66