def visualize()

in opendataqna.py [0:0]

• 19 lines of code
• 7 McCabe index (conditional complexity)

def visualize(session_id,user_question,generated_sql,sql_results):
    try:
        Rewriter=ResponseAgent('gemini-1.5-pro')
        
        if session_id is None or session_id=="":
            print("This is a new session")
        else:
            session_history =firestoreconnector.get_chat_logs_for_session(session_id) if USE_SESSION_HISTORY else None
            if session_history is None or not session_history:
                print("No records for the session. Not rewriting the question\n")
            else:
                concated_questions,re_written_qe=Rewriter.rewrite_question(user_question,session_history)
                user_question=re_written_qe
        
        _viz=VisualizeAgent()
        js_chart = _viz.generate_charts(user_question, generate_sql,sql_results)
        invalid_response=False

    except Exception as e: 
        js_chart="Issue was encountered while Generating Charts ::"  + str(e)
        invalid_response=True

    return js_chart,invalid_response