PDFOffice_Windows_exe/PDF Office/Helper/ChatGTPAIHelper.cs

using ComPDFKit_Conversion.Options;
using DryIoc;
using ImTools;
using Newtonsoft.Json;
using Newtonsoft.Json.Linq;
using PDF_Master.Properties;
using PDF_Master.ViewModels.Dialog.ChatGPTAIDialogs;
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Net;
using System.Net.Http;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading;
using System.Threading.Tasks;
using System.Web;
using static Dropbox.Api.Files.SearchMatchType;
using static Dropbox.Api.TeamLog.EventCategory;
using static Google.Apis.Requests.BatchRequest;

namespace PDF_Master.Helper
{
    public static class ChatGTPAIHelper
    {
#if DEBUG
        //测试环境
        //纠错重写
        private static string host = "https://ai.compdf.com";
        //百度翻译
        private static string translate = "http://101.132.103.13:8030";
#else
        //纠错重写
        private static string host ="http://prod-cn.your-api-server.com";
        //百度翻译
        private static string translate ="https://api-server.compdf.com";
#endif

        /// <summary>
        /// 翻译文档Key接口
        /// </summary>
        private static string Uri_fileKeytranslate = translate + "/v1/translate/fileUpload";

        /// <summary>
        /// 翻译文档接口
        /// </summary>
        private static string Uri_filetranslate = translate + "/v1/translate/fileTranslateHandle";

        /// <summary>
        /// 翻译文本接口
        /// </summary>
        private static string Uri_texttranslate = translate + "/v1/translate/textTrans";

        /// <summary>
        /// 错别字纠正
        /// </summary>
        private static string Uri_Correction = host + "/api/correct-typos";

        /// <summary>
        /// 错别字纠正
        /// </summary>
        private static string Uri_Rewrite = host + "/api/rewrite";


        #region AI服务器对接接口
        /// <summary>
        /// 翻译指定内容，可以指定语言
        /// </summary>
        /// <param name="content"></param>
        /// <param name="language">1=中文 2=英文 3=法语 4=汉语</param>
        /// <returns></returns>
        public async static Task<string> fileKeyTranslate(string content, string filename)
        {

            ServicePointManager.DefaultConnectionLimit = 200;
            HttpWebRequest request = (HttpWebRequest)HttpWebRequest.Create(Uri_fileKeytranslate);
            //关闭请求
            //弹窗内进度条关闭
            DocumentaryTranslationDialogViewModel.CancelProgress = () =>
            {
                if (request != null)
                {
                    request.Abort();
                }
            };
            //进度条关闭
            App.mainWindowViewModel.ProcessCloseAction = () =>
            {
                if (request != null)
                {
                    request.Abort();
                }
            };
            request.Method = "Post";
            //request.Accept = "application/vnd.api+json;version=1";
            request.UserAgent = "Apifox/1.0.0 (https://www.apifox.cn)";
            request.Timeout = 20000;
            request.ServicePoint.Expect100Continue = false;
            request.KeepAlive = false;
            var formData = new MultipartFormDataContent();
            byte[] data = File.ReadAllBytes(@content);
            MemoryStream stream = new MemoryStream(data);
            formData.Add(new ByteArrayContent(data), "file", filename);
            formData.Add(new StringContent("2"), "projectId");
            formData.Add(new StringContent("1.0.1"), "version");
            if (!string.IsNullOrEmpty(Settings.Default.UserDate.id))
            {
                formData.Add(new StringContent(Settings.Default.UserDate.id), "userId");
            }
            else
            {
                formData.Add(new StringContent("1"), "userId");
            }

            // 设置请求体格式
            request.ContentType = formData.Headers.ContentType.ToString();
            request.ContentLength = formData.Headers.ContentLength.Value;

            // 将FormData写入请求体
            using (var requestStream = request.GetRequestStream())
            {
                await formData.CopyToAsync(requestStream);
            }
            try
            {

                // 获取HTTP响应
                HttpWebResponse response2 = (HttpWebResponse)request.GetResponse();
                using (StreamReader reader = new StreamReader(response2.GetResponseStream()))
                {
                    string responseData = reader.ReadToEnd();
                    Console.WriteLine(responseData);
                    reader.Close();
                    JObject jobject = (JObject)JsonConvert.DeserializeObject(responseData);
                    if (response2 != null)
                    {
                        response2.Close();
                    }
                    if (request != null)
                    {
                        request.Abort();
                    }
                    string translation = "";
                    if (jobject["code"].ToObject<string>().ToLower() == "200")
                    {
                        translation = jobject["data"]["fileKey"].ToObject<string>().ToLower();
                        return translation;
                    }
                    return "Code" + jobject["code"].ToObject<string>().ToLower();


                }
            }
            catch (WebException ex)
            {
                Trace.WriteLine("HTTP异常：" + ex.Message);
                return "Code" + "300";
            }
        }

