Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,7 +2,7 @@ import asyncio
 import streamlit as st
 from typing import Dict, Any, List
 from agents import Agent, Runner, trace, set_default_openai_key
-from firecrwal import FirecrawlApp
+from firecrawl import FirecrawlApp
 from agents.tool import function_tool
 
 #Setup page configuration using streamlit
@@ -11,3 +11,31 @@ st.set_page_config(
     page_icon = "",
     layout = "wide"
 )
+
+if "openai_api_key" not in st.session_state:
+    st.session_state.openai_api_key = ""
+if "firecrawl_api_key" not in st.session_state:
+    st.session_state.firecrawl_api_key = ""
+
+with st.sidebar:
+    st.title("API Configuration")
+    openai_api_key = st.text_input(
+        "OpenAI API key",
+        value = st.session_state.openai_api_key,
+        type = "password"
+    )
+
+    firecrawl_api_key = st.text_input(
+        "OpenAI API key",
+        value = st.session_state.firecrawl_api_key,
+        type = "password"
+    )
+
+    if openai_api_key:
+        st.session_state.openai_api_key = openai_api_key
+        set_default_openai_key(openai_api_key)
+    if firecrawl_api_key:
+        st.session_state.firecrawl_api_key = firecrawl_api_key
+        set_default_openai_key(firecrawl_api_key)
+    
+