Update app.py

app.py

@@ -65,35 +65,38 @@ max_phrase_length = max(len(phrase) for phrase, _ in replacements)
 MIN_FLUSH_SIZE = max(50, max_phrase_length * 2)
 
 def think(request):
-  question = request
 
-  url = "https://openrouter.ai/api/v1/chat/completions"
-  headers = {
+    url = "https://openrouter.ai/api/v1/chat/completions"
+    headers = {
       "Authorization": f"Bearer {openai_api_key}",
       "Content-Type": "application/json"
-  }
+    }
 
-  def do_req(model, content, include_reasoning=False, reasoning=""):
-      messages = [{"role": "user", "content": content}]
+    def do_req(model, content, include_reasoning=False, reasoning=""):
+        messages = content
 
-      if reasoning != "":
-          messages.append({"role": "assistant", "content": "<think>\n" + reasoning + "</think>\n"})
+        if messages[-1]["role"] == "user":
+            messages[-1]["content"] += " Please think this through, but don't output an answer."
+      
+        #messages = [{"role": "user", "content": content}]
 
-      payload = {
-          "model": model,
-          "messages": messages,
-          "include_reasoning": include_reasoning,
-          "stop": "</think>"
-      }
+        #if reasoning != "":
+        #    messages.append({"role": "assistant", "content": "<think>\n" + reasoning + "</think>\n"})
+        print(messages)
+        payload = {
+            "model": model,
+            "messages": messages,
+            "include_reasoning": include_reasoning,
+            "stop": "</think>"
+        }
 
-      return requests.post(url, headers=headers, data=json.dumps(payload))
+        return requests.post(url, headers=headers, data=json.dumps(payload))
 
 
-  # R1 will reliably return "done" for the content portion of the response
-  content = f"{question} Please think this through, but don't output an answer"
-  reasoning_response = do_req("deepseek/deepseek-r1", content, True)
-  reasoning = reasoning_response.json()['choices'][0]['message']['reasoning']
-  return reasoning
+    # R1 will reliably return "done" for the content portion of the response
+    reasoning_response = do_req("deepseek/deepseek-r1", request, True)
+    reasoning = reasoning_response.json()['choices'][0]['message']['reasoning']
+    return reasoning
 
 def apply_replacements(text):
     """