U h:@sddlmZmZmZmZddlmZmZddlm Z ddl m Z ddl m Z ddlmZmZddlZddlZddlZddlZddlZejejdd eeZe d d Zed d ZeeedddZeeedddZee dddZ!edeeeeedddZ"dS))capture_exceptioncapture_messageset_tag set_context)monitor_openai_apimonitor_critical_system)OpenAI) ChatOpenAI)LangChainTracer)OptionalListNz4%(asctime)s - %(name)s - %(levelname)s - %(message)s)levelformatzopenai-cost-tracking) project_namec Cs^zddlm}|WStk rX}z(td|t|ddiWYdSd}~XYnXdS)z Get OpenAI API key from settingsr)OPENAI_API_KEYzError loading OpenAI API key: functionget_openai_api_keyN)config.settingsr Exceptionloggererrorr)rer?/var/www/html/testcasegenerator.evdpl.com/ai/image_generator.pyrs r) image_urlreturnc Cs\zt|}t|jdWStk rV}ztd|WYdSd}~XYnXdS)z Encode image from URL to base64.utf-8zError encoding image from URL: N) requestsgetbase64 b64encodecontentdecoderrr)rresponserrrrencode_image_from_url!s  r$) image_pathrc Csvz8t|d$}t|dW5QRWSQRXWn8tk rp}ztd|WYdSd}~XYnXdS)z'Encode image from local path to base64.rbrz Error encoding image from path: N)openrr readr"rrr)r% image_filerrrrencode_image_from_path*s  ,r*) test_typerc Csddddddddddddd }|||}dd d d d d d d ddd d ddd d ddd d ddd d d}||i}|std|d|d|S)z.Get the configuration for a specific test typedashboard_functionaldashboard_negative dashboard_ui dashboard_uxdashboard_compatibilitydashboard_performance) r,r-r.r/r0r1zFunctional - Positive TestszFunctional - Negative TestzUI TestszUX TestszCompatibility TestszPerformance TestsTC_FUNCzEfunctional test cases focusing on valid inputs and expected behaviors2)prefix description max_countTC_NEGzNnegative test cases focusing on invalid inputs, error handling, and edge casesTC_UIz4UI test cases focusing on visual elements and layoutTC_UXzDuser experience test cases focusing on user interaction and workflow TC_COMPATz@compatibility test cases across different browsers and platformsTC_PERFz@performance test cases focusing on load times and responsiveness)r,r-r.r/r0r1zUnknown test type: z (mapped to ))rrwarning)r+Zform_to_internalZ internal_type base_configsconfigrrrget_test_type_config3sV ! r@)r%selected_typesrc s|stddS|r"t|dkr0tddSt}|rJ|dksJ|dkr`d}t|t|td|td |d g}ztj|sd |}t|t |t |std WdSg}|D]X}td |t |}|s t d|qd|d|dd|dd|dd d}d} |D]} |rTq\ztd| d|dt | d|dtgdtddfdd } | } | j}|rtd!|d"| d#|d$|} || Wq\nt d%|d"| WnRtk rV}z2t|} t d&| d|d'| WYqBW5d}~XYnXqB|s | r d(}d)| ksd*| krd+|d'| }t|td,td-nrd.| ksd/| krd0|d'| }t|td1td2n(d3|d4| }t|td5| td6|q|sHtd7WdStd8t|d9d$d$|WStk r}z@tjd:t|dd;t||||d<td=t|W5d}~XYnXdS)>z9Generate test cases from an image using OpenAI Vision APIz/No image path provided for test case generationNrz/No test types selected for test case generationyour_openai_api_key_heremissing_api_keyuA⚠️ Invalid or missing OPENAI_API_KEY in environment variablesz,Generating test cases from image for types: z Image path: zgpt-4ozImage file not found at path: zFailed to encode imagez#Starting generation for test type: zSkipping unknown test type: zF Analyze the image thoroughly and generate test cases for z (up to r6z% maximum). Focus ONLY on r5am. IMPORTANT: Analyze the image content thoroughly and generate the appropriate number of relevant test cases. Consider the complexity and scope of the image - generate only what's truly needed. Do not force additional test cases just to reach the maximum. ANALYSIS REQUIREMENTS: - Analyze all visual elements: buttons, forms, text, images, icons, layout - Consider all possible user interactions visible in the image - Identify edge cases and boundary conditions - Consider different user roles and access levels if visible - Think about potential failure scenarios and error conditions - Analyze responsive design and cross-platform compatibility if applicable Use this format for each test case: Title: r4a_[Number]_[Brief_Title] Scenario: [Detailed scenario description covering all aspects visible in the image] Steps to reproduce: 1. [Step 1] 2. [Step 2] ... Expected Result: [What should happen] Priority: [High/Medium/Low] Ensure each test case covers a unique scenario and adds value. Focus on the most important and relevant test scenarios. z1Sending request to OpenAI Vision API using model z for z test casesg?i.)model temperatureopenai_api_key max_tokens callbacksT)criticalc s4ddddddddd id gdgS) NsystemzYou are a senior QA engineer generating test cases from the provided image. Analyze the image thoroughly and generate the appropriate number of test cases based on the image complexity. Focus on quality and relevance over quantity.)roler!usertext)typerMrurlzdata:image/jpeg;base64,)rNr)invokerZ base64_image current_llmpromptrrmake_openai_vision_callsz>generate_test_case_from_image..make_openai_vision_callzSuccessfully generated z test cases using model z TEST TYPE: z zReceived empty response for zError using model z test cases: model_not_foundinvalid_request_errorz!Error with all OpenAI models for zYPlease check if your OpenAI account has access to gpt-4o models with vision capabilities.ziThe OpenAI model required for image processing is not available. Please check your OpenAI account access. authorizationzapi keyzAuthorization error for z6Please check your OpenAI API key in the configuration.z;OpenAI API authorization failed. Please check your API key.zError generating z test cases with all models: zFailed to generate test cases: z$Completed generation for test type: z!Failed to generate any test casesz&Successfully generated test cases for z test typesz(Error in generate_test_case_from_image: )exc_info)r%rA vision_modelsz*Failed to generate test cases from image: )rrlenr ValueErrorinfoospathexistsFileNotFoundErrorr*r@r=r tracerrr!stripappendrstrlowerjoinr)r%rAapi_key error_msgrZall_test_casesr+r? test_cases last_errorrDrTr#test_cases_with_headerrrrQrgenerate_test_case_from_imageos        K               rn)N)#utils.error_loggerrrrrutils.error_monitorrropenairlangchain_openair %langchain.callbacks.tracers.langchainr typingr r rrr^syslogging basicConfigINFO getLogger__name__rrbrrer$r*dictr@rnrrrrs(        <