Update app.py

app.py

@@ -73,14 +73,23 @@ Follow up: Could you try to increase the shared memory and try setting num_worke
 Intermediate Answer: It worked! Can you explain what happened here? 
 So the final answer is: The error usually means that your system doesn’t provide enough shared memory for multiple workers (used via num_workers>0). Check the shared memory limitation of your system and try to increase it.
 
-
 StackTrace: 
 
-RuntimeError: size mismatch (got input: [100000], target: [1000]) 
+Traceback (most recent call last):
+  File "main.py", line 39, in <module>
+    request = create_request(page)
+  File "main.py", line 15, in create_request
+    url = base_url + data
+TypeError: can only concatenate str (not "bytes") to str
 
-Question: Any idea how I can solve this problem?
-Are follow up questions needed here: No
-So the final answer is: I don’t know which line of code creates the shape mismatch but would guess it’s raised in the loss calculation: loss = criterion(output.view(-1), batch['input_ids'].view(-1).to(device)). Print the shapes of both tensors and make sure they are expected in the used criterion. PS you can post code snippets by wrapping them into three backticks ```, which would make your code easier to read.
+
+Question: How do I fix this? 
+Are follow up questions needed here: Yes
+Follow up: Could you try to decode the data before passing it to the url? 
+Intermediate Answer: Yes, it made the data a string and worked!
+So the final answer is: You can try to decode the data before passing it to the url like this: 
+
+data = urllib.parse.unquote(data)
 
 StackTrace: ''',
 '''
@@ -163,16 +172,37 @@ def initial_query_builder(language, code, question, intermediate = "\nIntermedia
   cur_prompt = prompt[0] +  language + prompt[1] + code + prompt[2] + question + prompt[3]
 
   # print("prompt: ", cur_prompt, end ='')
-
-  ret_text = call_gpt(cur_prompt, intermediate)
-  print("ret_text: ", ret_text)
-  print("get_last_line(ret_text): ", get_last_line(ret_text))
+  ## check if follow up in the query, if not, make sure it contains the final answer. otherwise re-run until at least one of the 2 is in the response. break after 3 attempts. 
+  attempts = 0
+  ret_text = ''
+  while followup not in ret_text and finalans not in ret_text:
+    attempts +=1 
+    ret_text = call_gpt(cur_prompt, intermediate)
+    print(str(attempts) + " ret_text:", ret_text)
+    if attempts == 3: 
+      break
+  if "final answer is" in ret_text: 
+    updated_prompt = cur_prompt + re.findall(r".*?(?=is:)", ret_text)[0] + " is: Let's think step-by-step. "
+    ret_text = call_gpt(updated_prompt, intermediate)
   return ret_text
 
 def subsequent_query_builder(curr_prompt, external_answer, intermediate = "\nIntermediate Answer:", followup = "\nFollow up:", finalans= '\nSo the final answer is:'):
+  print("curr_prompt: ", curr_prompt)
   curr_prompt += intermediate + ' ' + external_answer + '.'
-  print(intermediate + ' ' + yellowfy(external_answer) + '.', end='' )
-  ret_text = call_gpt(curr_prompt, intermediate)
+   ## check if follow up in the query, if not, make sure it contains the final answer. otherwise re-run until at least one of the 2 is in the response. break after 3 attempts. 
+  attempts = 0
+  ret_text = ''
+  while followup not in ret_text and finalans not in ret_text:
+    attempts +=1 
+    ret_text = call_gpt(curr_prompt, intermediate)
+    print("subsequent query " + str(attempts) + " ret_text:", ret_text)
+    if attempts == 3: 
+      break
+  print("ret_text: ", ret_text)
+  if "final answer is" in ret_text: 
+    updated_prompt = curr_prompt + re.findall(r".*?(?=is:)", ret_text)[0] + " is: Let's think step-by-step. "
+    # print("updated_prompt: ", updated_prompt)
+    ret_text = call_gpt(updated_prompt, intermediate)
   return ret_text
 
 """subsequent query builder: 
@@ -196,7 +226,7 @@ def prompt_builder(history, intermediate = "\nIntermediate Answer:", followup =
   curr_prompt = prompt[0] +  language + prompt[1] + stacktrace + prompt[2] + question + prompt[3]
 
   #set subsequent conversation thread
-  if len(history) > 2: #subsequent conversations have occurred 
+  if len(history) >= 2: #subsequent conversations have occurred 
     curr_prompt += history[1][1] ## get the first response to the stacktrace prompt
     for conversation in history[2:]:
       #grab intermediate answer