        public static string HttpDownloadFile(string url, string path)
        {
            // 设置参数
            HttpWebRequest request = WebRequest.Create(url) as HttpWebRequest;
            //发送请求并获取相应回应数据
            HttpWebResponse response = request.GetResponse() as HttpWebResponse;
            //直到request.GetResponse()程序才开始向目标网页发送Post请求
            Stream responseStream = response.GetResponseStream();

            //创建本地文件写入流
            Stream stream = new FileStream(path, FileMode.Create);

            byte[] bArr = new byte[1024];
            int size = responseStream.Read(bArr, 0, (int)bArr.Length);
            while (size > 0)
            {
                stream.Write(bArr, 0, size);
                size = responseStream.Read(bArr, 0, (int)bArr.Length);
            }
            stream.Close();
            responseStream.Close();
            return path;
        }

        

        /// <summary>
        /// 翻译指定内容，可以指定语言
        /// </summary>
        /// <param name="fileKey">文本Key</param>
        /// <param name="fromlanguage">文本语言</param>
        /// <param name="tolanguage">需要翻译的语言</param>
        /// <returns></returns>
        public static async Task<string> fileTranslate(string content, string fromlanguage, string tolanguage, bool isprocess = true)
        {
            intProcess(isprocess,"Translating", System.Windows.Visibility.Visible);
            
            FileInfo file = new FileInfo(content);
            string fileKey = await fileKeyTranslate(content, file.Name);
            App.mainWindowViewModel.Value = 1;
            if (fileKey.Contains("Code"))
            {
                clossProcess();
                return fileKey.Replace("Code", "");
            }
            HttpWebResponse response = null;
            ServicePointManager.DefaultConnectionLimit =200;
            HttpWebRequest request = (HttpWebRequest)HttpWebRequest.Create(Uri_filetranslate);
            //关闭请求
            DocumentaryTranslationDialogViewModel.CancelProgress=()=> 
            {
                if (request != null)
                {
                    request.Abort();
                }
            };

            App.mainWindowViewModel.ProcessCloseAction = () =>
            {
                if (request != null)
                {
                    request.Abort(); 
                }
            }; 
            request.Method = "Post";
            request.ContentType = "application/json";
            //request.Accept = "application/vnd.api+json;version=1";
            request.UserAgent = "Apifox/1.0.0 (https://www.apifox.cn)";
            //设置无限请求时间：由于文本翻译时间不确定且时间长
            request.Timeout = Timeout.Infinite;
            request.ServicePoint.Expect100Continue = false;
            request.KeepAlive = false;
            StringWriter sw = new StringWriter();
            using (JsonWriter writer = new JsonTextWriter(sw))
            {
                writer.WriteStartObject();
                writer.WritePropertyName("fileKey");
                writer.WriteValue(fileKey);
                writer.WritePropertyName("from");
                writer.WriteValue(fromlanguage);
                writer.WritePropertyName("to");
                writer.WriteValue(tolanguage);
                writer.WritePropertyName("projectId");
                writer.WriteValue("2");
                writer.WritePropertyName("version");
                writer.WriteValue("1.0.1");
                writer.WritePropertyName("userId");
                writer.WriteValue(Settings.Default.UserDate.id);
                writer.WriteEndObject();
            }
            try
            {
                App.mainWindowViewModel.Value = 2;
                string postBody = sw.ToString();
                using (StreamWriter writer = new StreamWriter(request.GetRequestStream()))
                {
                    writer.Write(postBody);
                    writer.Close();
                }
                App.mainWindowViewModel.Value = 3;
                response = (HttpWebResponse)request.GetResponse();
                using (StreamReader reader = new StreamReader(response.GetResponseStream()))
                {
                    App.mainWindowViewModel.Value = 4;
                    string responseData = reader.ReadToEnd();
                    Console.WriteLine(responseData);
                    reader.Close();
                    JObject jobject = (JObject)JsonConvert.DeserializeObject(responseData);
                    App.mainWindowViewModel.Value = 6;
                    if (response != null)
                    {
                        response.Close();
                    }
                    if (request != null)
                    {
                        request.Abort();
                    }

                    if (jobject["code"].ToObject<string>().ToLower() == "200")
                    {
                        translate = jobject["data"]["ossDownUrl"].ToObject<string>().ToLower();

                        //return jobject["code"].ToObject<string>().ToLower();
                        using (var client = new WebClient())
                        {
                            App.mainWindowViewModel.Value = 8;
                            string folderPath = file.FullName.Remove(file.FullName.LastIndexOf("."), file.FullName.Length - file.FullName.LastIndexOf(".")) + "_aiTranslation.pdf";
                            folderPath = CommonHelper.CreateFilePath(folderPath);
                            client.DownloadProgressChanged += (sender, e) =>
                            {

                            };
                            client.DownloadFile(translate, folderPath);
                            clossProcess();
                            return folderPath;
                        }
                    }
                    clossProcess();
                    return jobject["code"].ToObject<string>().ToLower();
                }
            }
            catch(WebException ex)
            {
                clossProcess();
                Trace.WriteLine("HTTP异常：" + ex.Message);
                return "300";
            }

        }


