Debugging shit

app.py

@@ -60,47 +60,6 @@ def verify_token(token: str) -> bool:
     except jwt.PyJWTError:
         return False
 
-def stream_solution(instruction: str, token: str):
-    if not verify_token(token):
-        return {"error": "Invalid token"}
-    
-    system_prompt = "You are a Python coding assistant specialized in solving LeetCode problems. Provide only the complete implementation of the given function. Ensure proper indentation and formatting. Do not include any explanations or multiple solutions."
-    full_prompt = f"""### Instruction:
-{system_prompt}
-
-Implement the following function for the LeetCode problem:
-
-{instruction}
-
-### Response:
-Here's the complete Python function implementation:
-
-```python
-"""
-    
-    async def generate():
-        generated_text = ""
-        try:
-            for chunk in llm(full_prompt, stream=True, **generation_kwargs):
-                token = chunk["choices"][0]["text"]
-                generated_text += token
-                yield token
-
-                # Optionally send progress updates:
-                progress = len(generated_text) / len(full_prompt)
-                yield {"progress": progress}  # Send a progress update
-
-            formatted_code = extract_and_format_code(generated_text)
-            yield formatted_code
-        except Exception as e:
-            # Handle exceptions and return an error message
-            yield {"error": str(e)}
-
-    logger.info(StreamingResponse(generate(), media_type="application/x-ndjson"))
-    logger.info(StreamingResponse(generate(), media_type="text/plain"))
-    logger.info(StreamingResponse(generate(), media_type="application/json"))
-    return StreamingResponse(generate(), media_type="text/plain")
-
 def extract_and_format_code(text):
     # Extract code between triple backticks
     code_match = re.search(r'```python\s*(.*?)\s*```', text, re.DOTALL)
@@ -185,6 +144,7 @@ Here's the complete Python function implementation:
         formatted_code = extract_and_format_code(generated_text)
         yield json.dumps({"complete": True, "formatted_code": formatted_code}) + "\n"
 
+    logger.info("Streaming response")
     return StreamingResponse(generate(), media_type="application/x-ndjson")
 
 def random_problem(token: str) -> Dict[str, Any]: