ږJh*ddlmZddlmZddlmZmZddlZddlZddl Z ddl Z ddl Z e je jde jeZ eredk(rej#ddZnee Zej'd d ed eefdZded eefdZded efdZddedeed eefdZy#e$r Zej#d edZYdZ[PdZ[wwxYw))OpenAI)OPENAI_API_KEY)OptionalListNz4%(asctime)s - %(name)s - %(levelname)s - %(message)s)levelformatyour_openai_api_key_hereuA⚠️ Invalid or missing OPENAI_API_KEY in environment variables)api_keyz&OpenAI client initialized successfullyz"Error initializing OpenAI client: image_urlreturnc tj|}tj|jj dS#t $r"}tjd|Yd}~yd}~wwxYw)z Encode image from URL to base64.utf-8zError encoding image from URL: N) requestsgetbase64 b64encodecontentdecode Exceptionloggererror)r responsees 6D:\Projects\AI-TestCaseGenerator\ai\image_generator.pyencode_image_from_urlrs^<< * 0 0188AA  6qc:;sAA A0A++A0 image_pathc t|d5}tj|jj dcdddS#1swYyxYw#t $r"}t jd|Yd}~yd}~wwxYw)z'Encode image from local path to base64.rbrNz Error encoding image from path: )openrrreadrrrr)r image_filers rencode_image_from_pathr"#sr *d # Gz##JOO$56==gF G G G  7s;<s4 A2A  A AAA BA<<B test_typec ddddddddddddd }|j||}dd d d d d d d dddd dddd dddd dddd d}|j|i}|stjd|d|d|S)z.Get the configuration for a specific test typedashboard_functionaldashboard_negative dashboard_ui dashboard_uxdashboard_compatibilitydashboard_performance) r%r&r'r(r)r*zFunctional - Positive TestszFunctional - Negative TestzUI TestszUX TestszCompatibility TestszPerformance TestsTC_FUNCzEfunctional test cases focusing on valid inputs and expected behaviors)prefix descriptioncountTC_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(r)r*zUnknown test type: z (mapped to ))rrwarning)r#form_to_internal internal_type base_configsconfigs rget_test_type_configr<,s!72&&#M b! k Q a "]$  ]" 5LB  mR 0F ,YK|M?RSTU Mselected_typesc <|stjdy|rt|dk(rtjdyt"d}tj|t |tj d|tj d|gd} t jj|s%d |}tj|t|t|}|stjd yg}|D]:}tj d |t|}|stjd |Bd |dd|d|dd|dd }d} d} |D]} | rn tj d| d|dtjjj| ddddd|dddd |id!gdgd"d#$} | j dj"j$j'} | r6tj d%|d&| d'|d(| } |j)| ntjd)|d&| | s| rd,}d-| vsd.| vr=d/|d+| }tj|tjd0t d1d2| j/vsd3| j/vr=d4|d+| }tj|tjd5t d6d7|d8| }tj|t d9| tj d:|=|stjd;ytj dt-|d?@t dAt-|d}~wwxYw)Bz9Generate test cases from an image using OpenAI Vision APIz/No image path provided for test case generationNrz/No test types selected for test case generationzJOpenAI client not initialized. Please check your API key in the .env file.z,Generating test cases from image for types: z Image path: )zgpt-4ozgpt-4-vision-previewzgpt-4-turbo-previewzImage file not found at path: zFailed to encode imagez#Starting generation for test type: zSkipping unknown test type: z4 Analyze the image and generate EXACTLY r/ z' test cases. Focus ONLY on r.zM. Use this EXACT format for each test case: Title: r-ab_[Number]_[Brief_Title] Scenario: [Detailed scenario description based on the image] Steps to reproduce: 1. [Step 1] 2. [Step 2] ... Expected Result: [What should happen] Actual Result: [To be filled during execution] Priority: [High/Medium/Low] z1Sending request to OpenAI Vision API using model z for z test casessystemzDYou are a QA engineer generating test cases from the provided image.)rolerusertext)typerDr urlzdata:image/jpeg;base64,)rEr igffffff?)modelmessages max_tokens temperaturezSuccessfully 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 zXPlease check if your OpenAI account has access to GPT-4 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: T)exc_infoz*Failed to generate test cases from image: )rrlenclient ValueErrorinfoospathexistsFileNotFoundErrorr"r<r7chat completionscreatechoicesmessagerstripappendrstrlowerjoin)rr> error_msg vision_models base64_imageall_test_casesr#r;prompt test_cases last_errorrGrtest_cases_with_headerrs rgenerate_test_case_from_imagerjhs  FG S0A5 FG~`  Y## KK>~>NOP KK,zl+,NMxPww~~j)8 EI LL ##I. .-j9  LL1 2'^ LI KK=i[I J))4F!=i[IJ44:7O3DAi[Q!-0128$% & F"JJ'- )KK"STYSZZ_`i_jju vw &{{66==#)1-q )/-3V$D0;,15L\N3[6*%&,"  "$$($'+ > H0"*!1!1!!4!ykj\1Z.&--.DE)Ei[Phinho'pqQ- `* $ 26MQ[6["CI;m\f[g hILL+LL!{|$&QRR$ (8(8(::i:K[K[K]>]":9+]S]R^ _ILL+LL!YZ$%bcc"3I;>[\f[g hILL+$'Fzl%STT KK>ykJ K}^ L@ LL< = rvs*!   ',,/ef   8 $  ^/II XY/ <= SXc]sx}:C:D:xNPcNP49NPX`adXeNPe LL5aS9: Fs5B??C$CC$