georgeek commited on
Commit
d8d0296
·
1 Parent(s): cbc938b
Files changed (1) hide show
  1. bill.py +11 -2
bill.py CHANGED
@@ -60,7 +60,7 @@ def check_related_keys(question, user_id):
60
  bill_keys.update(bill.keys())
61
  return [key for key in bill_keys if key.lower() in question.lower()]
62
 
63
- def process_query(query, user_id):
64
  user_data = load_user_data(user_id)
65
  bill_info = user_data.get("bills", [])
66
  related_keys = check_related_keys(query, user_id)
@@ -85,12 +85,21 @@ def process_query(query, user_id):
85
  st.warning("Prea multe caractere în context, solicitarea nu va fi trimisă.")
86
  return None
87
 
 
 
 
 
 
 
 
 
88
  return context
89
 
90
  def main():
91
-
92
  st.title("Telecom Bill Chat with LLM Agent")
93
 
 
 
94
  if "user_id" not in st.session_state:
95
  st.session_state.user_id = None
96
 
 
60
  bill_keys.update(bill.keys())
61
  return [key for key in bill_keys if key.lower() in question.lower()]
62
 
63
+ def process_query(query, user_id, model):
64
  user_data = load_user_data(user_id)
65
  bill_info = user_data.get("bills", [])
66
  related_keys = check_related_keys(query, user_id)
 
85
  st.warning("Prea multe caractere în context, solicitarea nu va fi trimisă.")
86
  return None
87
 
88
+ # Update this part to run the chosen model
89
+ if model == "4o":
90
+ # Code to run model 4o
91
+ st.write("Running model 4o")
92
+ elif model == "4o-mini":
93
+ # Code to run model 4o-mini
94
+ st.write("Running model 4o-mini")
95
+
96
  return context
97
 
98
  def main():
 
99
  st.title("Telecom Bill Chat with LLM Agent")
100
 
101
+ # Create a sidebar menu to choose between models
102
+ model = st.sidebar.selectbox("Choose OpenAI Model", ["4o", "4o-mini"])
103
  if "user_id" not in st.session_state:
104
  st.session_state.user_id = None
105