From 8bcfe38c601da406aa4163127a2b64e3225bf2fe Mon Sep 17 00:00:00 2001
From: Tabalbar <35385849+Tabalbar@users.noreply.github.com>
Date: Wed, 10 Jul 2024 15:30:38 -1000
Subject: [PATCH] save before user study sessions

---
 src/__pycache__/main.cpython-39.pyc | Bin 8299 -> 8833 bytes
 src/main.py                         |  33 ++++++++++++++++++++++------
 2 files changed, 26 insertions(+), 7 deletions(-)

diff --git a/src/__pycache__/main.cpython-39.pyc b/src/__pycache__/main.cpython-39.pyc
index e177fadfb9eac59948c6073b06ae61fba11a35e2..ce84e4bae5bcac0e2c564eac3f7716a5b39aa621 100644
GIT binary patch
delta 1483
zcmZuxZ)hAv6rY*fY%YK9Ztvgj{lC4tv@uPa-d$5G!GhH^&{ip!V4*1C>Tbd{>7JR}
z#3am;N))728y$o|FR<q)5wvJT&<|ok6r>=Hif|yJep3-d^rQIZaw&m!W^eY*oA-O~
z_j}CV?~lE7+F$YcdJ&8Z$8XGhc=Q9m(vNp|w*S*)kJ0OP+kqy&iHv?mZlWz|9b;r8
z!^d0w`Lzv9i?H$UI+;(rnd~a8uZf=}j?eMiLE$yJ=Lp0;R7H~l9OxubI`N@Se85p`
zw2q)akOkau6Wv2D11i|5(ncF0BW#?CG>8+ZA}44^M455)+$NL_Ibl0$MAz|+gcGq<
zS7VVTcA^)Mfh3sg9*nF|32or|$9&$GU5l}pp~jILH#8&GBuybgbA>k^M`E{u7w$0C
zjTwo(vyP=Q%}6!~3$pN|BI;$}5*NETGE#Bms2w{EwgwV1yAp&%Xj6jqF||q7fO5h~
zg62d=FQG03XFGbqyFdH^(VP4GI*XhXOF%0gchW4))Fre*=r2G&$<ok<49gh$oCka@
zmI9F)dn&W5(vm?bS@>h`4fZ=gBgazf1PFAyDd3xG5+IOubmn!_uus5Va&k`IDL4Zx
z=!UuL*OK{07mK%h;#tV=y8<3M#drJ#T;TuuPZkCmA!pFe+3Mvy8?+0?f!9b|YMc!k
z2ODG(4GnHT6L=95YlKfL*Vc}nrnE-&s-@4Gwq3Pm^}1cLt2L{BShwklaMUbQzfz;7
zZdb3Et*(jcvT3YphhR*a^XBty(0Ii*>%z4<bYDV4oE_Hdre%n9mHBzS?OIKii%*tE
zj}%MrOcje`?PsLe9v8)_I6QHrRGcc6#>&O9<D(;wlpj7(9=$k3T0Ov%+M;f&8!`)n
z<KQduB7AdniC+)qj;)sN<5ID0TD`Q`CiEbA{JmfUuky2@Ma=l;p-5sG(zQbOU{9Ns
zdd&hEbY=U`Pz+1boVmiK$P&LA`5Pd1bWGW0AaI_g3Z>WiXVG)=mY}u518Nk~A5dS#
zU-S1>eTXEmm-sz0@nbJcKTL^uVG3VJ_z1z|Ke>zls^$+WoiL>nCSk9q7x%(X0SH)F
z!4bl9+DUwSds;h><<~{oYy6{_+4r_Ed&JV@1MwVAZcoIAG2Y-bF{K^D2-W-H5Rpim
ziQq!-@*fjl^?fTa--+dWek+;66Z}qcO!@&F{AW2D-T`A=z$GxeSi*FNms5Yqn*y=J
zhtj(G4VW!#wdC5&Ox?8il1--z$#Vj!z_}<a!Bk*EQ!xFBZ>3%NOOfRZ{zPUtD73Vs
zSz245;-8grzLq%z{NK%-#c%L`GFk0S5h0&{!L&}D5w~KgN^2H<iw|cP(zgX<N-THV
kQMF5#;qFqw9!h<ZE#j;Ex9p-Hg}peYd4o!~x33rf4=3P<cK`qY

delta 1053
zcmZ8gU1%It6uxJ&olKkk|C!nO*`3{NA!;79ico^VSX5980b44FW!ZE#VH22j@62up
z-mz%JAG9`jP!txOMIU`B>61u6MCyaKTEvG%p@R4#f`#Ivc<v^x;#@fM-S3=pzwgXt
zUV7~JrF1KmI>PA5zwp-TZ}XSZ*(BVLZt^oo;;rQD8GfU~PO*@Ascirz)Hp9@M(~^x
zWuk1zwi&Q6Cn)DlZzwy!)3Un5qFh*z?4#`L`&65c3SrSpZ^K0^Du$(4$InwHoWvhf
z?s-F$y^P7?vX}L8BV{C2j99XHlgZr<^j3r(8(!gd?f~h+@QOo4<V4{iS!I!~DR(CG
zN+yf+1FRD^XaYk{@PE(e$@HJnG>vA88KI99GHgZm8jBoZ(Eul+il_*Ez;+aVf{a+A
zGJ>cos$O|xins;gkX=Jm_wD*BSyp-{Z@&p^ByI|4TcOdWW1Cc*iHaN5grUU_5f%}N
zs2<Hmjp(Sz#T8u27`8P$B+SWlO7G?Wd}oDUMxJSajqha^8_qBv&4u-_v_325!iHD5
zpiJ({6mwp6sGMSVySsm7HUWRmIe08@;#c`sR8CjU16(fD@#De}HgT!gzjuqsaoTHz
z?cnKlE9m+ie~tGgnQw(5@2vF0_P>e2?yto%prVI|5*zKU0Q>qa*ur&VA-jLNY@8aj
zIDZ2#8&5eONXdaDGa+yJL6hIn*hIVhI(&+slxw9aO-WCwN)i&vK1vh9KgzSJ1X)1W
zECJwy<|6Fv2IexTZ_0yT!>_IO%tsO)mf=G@Zr4HEeb$}_xPY%Zr;Qpg79=Ujlp<)8
zl?HwlZ#q}*`cC$IFT-^i>^s3BG}fBUj^7EJ%?I#CWg+ncarlc`DqJIQOrDw`pg{uo
zHGI5!Q+=1V1Q*?!{tdw~_+z!Zx*D{@0}Tw^hJ9RiW+^ZIBxq%6S>%iVgqPe{{ZgLt
z1wK=o&&d(Ti8bEsarw9M9mLu_G}d12Nq8F%*InbHtWcMqZ~KotF`jPwgAVWd{2hF_
s-mC1%meVp^om6#K*4yVoF5Q#VxAptrMf|(oPijy$v|M&tn>hmi04I<YEdT%j

diff --git a/src/main.py b/src/main.py
index b0d1b34..23a3f56 100644
--- a/src/main.py
+++ b/src/main.py
@@ -152,7 +152,7 @@ async def createChartOptions(request: TextRequest):
   user_prompt = request.prompt #"""show me car brands vs price, sort by decending prices"""
   user_prompt_modified, user_prompt_reasoning = llm_re.prompt_reiterate(user_prompt, f"""For extra context, here is the current verbal context of the last few utterances: {" ".join(last_utterances)}. 
                                                  Here is is the current chart context of what charts were last created, selected, and iteracted by the user: {chart_context} """)
-
+  print(f'For extra context, here is the current verbal context of the last few utterances: {" ".join(last_utterances)}. Here is is the current chart context of what charts were last created, selected, and iteracted by the user: {chart_context}')
   print("****Selecting Stations****")
 
   stations, station_reasoning = llm_base.prompt_select_stations(user_prompt_modified)
@@ -161,7 +161,13 @@ async def createChartOptions(request: TextRequest):
   print()
   
   if len(stations) == 0:
-    return {}
+    print("Error in getting stations, trying one more time")
+    print(stations)
+    stations, station_reasoning = llm_base.prompt_select_stations(user_prompt_modified)
+    if(len(stations) == 0):
+      return {}
+    else:
+      print("Fixed")
   
   print("****Selecting Dates****")
   # print("****Starting Extracting Stations****")
@@ -170,8 +176,13 @@ async def createChartOptions(request: TextRequest):
   # print(dates_reasoning)
   print()
   
-  if len(stations) == 0:
-    return {}
+  if len(dates.keys()) == 0:
+    print("Error in getting dates, sending all dates")
+    print(dates)
+    dates = {
+      "startDate": "2024-01-01T00:00:00.000000Z",
+      "endDate": "2024-07-10T11:20:54.826724Z"
+    }
   
   station_info = {}
   for idx,id in enumerate(stations):
@@ -191,10 +202,17 @@ async def createChartOptions(request: TextRequest):
   print("****Selecting Attributes****")
   print()
   
-  if len(stations) == 0:
-    return {}
-  
   chosen_attributes_names, attrib_reasoning = llm_base.prompt_attributes(user_prompt_modified, available_variable_names)
+  
+  if len(chosen_attributes_names) == 0:
+    print("Error in getting attributes, trying one more time")
+    print(chosen_attributes_names)
+    chosen_attributes_names, attrib_reasoning = llm_base.prompt_attributes(user_prompt_modified, available_variable_names)
+    if len(chosen_attributes_names) == 0:
+      return {}
+    else:
+      print("Fixed")
+  
   chosen_attribute_ids = []
   for attr in chosen_attributes_names:
       # print(attr, "--------------------", available_variable_names)
@@ -212,6 +230,7 @@ async def createChartOptions(request: TextRequest):
   print("****Selecting Transformations****", chosen_attribute_ids)
   transformations, trans_reasoning = llm_transform.prompt_transformations(user_prompt_modified, chosen_attribute_ids)
   chartType, chart_frequencies, chart_reasoning, chart_scope = llm.prompt_charts_via_chart_info(user_prompt_modified, chosen_attributes_names)
+  
   for id in station_info.keys():
     station_chart_info[id] = {'attributes': chosen_attribute_ids, 'transformations': transformations, 'chartType': chartType, 'available_attribute_info': station_info[id], 'dates': dates} #TODO check if values exist
   # print(f"************Generated a {station_chart_info}**************")