mirror of
				https://github.com/MindWorkAI/AI-Studio.git
				synced 2025-11-04 12:20:21 +00:00 
			
		
		
		
	Implemented the base64 image encoding
This commit is contained in:
		
							parent
							
								
									657fda4961
								
							
						
					
					
						commit
						2af75aed5f
					
				@ -49,4 +49,62 @@ public sealed class ContentImage : IContent
 | 
				
			|||||||
    /// The image source.
 | 
					    /// The image source.
 | 
				
			||||||
    /// </summary>
 | 
					    /// </summary>
 | 
				
			||||||
    public required string Source { get; set; }
 | 
					    public required string Source { get; set; }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    /// <summary>
 | 
				
			||||||
 | 
					    /// Read the image content as a base64 string.
 | 
				
			||||||
 | 
					    /// </summary>
 | 
				
			||||||
 | 
					    /// <remarks>
 | 
				
			||||||
 | 
					    /// The images are directly converted to base64 strings. The maximum
 | 
				
			||||||
 | 
					    /// size of the image is around 10 MB. If the image is larger, the method
 | 
				
			||||||
 | 
					    /// returns an empty string.
 | 
				
			||||||
 | 
					    ///
 | 
				
			||||||
 | 
					    /// As of now, this method does no sort of image processing. LLMs usually
 | 
				
			||||||
 | 
					    /// do not work with arbitrary image sizes. In the future, we might have
 | 
				
			||||||
 | 
					    /// to resize the images before sending them to the model.
 | 
				
			||||||
 | 
					    /// </remarks>
 | 
				
			||||||
 | 
					    /// <param name="token">The cancellation token.</param>
 | 
				
			||||||
 | 
					    /// <returns>The image content as a base64 string; might be empty.</returns>
 | 
				
			||||||
 | 
					    public async Task<string> AsBase64(CancellationToken token = default)
 | 
				
			||||||
 | 
					    {
 | 
				
			||||||
 | 
					        switch (this.SourceType)
 | 
				
			||||||
 | 
					        {
 | 
				
			||||||
 | 
					            case ContentImageSource.BASE64:
 | 
				
			||||||
 | 
					                return this.Source;
 | 
				
			||||||
 | 
					            
 | 
				
			||||||
 | 
					            case ContentImageSource.URL:
 | 
				
			||||||
 | 
					            {
 | 
				
			||||||
 | 
					                using var httpClient = new HttpClient();
 | 
				
			||||||
 | 
					                using var response = await httpClient.GetAsync(this.Source, HttpCompletionOption.ResponseHeadersRead, token);
 | 
				
			||||||
 | 
					                if(response.IsSuccessStatusCode)
 | 
				
			||||||
 | 
					                {
 | 
				
			||||||
 | 
					                    // Read the length of the content:
 | 
				
			||||||
 | 
					                    var lengthBytes = response.Content.Headers.ContentLength;
 | 
				
			||||||
 | 
					                    if(lengthBytes > 10_000_000)
 | 
				
			||||||
 | 
					                        return string.Empty;
 | 
				
			||||||
 | 
					                    
 | 
				
			||||||
 | 
					                    var bytes = await response.Content.ReadAsByteArrayAsync(token);
 | 
				
			||||||
 | 
					                    return Convert.ToBase64String(bytes);
 | 
				
			||||||
 | 
					                }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					                return string.Empty;
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            case ContentImageSource.LOCAL_PATH:
 | 
				
			||||||
 | 
					                if(File.Exists(this.Source))
 | 
				
			||||||
 | 
					                {
 | 
				
			||||||
 | 
					                    // Read the content length:
 | 
				
			||||||
 | 
					                    var length = new FileInfo(this.Source).Length;
 | 
				
			||||||
 | 
					                    if(length > 10_000_000)
 | 
				
			||||||
 | 
					                        return string.Empty;
 | 
				
			||||||
 | 
					                    
 | 
				
			||||||
 | 
					                    var bytes = await File.ReadAllBytesAsync(this.Source, token);
 | 
				
			||||||
 | 
					                    return Convert.ToBase64String(bytes);
 | 
				
			||||||
 | 
					                }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					                return string.Empty;
 | 
				
			||||||
 | 
					            
 | 
				
			||||||
 | 
					            default:
 | 
				
			||||||
 | 
					                return string.Empty;
 | 
				
			||||||
 | 
					        }
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
		Loading…
	
		Reference in New Issue
	
	Block a user