        /// <summary>
        /// 翻译指定内容，可以指定语言
        /// </summary>
        /// <param name="content">翻译内容</param>
        /// <param name="fromlanguage">文本语言</param>
        /// <param name="tolanguage">需要翻译的语言</param>
        /// <returns></returns>
        public static String textTranslate(string content, string fromlanguage, string tolanguage, ref string translate)
        {

            HttpWebResponse response = null;
            ServicePointManager.DefaultConnectionLimit = 200;
            HttpWebRequest request = (HttpWebRequest)HttpWebRequest.Create(Uri_texttranslate);
            request.Method = "Post";
            request.ContentType = "application/json";
            //request.Accept = "application/vnd.api+json;version=1";
            request.UserAgent = "Apifox/1.0.0 (https://www.apifox.cn)";
            request.Timeout = 20000;
            request.ServicePoint.Expect100Continue = false;
            StringWriter sw = new StringWriter();
            using (JsonWriter writer = new JsonTextWriter(sw))
            {
                writer.WriteStartObject();
                writer.WritePropertyName("q");
                writer.WriteValue(content);
                writer.WritePropertyName("from");
                writer.WriteValue(fromlanguage);
                writer.WritePropertyName("to");
                writer.WriteValue(tolanguage);
                writer.WritePropertyName("projectId");
                writer.WriteValue("2");
                writer.WritePropertyName("version");
                writer.WriteValue("1.0.1");
                writer.WritePropertyName("userId");
                writer.WriteValue(Settings.Default.UserDate.id);
                writer.WriteEndObject();
            }
            try
            {
                string postBody = sw.ToString();
                using (StreamWriter writer = new StreamWriter(request.GetRequestStream()))
                {
                    writer.Write(postBody);
                    writer.Close();
                }

                response = (HttpWebResponse)request.GetResponse();
                using (StreamReader reader = new StreamReader(response.GetResponseStream()))
                {
                    string responseData = reader.ReadToEnd();
                    Console.WriteLine(responseData);
                    reader.Close();
                    JObject jobject = (JObject)JsonConvert.DeserializeObject(responseData);
                    if (response != null)
                    {
                        response.Close();
                    }
                    if (request != null)
                    {
                        request.Abort();
                    }

                    if (jobject["code"].ToObject<string>().ToLower() == "200")
                    {
                        translate = jobject["data"]["dst"].ToObject<string>().ToLower();
                    }
                    //return jobject["code"].ToObject<string>().ToLower();

                    return jobject["code"].ToObject<string>().ToLower();
                }
            }
            catch
            {
                return "300";
            }

        }

        /// <summary>
        /// 纠错
        /// </summary>
        /// <param name="content"></param>
        /// <returns></returns>
        public async static Task<string> Correction(string content)
        {
            intProcess(true, "Correction");
            System.Collections.Specialized.NameValueCollection namevalue = new System.Collections.Specialized.NameValueCollection();
            App.mainWindowViewModel.Value = 1;
            //需要纠错的内容
            namevalue["project_id"] = "2";
            namevalue["version"] = "1.0.1";
            if (!string.IsNullOrEmpty(Settings.Default.UserDate.id))
            {
                namevalue["user_id"] = Settings.Default.UserDate.id;
            }
            else
            {
                namevalue["user_id"] = "1";
            }
            namevalue["content"] = content;
            App.mainWindowViewModel.Value = 3;
            return await PostString(Uri_Correction, namevalue, "content");
        }

