Update app.py

app.py

@@ -22,7 +22,6 @@ warnings.filterwarnings("ignore", category=DeprecationWarning)
 # Load environment variables
 load_dotenv()
 GROQ_API_KEY=os.getenv("GROQ_API_KEY")
-print(GROQ_API_KEY)
 llm = ChatGroq(model="llama-3.1-70b-versatile")
 
 PROMPT_TEMPLATE = """
@@ -45,6 +44,10 @@ Instructions:
 Begin extraction.
 """
 def get_llm_response(entity, query):
+    # Ensure entity and query are not None
+    entity = entity or "Unknown Entity"
+    query = query or "Information not provided"
+    
     # Format the prompt with the entity and query
     formatted_prompt = PROMPT_TEMPLATE.format(entity=entity, query=query)
     
@@ -161,7 +164,7 @@ elif selected == "Define Query":
     if st.button("Extract Information"):
         if entity and query:
             response_text = get_llm_response(entity, query)
-            st.write(reponse_text)
+            st.write(response_text)
 
 # Extract Information Section with Progress Bar
 elif selected == "Extract Information":
@@ -181,9 +184,6 @@ elif selected == "Extract Information":
             user_message = st.session_state["query_template"].replace("{company}", str(entity))
             formatted_prompt = PROMPT_TEMPLATE.format(entity=entity, query=user_message)
             
-            # Append user message to the flow history
-            st.session_state.flowmessages.append(HumanMessage(content=user_message))
-
             # Generate response from the model
             response = llm([SystemMessage(content=formatted_prompt)])