Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- 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=[
|