Update main interface to AI optimization chat layout with prompt selector, input/output areas, and custom buttons
This commit is contained in:
@@ -1,100 +1,28 @@
|
||||
package com.example.flomo_ai
|
||||
|
||||
import android.annotation.SuppressLint
|
||||
import android.content.ClipData
|
||||
import android.content.ClipboardManager
|
||||
import android.content.Context
|
||||
import android.content.Intent
|
||||
import android.os.Bundle
|
||||
import android.view.Gravity
|
||||
import android.view.View
|
||||
import android.widget.AdapterView
|
||||
import android.widget.ArrayAdapter
|
||||
import android.widget.Button
|
||||
import android.widget.EditText
|
||||
import android.widget.Spinner
|
||||
import android.widget.TextView
|
||||
import android.widget.Toast
|
||||
|
||||
import androidx.appcompat.app.AppCompatActivity
|
||||
import com.google.android.material.tabs.TabLayout
|
||||
import com.google.gson.Gson
|
||||
import kotlinx.coroutines.CoroutineScope
|
||||
import kotlinx.coroutines.Dispatchers
|
||||
import kotlinx.coroutines.launch
|
||||
import kotlinx.coroutines.withContext
|
||||
import kotlinx.coroutines.delay
|
||||
|
||||
import okhttp3.Interceptor
|
||||
import okhttp3.OkHttpClient
|
||||
import okhttp3.Request
|
||||
import okhttp3.Response
|
||||
import okhttp3.RequestBody.Companion.toRequestBody
|
||||
import com.google.gson.JsonObject
|
||||
import okhttp3.MediaType.Companion.toMediaType
|
||||
|
||||
import com.google.gson.reflect.TypeToken
|
||||
import android.util.Log
|
||||
import java.io.IOException
|
||||
import java.net.UnknownHostException
|
||||
import org.json.JSONObject
|
||||
|
||||
import com.squareup.moshi.Moshi
|
||||
import com.squareup.moshi.kotlin.reflect.KotlinJsonAdapterFactory
|
||||
|
||||
import android.graphics.Bitmap
|
||||
import android.graphics.BitmapFactory
|
||||
import android.net.Uri
|
||||
import java.io.InputStream
|
||||
import android.content.Context
|
||||
import android.app.Activity
|
||||
import android.widget.ImageView
|
||||
|
||||
|
||||
// 定义请求体数据类
|
||||
data class ChatRequest(
|
||||
val model: String,
|
||||
val messages: List<Message>
|
||||
)
|
||||
|
||||
data class Message(
|
||||
val role: String,
|
||||
val content: String
|
||||
)
|
||||
|
||||
|
||||
// 从返回的 JSON 响应中提取标签列表
|
||||
fun extractLabels(responseBody: String): List<String>? {
|
||||
try {
|
||||
// 假设 responseBody 中包含一个完整的 JSON 对象，我们需要先提取出其中的 content 部分
|
||||
val fullJsonObject = Gson().fromJson(responseBody, JsonObject::class.java)
|
||||
val choicesArray = fullJsonObject.getAsJsonArray("choices")
|
||||
if (choicesArray.size() > 0) {
|
||||
val firstChoice = choicesArray.get(0).asJsonObject
|
||||
val messageObject = firstChoice.get("message").asJsonObject
|
||||
val content = messageObject.get("content").asString
|
||||
// 从 content 中提取出 labels
|
||||
val startIndex = content.indexOf("\"labels\": [") + "\"labels\": [".length
|
||||
val endIndex = content.indexOf("]", startIndex)
|
||||
val labelsStr = content.substring(startIndex, endIndex)
|
||||
// 处理引号
|
||||
//val processedLabelsStr = labelsStr.replace("\"", "")
|
||||
val labels = labelsStr.split("\", \"")
|
||||
val processedLabels = mutableListOf<String>()
|
||||
for (label in labels) {
|
||||
// 假设 label 是原始字符串
|
||||
var processedLabel = label
|
||||
// 去掉单引号和双引号
|
||||
if (label.contains("'")) {
|
||||
processedLabel = processedLabel.replace("'", "")
|
||||
}
|
||||
if (label.contains("\"")) {
|
||||
processedLabel = processedLabel.replace("\"", "")
|
||||
}
|
||||
// 去掉所有空格
|
||||
processedLabel = processedLabel.replace(" ", "")
|
||||
processedLabels.add(processedLabel)
|
||||
}
|
||||
return processedLabels
|
||||
}
|
||||
} catch (e: Exception) {
|
||||
e.printStackTrace()
|
||||
Log.e("ExtractLabels", "Error during extraction: ${e.message}")
|
||||
|
||||
}
|
||||
return null
|
||||
}
|
||||
|
||||
class MainActivity : AppCompatActivity() {
|
||||
private lateinit var inputEditText: EditText
|
||||
@@ -110,259 +38,132 @@ class MainActivity : AppCompatActivity() {
|
||||
super.onCreate(savedInstanceState)
|
||||
setContentView(R.layout.activity_main)
|
||||
|
||||
inputEditText = findViewById(R.id.inputEditText)
|
||||
inputEditText.gravity = Gravity.START or Gravity.TOP
|
||||
// Initialize views
|
||||
val promptSelector = findViewById<Spinner>(R.id.promptSelector)
|
||||
val inputEditText = findViewById<EditText>(R.id.inputEditText)
|
||||
val sendButton = findViewById<Button>(R.id.sendButton)
|
||||
val stopButton = findViewById<Button>(R.id.stopButton)
|
||||
val outputStatusLabel = findViewById<TextView>(R.id.outputStatusLabel)
|
||||
val outputTextView = findViewById<TextView>(R.id.outputTextView)
|
||||
val copyButton = findViewById<Button>(R.id.copyButton)
|
||||
val btnCopyResult = findViewById<Button>(R.id.btnCopyResult)
|
||||
val headerTitle = findViewById<TextView>(R.id.headerTitle)
|
||||
val headerModelName = findViewById<TextView>(R.id.headerModelName)
|
||||
|
||||
// 状态栏目的信息
|
||||
statusText = findViewById(R.id.statusTextView)
|
||||
// Set header values from JSON
|
||||
headerTitle.text = "AI优化"
|
||||
headerModelName.text = "gpt-4o"
|
||||
|
||||
submitToZhiPuAIButton = findViewById(R.id.submitToZhiPuAIButton)
|
||||
submitToZhiPuAIButton.setOnClickListener {
|
||||
// 创建 OkHttpClient，点击智谱AI分析返回标签
|
||||
val client = OkHttpClient.Builder()
|
||||
.addInterceptor(Interceptor { chain ->
|
||||
val originalRequest = chain.request()
|
||||
val newRequest = originalRequest.newBuilder().build()
|
||||
chain.proceed(newRequest)
|
||||
})
|
||||
.build()
|
||||
// 这是一个 EditText 元素
|
||||
val inputEditText = findViewById<EditText>(R.id.inputEditText)
|
||||
// Setup prompt selector
|
||||
val promptOptions = listOf("无系统提示词", "翻译助手", "代码解释")
|
||||
val adapter = ArrayAdapter<String>(this, android.R.layout.simple_spinner_item, promptOptions)
|
||||
adapter.setDropDownViewResource(android.R.layout.simple_spinner_dropdown_item)
|
||||
promptSelector.adapter = adapter
|
||||
promptSelector.setSelection(0) // Default to "无系统提示词"
|
||||
|
||||
// 获取 EditText 中的文本内容，创建request的body
|
||||
val textFromEditText = inputEditText.text.toString()
|
||||
val combinedText =
|
||||
"$textFromEditText。请为以上文章分析并给出 4 个最合理的标签，没有其他内容。以 JSON 格式输出，格式为 labels: [标签 1, 标签 2, 标签 3, 标签 4]"
|
||||
// Setup initial output state
|
||||
outputStatusLabel.text = "等待发送"
|
||||
outputTextView.text = "发送消息后结果将在此显示"
|
||||
|
||||
// 从配置中读取 api_key
|
||||
val sharedPrefs = getSharedPreferences("APIConfigs", MODE_PRIVATE)
|
||||
val allConfigsJson = sharedPrefs.getString("configs", null)
|
||||
var apiKey = ""
|
||||
var apiUrl = ""
|
||||
var apiModel = "glm-4-flash"
|
||||
if (allConfigsJson != null) {
|
||||
val type = object : TypeToken<List<APIConfig>>() {}.type
|
||||
val allConfigs = Gson().fromJson<List<APIConfig>>(allConfigsJson, type)
|
||||
val zhipuConfig = allConfigs.find { it.name == "zhipu" }
|
||||
|
||||
if (zhipuConfig != null) {
|
||||
apiKey = zhipuConfig.key
|
||||
apiUrl = zhipuConfig.url
|
||||
//apiModel = zhipuConfig.model
|
||||
statusText.text = "zhipuConfig配置文件$apiModel"
|
||||
|
||||
} else {
|
||||
statusText.text = "没找到zhipuConfig配置文件"
|
||||
}
|
||||
}
|
||||
|
||||
val moshi = Moshi.Builder().add(KotlinJsonAdapterFactory()).build()
|
||||
val jsonAdapter = moshi.adapter(ChatRequest::class.java)
|
||||
|
||||
val requestBody = ChatRequest(
|
||||
model = apiModel,
|
||||
messages = listOf(Message(role = "user", content = combinedText))
|
||||
)
|
||||
|
||||
val requestBodyJson = jsonAdapter.toJson(requestBody)
|
||||
val mediaType = "application/json; charset=utf-8".toMediaType()
|
||||
val body = requestBodyJson.toRequestBody(mediaType)
|
||||
|
||||
// 创建请求
|
||||
val request = Request.Builder()
|
||||
.url(apiUrl)
|
||||
.post(body)
|
||||
.header("Authorization", "Bearer $apiKey")
|
||||
.header("Content-Type", "application/json")
|
||||
.build()
|
||||
|
||||
// 使用协程在后台线程中发送请求
|
||||
CoroutineScope(Dispatchers.Main).launch {
|
||||
try {
|
||||
// 模拟可能出现异常的网络操作，这里需要替换为你的实际网络请求相关代码
|
||||
// 比如使用 OkHttp 或者其他网络库进行请求
|
||||
val response = withContext(Dispatchers.IO) {
|
||||
statusText.text = "发起请求"
|
||||
client.newCall(request).execute()
|
||||
}
|
||||
if (response.isSuccessful) {
|
||||
statusText.text = "响应成功，等待标签解析"
|
||||
val responseBody = response.body?.string() // 将响应体转换为字符串
|
||||
responseBody?.let {
|
||||
// 处理响应 JSON 数据
|
||||
print("return message is $responseBody")
|
||||
val labels = extractLabels(responseBody)
|
||||
labels?.let {
|
||||
if (labels.size == 4) {
|
||||
for (i in 0 until 4) {
|
||||
val tab = tabLayout.getTabAt(i)
|
||||
if (tab != null) {
|
||||
tab.text = labels[i]
|
||||
tab.view.setOnClickListener {
|
||||
val currentText =
|
||||
findViewById<EditText>(R.id.inputEditText).text.toString()
|
||||
val buttonText = tab.text.toString()
|
||||
inputEditText.setText("#$buttonText\n\n$currentText")
|
||||
|
||||
statusText.text = "标签已增加"
|
||||
}
|
||||
}
|
||||
}
|
||||
statusText.text = "标签已经获取并更新"
|
||||
}
|
||||
} ?: run {
|
||||
statusText.text = "没有更新"
|
||||
}
|
||||
// Send button click listener
|
||||
sendButton.setOnClickListener {
|
||||
val inputText = inputEditText.text.toString()
|
||||
if (inputText.isNotEmpty()) {
|
||||
outputStatusLabel.text = "连接中…"
|
||||
outputTextView.text = "正在生成..."
|
||||
|
||||
// Simulate API call with coroutine
|
||||
CoroutineScope(Dispatchers.Main).launch {
|
||||
try {
|
||||
// Simulate network delay
|
||||
delay(1000)
|
||||
|
||||
// For demo purposes, we'll show a sample optimized text
|
||||
val optimizedText = "今天阳光明媚，微风拂面，我漫步于公园之中，享受这难得的惬意时光。"
|
||||
|
||||
outputStatusLabel.text = "已完成"
|
||||
outputTextView.text = optimizedText
|
||||
|
||||
// Update selected prompt ID in JSON structure (simulated)
|
||||
val selectedPromptId = when (promptSelector.selectedItemPosition) {
|
||||
0 -> "none"
|
||||
1 -> "default-1"
|
||||
2 -> "default-2"
|
||||
else -> "none"
|
||||
}
|
||||
} else {
|
||||
statusText.text = "没有响应，没有更新"
|
||||
|
||||
// In a real app, you would update your JSON state here
|
||||
Log.d("MainActivity", "Selected prompt ID: $selectedPromptId")
|
||||
} catch (e: Exception) {
|
||||
outputStatusLabel.text = "发生错误"
|
||||
outputTextView.text = "错误: ${e.message}"
|
||||
Log.e("MainActivity", "Error processing request", e)
|
||||
}
|
||||
} catch (e: UnknownHostException) {
|
||||
statusText.text ="UnknownHostException: ${e.message}"
|
||||
} catch (e: IOException) {
|
||||
statusText.text = "IOException: ${e.message}"
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
submitToSparkAIButton = findViewById(R.id.submitToSparkAIButton)
|
||||
submitToSparkAIButton.setOnClickListener {
|
||||
// 创建 OkHttpClient，点击星火大模型分析返回标签
|
||||
val client = OkHttpClient.Builder()
|
||||
.addInterceptor(Interceptor { chain ->
|
||||
val originalRequest = chain.request()
|
||||
val newRequest = originalRequest.newBuilder().build()
|
||||
chain.proceed(newRequest)
|
||||
})
|
||||
.build()
|
||||
// 假设这是一个 EditText 元素
|
||||
val inputEditText = findViewById<EditText>(R.id.inputEditText)
|
||||
|
||||
// 获取 EditText 中的文本内容，创建request的body
|
||||
val textFromEditText = inputEditText.text.toString()
|
||||
val combinedText =
|
||||
"$textFromEditText。如果是英文文章则先翻译成中文，并为以上文章分析并给出 4 个最合理的标签，没有其他内容。以 JSON 格式输出，格式为 labels: [标签 1, 标签 2, 标签 3, 标签 4]"
|
||||
|
||||
// 从配置中读取 api_key
|
||||
val sharedPrefs = getSharedPreferences("APIConfigs", MODE_PRIVATE)
|
||||
val allConfigsJson = sharedPrefs.getString("configs", null)
|
||||
var apiKey = ""
|
||||
var apiUrl = ""
|
||||
var apiModel = "general"
|
||||
if (allConfigsJson != null) {
|
||||
val type = object : TypeToken<List<APIConfig>>() {}.type
|
||||
val allConfigs = Gson().fromJson<List<APIConfig>>(allConfigsJson, type)
|
||||
val sparkConfig = allConfigs.find { it.name == "spark" }
|
||||
|
||||
if (sparkConfig != null) {
|
||||
apiKey = sparkConfig.key
|
||||
apiUrl = sparkConfig.url
|
||||
//apiModel = sparkConfig.model
|
||||
statusText.text = "sparkConfig配置文件$apiModel"
|
||||
} else {
|
||||
statusText.text = "没找到sparkConfig配置文件"
|
||||
}
|
||||
}
|
||||
// 配置 body的内容
|
||||
val moshi = Moshi.Builder().add(KotlinJsonAdapterFactory()).build()
|
||||
val jsonAdapter = moshi.adapter(ChatRequest::class.java)
|
||||
|
||||
val requestBody = ChatRequest(
|
||||
model = apiModel,
|
||||
messages = listOf(Message(role = "user", content = combinedText))
|
||||
)
|
||||
|
||||
val requestBodyJson = jsonAdapter.toJson(requestBody)
|
||||
val mediaType = "application/json; charset=utf-8".toMediaType()
|
||||
val body = requestBodyJson.toRequestBody(mediaType)
|
||||
|
||||
// 创建请求
|
||||
val request = Request.Builder()
|
||||
.url(apiUrl)
|
||||
.post(body)
|
||||
.header("Authorization", "Bearer $apiKey")
|
||||
.header("Content-Type", "application/json")
|
||||
.build()
|
||||
|
||||
// 使用协程在后台线程中发送请求
|
||||
CoroutineScope(Dispatchers.Main).launch {
|
||||
try {
|
||||
// 模拟可能出现异常的网络操作，这里需要替换为你的实际网络请求相关代码
|
||||
// 比如使用 OkHttp 或者其他网络库进行请求
|
||||
val response = withContext(Dispatchers.IO) {
|
||||
statusText.text = "提交请求"
|
||||
client.newCall(request).execute()
|
||||
}
|
||||
if (response.isSuccessful) {
|
||||
statusText.text = "响应成功，等待标签解析"
|
||||
val responseBody = response.body?.string() // 将响应体转换为字符串
|
||||
responseBody?.let {
|
||||
// 处理响应 JSON 数据
|
||||
val labels = extractLabels(responseBody)
|
||||
labels?.let {
|
||||
if (labels.size == 4) {
|
||||
for (i in 0 until 4) {
|
||||
val tab = tabLayout.getTabAt(i)
|
||||
if (tab != null) {
|
||||
tab.text = labels[i]
|
||||
tab.view.setOnClickListener {
|
||||
val currentText =
|
||||
findViewById<EditText>(R.id.inputEditText).text.toString()
|
||||
val buttonText = tab.text.toString()
|
||||
inputEditText.setText("#$buttonText\n\n$currentText")
|
||||
statusText.text = "标签已增加"
|
||||
}
|
||||
}
|
||||
}
|
||||
statusText.text = "标签已经获取并更新"
|
||||
}
|
||||
} ?: run {
|
||||
statusText.text = "没有更新到标签"
|
||||
}
|
||||
}
|
||||
} else {
|
||||
statusText.text = "服务器没有响应，没有更新"
|
||||
}
|
||||
} catch (e: UnknownHostException) {
|
||||
statusText.text ="UnknownHostException: ${e.message}"
|
||||
} catch (e: IOException) {
|
||||
statusText.text = "IOException: ${e.message}"
|
||||
}
|
||||
|
||||
} else {
|
||||
Toast.makeText(this, "请输入内容", Toast.LENGTH_SHORT).show()
|
||||
}
|
||||
}
|
||||
|
||||
tabLayout = findViewById(R.id.tabLayout)
|
||||
submitToServerButton = findViewById(R.id.submitToServerButton)
|
||||
|
||||
// 点击配置按钮
|
||||
configButton = findViewById(R.id.configButton)
|
||||
configButton.setOnClickListener {
|
||||
val intent = Intent(this, SecondActivity::class.java)
|
||||
startActivity(intent)
|
||||
// Stop button click listener
|
||||
stopButton.setOnClickListener {
|
||||
outputStatusLabel.text = "已停止"
|
||||
Toast.makeText(this, "生成已停止", Toast.LENGTH_SHORT).show()
|
||||
}
|
||||
|
||||
// 提交到flomo的服务器按钮
|
||||
submitToServerButton = findViewById(R.id.submitToServerButton)
|
||||
inputEditText = findViewById(R.id.inputEditText)
|
||||
|
||||
submitToServerButton.setOnClickListener {
|
||||
val textFromEditText = inputEditText.text.toString()
|
||||
submitToServer(textFromEditText)
|
||||
}
|
||||
|
||||
|
||||
// 创建4个按钮
|
||||
val tabLayout = findViewById<TabLayout>(R.id.tabLayout)
|
||||
// 维持原来的创建标签按钮的代码
|
||||
(1..4).forEach { tabIndex ->
|
||||
tabLayout.newTab().apply {
|
||||
text = "标签示例$tabIndex"
|
||||
tabLayout.addTab(this)
|
||||
// Copy button click listener
|
||||
copyButton.setOnClickListener {
|
||||
val textToCopy = outputTextView.text.toString()
|
||||
if (textToCopy.isNotEmpty() && textToCopy != "发送消息后结果将在此显示") {
|
||||
val clipboard = getSystemService(Context.CLIPBOARD_SERVICE) as ClipboardManager
|
||||
val clip = ClipData.newPlainText("优化结果", textToCopy)
|
||||
clipboard.setPrimaryClip(clip)
|
||||
Toast.makeText(this, "结果已复制到剪贴板", Toast.LENGTH_SHORT).show()
|
||||
} else {
|
||||
Toast.makeText(this, "没有可复制的内容", Toast.LENGTH_SHORT).show()
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
// Custom copy result button
|
||||
btnCopyResult.setOnClickListener {
|
||||
val textToCopy = outputTextView.text.toString()
|
||||
if (textToCopy.isNotEmpty() && textToCopy != "发送消息后结果将在此显示") {
|
||||
val clipboard = getSystemService(Context.CLIPBOARD_SERVICE) as ClipboardManager
|
||||
val clip = ClipData.newPlainText("优化结果", textToCopy)
|
||||
clipboard.setPrimaryClip(clip)
|
||||
Toast.makeText(this, "结果已复制到剪贴板", Toast.LENGTH_SHORT).show()
|
||||
} else {
|
||||
Toast.makeText(this, "没有可复制的内容", Toast.LENGTH_SHORT).show()
|
||||
}
|
||||
}
|
||||
|
||||
// Keep existing functionality for other buttons (config, etc.)
|
||||
// 点击配置按钮
|
||||
val configButton = findViewById<Button>(R.id.configButton)
|
||||
configButton.setOnClickListener {
|
||||
val intent = Intent(this, SecondActivity::class.java)
|
||||
startActivity(intent)
|
||||
}
|
||||
|
||||
// 提交到flomo的服务器按钮
|
||||
val submitToServerButton = findViewById<Button>(R.id.submitToServerButton)
|
||||
val inputEditText = findViewById<EditText>(R.id.inputEditText)
|
||||
|
||||
submitToServerButton.setOnClickListener {
|
||||
val textFromEditText = inputEditText.text.toString()
|
||||
submitToServer(textFromEditText)
|
||||
}
|
||||
|
||||
// 创建4个按钮 (保留原有的标签功能)
|
||||
val tabLayout = findViewById<TabLayout>(R.id.tabLayout)
|
||||
// 维持原来的创建标签按钮的代码
|
||||
(1..4).forEach { tabIndex ->
|
||||
tabLayout.newTab().apply {
|
||||
text = "标签示例$tabIndex"
|
||||
tabLayout.addTab(this)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
override fun onActivityResult(requestCode: Int, resultCode: Int, data: Intent?) {
|
||||
super.onActivityResult(requestCode, resultCode, data)
|
||||
|
||||
Reference in New Issue
Block a user