文件提交

Dockerfile.txt

@@ -0,0 +1,22 @@
+FROM openjdk:8-alpine
+MAINTAINER wfq
+WORKDIR /code
+COPY . .
+RUN wget https://dlcdn.apache.org/maven/maven-3/3.9.2/binaries/apache-maven-3.9.2-bin.tar.gz
+RUN tar -zxvf apache-maven-3.9.2-bin.tar.gz
+RUN pwd
+RUN ls -la
+ENV PATH /code/apache-maven-3.9.2/bin:$PATH
+ENV MAVEN_HOME=/code/apache-maven-3.9.2
+RUN mvn -v
+RUN mvn install  -Dmaven.test.skip=true
+## 更新apk源
+RUN apk update
+## 设置时区
+RUN apk add -U tzdata
+RUN cp /usr/share/zoneinfo/Asia/Shanghai /etc/localtime
+## 设置输出格式
+ENV LANG en_US.UTF-8
+ENV LANGUAGE en_US:en
+ENV LC_ALL en_US.UTF-8
+ENTRYPOINT [ "sh", "-c", "java $JAVA_OPTS -Duser.region=CN -Duser.timezone=GMT=GMT+8 -Djava.security.egd=file:/dev/./urandom -jar target/demo.jar"]

pom.xml

@@ -0,0 +1,68 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+	<modelVersion>4.0.0</modelVersion>
+
+	<groupId>com.bitservice</groupId>
+	<artifactId>Aidemo</artifactId>
+	<version>0.0.1-SNAPSHOT</version>
+	<packaging>jar</packaging>
+
+	<name>demo</name>
+	<description>Demo project for Spring Boot</description>
+	<parent>
+		<groupId>org.springframework.boot</groupId>
+		<artifactId>spring-boot-starter-parent</artifactId>
+		<version>2.3.12.RELEASE</version>
+		<relativePath /> <!-- lookup parent from repository -->
+	</parent>
+
+	<properties>
+		<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
+		<project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
+		<java.version>1.8</java.version>
+	</properties>
+
+	<dependencies>
+		<dependency>
+			<groupId>org.springframework.boot</groupId>
+			<artifactId>spring-boot-starter-web</artifactId>
+		</dependency>
+
+		<dependency>
+			<groupId>org.springframework.boot</groupId>
+			<artifactId>spring-boot-starter-test</artifactId>
+			<scope>test</scope>
+		</dependency>
+		<dependency>
+			<groupId>com.azure</groupId>
+			<artifactId>azure-ai-openai</artifactId>
+			<version>1.0.0-beta.2</version>
+		</dependency>
+		<dependency>
+			<groupId>com.alibaba</groupId>
+			<artifactId>fastjson</artifactId>
+			<version>1.2.47</version>
+		</dependency>
+		<dependency>
+			<groupId>org.apache.httpcomponents</groupId>
+			<artifactId>httpclient</artifactId>
+		</dependency>
+		<dependency>
+			<groupId>org.projectlombok</groupId>
+			<artifactId>lombok</artifactId>
+			<scope>provided</scope>
+		</dependency>
+	</dependencies>
+
+	<build>
+		<finalName>/demo</finalName>
+		<plugins>
+			<plugin>
+				<groupId>org.springframework.boot</groupId>
+				<artifactId>spring-boot-maven-plugin</artifactId>
+			</plugin>
+		</plugins>
+	</build>
+
+
+</project>

src/main/java/com/bit/AIController.java

@@ -0,0 +1,83 @@
+package com.bit;
+
+import java.io.IOException;
+
+import lombok.extern.slf4j.Slf4j;
+import org.apache.http.Header;
+import org.apache.http.HttpEntity;
+import org.apache.http.HttpResponse;
+import org.apache.http.client.ClientProtocolException;
+import org.apache.http.client.ResponseHandler;
+import org.apache.http.client.methods.CloseableHttpResponse;
+import org.apache.http.client.methods.HttpGet;
+import org.apache.http.client.methods.HttpPost;
+import org.apache.http.entity.StringEntity;
+import org.apache.http.impl.client.CloseableHttpClient;
+import org.apache.http.impl.client.HttpClientBuilder;
+import org.apache.http.util.EntityUtils;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+import com.alibaba.fastjson.JSONArray;
+import com.alibaba.fastjson.JSONObject;
+
+@RestController
+@Slf4j
+public class AIController {
+	@PostMapping("/txt2pic")
+	String txt2pic(String prompt, String size) {
+		String result = "";
+		String api_key = System.getenv("OPENAI_API_KEY");
+		String api_base = System.getenv("OPENAI_API_BASE");
+		String api_version = "2023-06-01-preview";
+		String url = api_base + "openai/images/generations:submit?api-version=" + api_version;
+		CloseableHttpClient httpclient = HttpClientBuilder.create().build();
+		HttpPost httpPost = new HttpPost(url);
+		ResponseHandler<String> handler = new ResponseHandler<String>() {
+			@Override
+			public String handleResponse(HttpResponse response) throws ClientProtocolException, IOException {
+				// TODO Auto-generated method stub
+				Header[] headers = response.getHeaders("operation-location");
+				return headers[0].getValue();
+			}
+		};
+		// 设置请求格式
+		String jsonData = "{\"prompt\":\"" + prompt + "\",\"size\":\"" + size + "\",\"n\":1}";
+		log.info("查询请求:"+jsonData);
+		StringEntity entity = new StringEntity(jsonData, "utf-8");// 解决中文乱码问题
+		entity.setContentEncoding("UTF-8");
+		entity.setContentType("application/json");
+		httpPost.setEntity(entity);
+		httpPost.addHeader("api-key", api_key);
+		httpPost.addHeader("Content-Type", "application/json");
+		CloseableHttpResponse response = null;
+		try {
+			result = httpclient.execute(httpPost, handler);
+			HttpGet httpget = new HttpGet(result);
+			httpget.addHeader("api-key", api_key);
+			httpget.addHeader("Content-Type", "application/json");
+			String status = "";
+			while (!status.equals( "succeeded")) {
+				Thread.sleep(1000);
+				response = httpclient.execute(httpget);
+				HttpEntity he = response.getEntity();
+				String ret=EntityUtils.toString(response.getEntity());
+				JSONObject obj = JSONObject.parseObject(ret);
+				status=obj.getString("status");
+				log.info("当前状态:"+status);
+				if (status .equals("succeeded")) {
+					JSONArray js=obj.getJSONObject("result").getJSONArray("data");
+					String images_url=js.getJSONObject(0).getString("url");
+					log.info("获取到图片url:"+images_url);
+					return images_url;
+				}
+			}
+		} catch (Exception e) {
+			// TODO Auto-generated catch block
+			e.printStackTrace();
+		}
+		return null;
+	}
+
+
+}

src/main/java/com/bit/DemoApplication.java

@@ -0,0 +1,20 @@
+package com.bit;
+
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.stereotype.Controller;
+import org.springframework.web.bind.annotation.GetMapping;
+
+@SpringBootApplication
+@Controller
+public class DemoApplication {
+
+	@GetMapping("/")
+	String home() {
+		return "index.html";
+	}
+
+	public static void main(String[] args) {
+		SpringApplication.run(DemoApplication.class, args);
+	}
+}