.

bill.py

@@ -60,7 +60,7 @@ def check_related_keys(question, user_id):
         bill_keys.update(bill.keys())
     return [key for key in bill_keys if key.lower() in question.lower()]
 
-def process_query(query, user_id):
+def process_query(query, user_id, model):
     user_data = load_user_data(user_id)
     bill_info = user_data.get("bills", [])
     related_keys = check_related_keys(query, user_id)
@@ -85,12 +85,21 @@ def process_query(query, user_id):
         st.warning("Prea multe caractere în context, solicitarea nu va fi trimisă.")
         return None
 
+    # Update this part to run the chosen model
+    if model == "4o":
+        # Code to run model 4o
+        st.write("Running model 4o")
+    elif model == "4o-mini":
+        # Code to run model 4o-mini
+        st.write("Running model 4o-mini")
+
     return context
 
 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"])
     if "user_id" not in st.session_state:
         st.session_state.user_id = None