.

bill.py

@@ -99,7 +99,7 @@ def main():
     st.title("Telecom Bill Chat with LLM Agent")
 
     # Create a sidebar menu to choose between models
-    model = st.sidebar.selectbox("Choose OpenAI Model", ["4o", "4o-mini"])
+    model_name = st.sidebar.selectbox("Choose OpenAI Model", ["gpt-4o", "gpt-4o-mini"])
     if "user_id" not in st.session_state:
         st.session_state.user_id = None
 
@@ -172,7 +172,7 @@ def main():
 
         # Now call GPT-4 with the entire conversation
         completion = client.chat.completions.create(
-            model="gpt-4",
+            model=model_name,
             messages=st.session_state["messages"]
         )
         response_text = completion.choices[0].message.content.strip()