        /// <summary>
        /// 重写
        /// </summary>
        /// <param name="content"></param>
        /// <returns></returns>
        public async static Task<string> Rewrite(string content)
        {
            intProcess(true, "Rewrite");
            System.Collections.Specialized.NameValueCollection namevalue = new System.Collections.Specialized.NameValueCollection();
            App.mainWindowViewModel.Value = 1;
            //需要纠错的内容
            namevalue["project_id"] = "2";
            namevalue["version"] = "1.0.1";
            if (!string.IsNullOrEmpty(Settings.Default.UserDate.id))
            {
                namevalue["user_id"] = Settings.Default.UserDate.id;
            }
            else
            {
                namevalue["user_id"] = "1";
            }
            namevalue["content"] = content;
            App.mainWindowViewModel.Value = 3;
            return await PostString(Uri_Rewrite, namevalue, "content");
        }

        /// <summary>
        /// 从链接中返回存在的链接(仅返回第一个找到的链接)
        /// </summary>
        /// <param name="content"></param>
        /// <returns></returns>
        public static List<string> GetLinkFromString(string content)
        {
            if (string.IsNullOrEmpty(content))
            {
                return null;
            }
            // 创建正则表达式模式
            string pattern = @"(https?://[^\s]+)";

            // 获取所有匹配项
            MatchCollection matches = Regex.Matches(content, pattern);

            List<string> list = new List<string>();
            // 遍历并打印所有匹配项
            foreach (Match match in matches)
            {
                list.Add(match.Value);
            }
            return list;
        }

        public static string Code = "";

        /// <summary>
        /// 从接口获取信息，获取message对应value值
        /// </summary>
        /// <param name="url"></param>
        /// <param name="namevalue"></param>
        /// <returns></returns>
        private async static Task<string> PostString(string url, System.Collections.Specialized.NameValueCollection namevalue, string key = "message")
        {
            App.mainWindowViewModel.Value = 5;
            string repsonseData = "";
            try
            {
                using (var client = new WebClient())
                {
                    byte[] bytes = await client.UploadValuesTaskAsync(url, namevalue);
                    //转换成字符串类型
                    var json = Encoding.Default.GetString(bytes);
                    //将Json格式字符串转换成键值对
                    //var values = JsonConvert.DeserializeObject<Dictionary<string, string>>(json);
                    JObject jobject = (JObject)JsonConvert.DeserializeObject(json);
                    App.mainWindowViewModel.Value = 8;
                    string unicode = "";
                    // 遍历字典对象输出键值对
                    Code = jobject["code"].ToObject<string>().ToLower();
                    if (Code == "200")
                    {
                        unicode = jobject["data"][key].ToObject<string>().ToLower();
                    }
                    //将Unicode格式转换成String
                    //repsonseData = Regex.Unescape(unicode);
                    repsonseData = unicode;
                }
                clossProcess();
            }
            catch (Exception ex)
            {
                Code = "300";
                clossProcess();
                return null;
            }
            return repsonseData;
        }

        /// <summary>
        /// 初始进度条
        /// </summary>
        /// <param name="isprocess">是否需要进度条</param>
        /// <param name="progressTitle">文案</param>
        public static void intProcess(bool isprocess = true, string progressTitle = "Translating...", System.Windows.Visibility ProcessCloseBtnVisible = System.Windows.Visibility.Collapsed)
        {
            if (isprocess)
            {
                App.mainWindowViewModel.ProgressTitle = progressTitle;
                App.mainWindowViewModel.MaxValue = 10;
                App.mainWindowViewModel.Value = 0;
                App.mainWindowViewModel.IsProcessVisible = System.Windows.Visibility.Visible;
                App.mainWindowViewModel.ProcessCloseBtnVisible = ProcessCloseBtnVisible;
            }
        }

        /// <summary>
        /// 关闭进度条
        /// </summary>
        public static void clossProcess()
        {
            App.mainWindowViewModel.Value = 9;
            App.mainWindowViewModel.IsProcessVisible = System.Windows.Visibility.Collapsed;
            App.mainWindowViewModel.ProcessCloseBtnVisible = System.Windows.Visibility.Visible;
        }
        #endregion

