Datastreaming with flask. about poe-api HOT 6 CLOSED

def chatbot2_response(query):
    max_retries = 20
    for attempt in range(1, max_retries + 1):
        try:
            for chunk in poe_client.send_message("capybara", query, with_chat_break=True):
                yield chunk["text_new"]
            break
        except Exception as e:
            logging.warning(f"SendMessageMutation returned an error: {e} | Retrying ({attempt}/{max_retries})")
            time.sleep(1)  # Add a sleep time to avoid overloading the server
    else:
        logging.error("SendMessageMutation failed after maximum retries")

@app.route("/chatbot2", methods=['GET', 'POST'])
def chatbot2():
    chatbot_2_question = flask.request.form.get('question')
    return Response(stream_with_context(chatbot2_response(chatbot_2_question)), mimetype='text/plain')

@app.route("/chatbot2", methods=['GET', 'POST'])
def chatbot2():
    chatbot_2_question = flask.request.form.get('question')
    def chatbot2_response(query):
        max_retries = 20
        for attempt in range(1, max_retries + 1):
            try:
                for chunk in poe_client.send_message("capybara", query, with_chat_break=True):
                    yield chunk["text_new"]
                break
            except Exception as e:
                logging.warning(f"SendMessageMutation returned an error: {e} | Retrying ({attempt}/{max_retries})")
                time.sleep(1)
        else:
            logging.error("SendMessageMutation failed after maximum retries")
    return Response(stream_with_context(chatbot2_response(chatbot_2_question)), mimetype='text/plain')

def chatbot2():
    chatbot_2_question = flask.request.form.get('question')
    def generator():
        for chunk in poe_client.send_message("capybara", chatbot_2_question):
            yield json.dumps(chunk)+"\n"
    return Response(generator(), content_type="text/event-stream")`````