Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,7 +1,7 @@
 import streamlit as st 
 from pydantic import BaseModel, Field
 from typing import List, Literal, Dict, Optional 
-from agency_swarm import Agent, Agency, set_openai_key, Basetool
+from agency_swarm import Agent, Agency, set_openai_key, basetool
 
 
 class AnalyzeProjectRequirement(BaseTool):
@@ -67,7 +67,7 @@ class AnalyzeProjectRequirement(BaseTool):
 
 def init_session_state()-> None:
     """Initiate session state variable"""
-    if "messages" is not st.session_state:
+    if "messages" not in st.session_state:
         st.session_state.message=[]
     if "api_key" not in st.session_state:
         st.session_state.api_key= None
@@ -78,22 +78,6 @@ def main() -> None:
 
     st.title("Greg.ai Services- Innovate Build, and Lighten up the future with AI")
 
-    ##API configuration
-
-    with st.sidebar:
-        st.header("API Configuration")
-        openai_api_key= st.text_input(
-            "OpenAI API Key",
-            type= "password",
-            help= "Enter your OpenAI Key that starts with sk-xxxxxxxxx"
-        )
-
-        if openai_api_key:
-            st.session_state_api_key= openai_api_key
-            st.success("API Key accepted")
-        else: 
-            st.warning("⚠️ API Key not found, please enter your key to proceed")
-            st.markdown("[Get your API key here](https://platform.openai.com/api-keys)")
-            return
-    
-    
+
+if __name__ == "__main__":
+    main()