        #region 支持翻译的语言

        /// <summary>
        /// 当前语言列表
        /// </summary>
        public static List<string> FromlanguageFamily { set; get; } = new List<string>();
        public static List<string> SetFromlanguageOrigin()
        {
            FromlanguageFamily.Clear();
            FromlanguageFamily.Add("Automatic Identification");
            FromlanguageFamily.Add("English");
            FromlanguageFamily.Add("Simplified Chinese");
            FromlanguageFamily.Add("Traditional Chinese");
            FromlanguageFamily.Add("Japanese");
            FromlanguageFamily.Add("Korean");
            FromlanguageFamily.Add("French");
            FromlanguageFamily.Add("Spanish");
            FromlanguageFamily.Add("Italian");
            FromlanguageFamily.Add("German");
            FromlanguageFamily.Add("Portuguese");
            FromlanguageFamily.Add("Russian");
            FromlanguageFamily.Add("Vietnamese");
            FromlanguageFamily.Add("Thai");
            FromlanguageFamily.Add("Arabic");
            FromlanguageFamily.Add("Greek");
            FromlanguageFamily.Add("Bulgarian");
            FromlanguageFamily.Add("Finnish");
            FromlanguageFamily.Add("Slovene");
            FromlanguageFamily.Add("Dutch");
            FromlanguageFamily.Add("Czech");
            FromlanguageFamily.Add("Swedish");
            FromlanguageFamily.Add("Polish");
            FromlanguageFamily.Add("Danish");
            FromlanguageFamily.Add("Romanian");
            FromlanguageFamily.Add("Hungarian");

            return FromlanguageFamily;
        }

        /// <summary>
        /// 需要翻译成对应语言列表
        /// </summary>
        public static List<string> TolanguageFamily { set; get; } = new List<string>();
        public static List<string> SetTolanguageOrigin()
        {
            TolanguageFamily.Clear();
            TolanguageFamily.Add("English");
            TolanguageFamily.Add("Simplified Chinese");
            TolanguageFamily.Add("Traditional Chinese");
            TolanguageFamily.Add("Japanese");
            TolanguageFamily.Add("Korean");
            TolanguageFamily.Add("French");
            TolanguageFamily.Add("Spanish");
            TolanguageFamily.Add("Italian");
            TolanguageFamily.Add("German");
            TolanguageFamily.Add("Portuguese");
            TolanguageFamily.Add("Russian");
            TolanguageFamily.Add("Vietnamese");
            TolanguageFamily.Add("Thai");
            TolanguageFamily.Add("Arabic");
            TolanguageFamily.Add("Greek");
            TolanguageFamily.Add("Bulgarian");
            TolanguageFamily.Add("Finnish");
            TolanguageFamily.Add("Slovene");
            TolanguageFamily.Add("Dutch");
            TolanguageFamily.Add("Czech");
            TolanguageFamily.Add("Swedish");
            TolanguageFamily.Add("Polish");
            TolanguageFamily.Add("Danish");
            TolanguageFamily.Add("Romanian");
            TolanguageFamily.Add("Hungarian");
            return TolanguageFamily;
        }

        /// <summary>
        /// 接口传值对应字符
        /// </summary>
        /// <param name="index">控制form与to的区别，form为0开始，to为1开始，因为to没有auto</param>
        /// <returns></returns>
        public static string UpdateLanguagebType(int index)
        {
            switch (index)
            {
                case 0:
                    return "auto";
                case 1:
                    return "en";
                case 2:
                    return "zh";
                case 3:
                    return "cht";
                case 4:
                    return "jp";
                case 5:
                    return "kor";
                case 6:
                    return "fra";
                case 7:
                    return "spa";
                case 8:
                    return "it";
                case 9:
                    return "de";
                case 10:
                    return "pt";
                case 11:
                    return "ru";
                case 12:
                    return "vie";
                case 13:
                    return "th";
                case 14:
                    return "ara";
                case 15:
                    return "el";
                case 16:
                    return "bul";
                case 17:
                    return "fin";
                case 18:
                    return "slo";
                case 19:
                    return "nl";
                case 20:
                    return "cs";
                case 21:
                    return "swe";
                case 22:
                    return "pl";
                case 23:
                    return "dan";
                case 24:
                    return "rom";
                case 25:
                    return "hu";
                default:
                    return "";
            }
        }
        #endregion
    }
}