mrciomnl commited on
Commit
194b21f
·
1 Parent(s): 8dffd49

fixed some bugs

Browse files
Files changed (1) hide show
  1. app.py +15 -10
app.py CHANGED
@@ -1,10 +1,15 @@
1
  import openai
2
  import streamlit as st
3
- from openai import OpenAI
4
  import os
5
 
6
  # Set OpenAI API Key
7
- client = OpenAI(api_key=os.getenv("OPENAI_API_KEY", "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"))
 
 
 
 
 
8
 
9
 
10
  # Streamlit UI
@@ -25,8 +30,8 @@ if st.button("Generate Text"):
25
  if not prompt.strip():
26
  st.error("Please enter a prompt.")
27
  else:
 
28
  try:
29
- full_prompt = f"Generate a {tone.lower()} {output_format.lower()} based on: {prompt}"
30
  response = client.chat.completions.create(
31
  model="gpt-3.5-turbo",
32
  messages=[{"role": "user", "content": full_prompt}],
@@ -34,13 +39,13 @@ if st.button("Generate Text"):
34
  max_tokens=max_length,
35
  n=n_responses
36
  )
37
-
38
- for i, choice in enumerate(response.choices):
39
- st.subheader(f"Response {i+1}")
40
- st.write(choice.message.content)
41
-
42
- except openai.OpenAIError as e:
43
- st.error(f"API Error: {e}")
44
 
45
  # Feedback Section
46
  st.subheader("Feedback")
 
1
  import openai
2
  import streamlit as st
3
+ from openai import OpenAI, OpenAIError
4
  import os
5
 
6
  # Set OpenAI API Key
7
+ api_key = os.getenv("OPENAI_API_KEY")
8
+ if not api_key:
9
+ st.error("API key is missing. Set OPENAI_API_KEY as an environment variable.")
10
+ st.stop()
11
+
12
+ client = OpenAI(api_key=api_key)
13
 
14
 
15
  # Streamlit UI
 
30
  if not prompt.strip():
31
  st.error("Please enter a prompt.")
32
  else:
33
+ full_prompt = f"{prompt}\n\nFormat: {output_format}\nTone: {tone}\nCreative Mode: {creative_mode}"
34
  try:
 
35
  response = client.chat.completions.create(
36
  model="gpt-3.5-turbo",
37
  messages=[{"role": "user", "content": full_prompt}],
 
39
  max_tokens=max_length,
40
  n=n_responses
41
  )
42
+ except OpenAIError as e:
43
+ if "insufficient_quota" in str(e):
44
+ st.error("You have exceeded your API quota. Please check your OpenAI account billing and usage.")
45
+ elif "model_not_found" in str(e):
46
+ st.error("The specified AI model does not exist or you do not have access to it.")
47
+ else:
48
+ st.error("An error occurred while generating text. Please try again later.")
49
 
50
  # Feedback Section
51
  st.subheader("Feedback")