carrief0908 commited on
Commit
f3ed951
·
verified ·
1 Parent(s): 8a60869

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -14
src/streamlit_app.py CHANGED
@@ -68,19 +68,6 @@ if 'user_requirements' not in st.session_state:
68
  st.session_state.user_requirements = {}
69
 
70
 
71
- # Helper function to call Claude API
72
- def call_claude(prompt, system_prompt="You are an expert software procurement consultant."):
73
- try:
74
- client = anthropic.Anthropic(api_key=st.session_state.api_key)
75
- message = client.messages.create(
76
- model="claude-3-5-haiku-latest",
77
- max_tokens=4096,
78
- system=system_prompt,
79
- messages=[{"role": "user", "content": prompt}]
80
- )
81
- return message.content[0].text
82
- except Exception as e:
83
- return f"Error: {str(e)}"
84
 
85
  # Sample software database
86
  SOFTWARE_DATABASE = [
@@ -201,7 +188,7 @@ if not st.session_state.software_database:
201
 
202
  def call_openai(prompt, system_prompt="You are an expert software procurement consultant."):
203
  try:
204
- client = OpenAI()
205
  completion = client.chat.completions.create(
206
  model="gpt-4o-mini", # or "gpt-4o" if you have access
207
  messages=[
 
68
  st.session_state.user_requirements = {}
69
 
70
 
 
 
 
 
 
 
 
 
 
 
 
 
 
71
 
72
  # Sample software database
73
  SOFTWARE_DATABASE = [
 
188
 
189
  def call_openai(prompt, system_prompt="You are an expert software procurement consultant."):
190
  try:
191
+ client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
192
  completion = client.chat.completions.create(
193
  model="gpt-4o-mini", # or "gpt-4o" if you have access
194
  messages=[