238 lines
12 KiB
Java
238 lines
12 KiB
Java
package com.bifan.txtreaderlib.ui;
|
||
|
||
import android.content.Context;
|
||
import android.content.SharedPreferences;
|
||
import android.os.Bundle;
|
||
import android.util.Log;
|
||
import android.view.View;
|
||
import android.widget.Button;
|
||
import android.widget.EditText;
|
||
import android.widget.TextView;
|
||
|
||
import androidx.annotation.Nullable;
|
||
import androidx.appcompat.app.AppCompatActivity;
|
||
import androidx.recyclerview.widget.LinearLayoutManager;
|
||
import androidx.recyclerview.widget.RecyclerView;
|
||
|
||
import com.bifan.txtreaderlib.R;
|
||
import com.bifan.txtreaderlib.adapter.MessageAdapter;
|
||
import com.bifan.txtreaderlib.bean.ChatMessage_bmob;
|
||
import com.bifan.txtreaderlib.bean.ChatRequest;
|
||
import com.bifan.txtreaderlib.bean.ChatResponse;
|
||
import com.bifan.txtreaderlib.bean.Message;
|
||
import com.bifan.txtreaderlib.interfaces.ChatGPTService;
|
||
|
||
import org.json.JSONObject;
|
||
|
||
import java.io.IOException;
|
||
import java.util.ArrayList;
|
||
import java.util.List;
|
||
import java.util.concurrent.TimeUnit;
|
||
|
||
import cn.bmob.v3.Bmob;
|
||
import cn.bmob.v3.BmobQuery;
|
||
import cn.bmob.v3.exception.BmobException;
|
||
import cn.bmob.v3.listener.FindListener;
|
||
import cn.bmob.v3.listener.SaveListener;
|
||
import okhttp3.OkHttpClient;
|
||
import retrofit2.Call;
|
||
import retrofit2.Callback;
|
||
import retrofit2.Response;
|
||
import retrofit2.Retrofit;
|
||
import retrofit2.converter.gson.GsonConverterFactory;
|
||
|
||
public class ChatWithGptActivity extends AppCompatActivity {
|
||
private RecyclerView recyclerView;
|
||
private MessageAdapter messageAdapter;
|
||
private EditText editTextMessage;
|
||
private TextView chatbot_title;
|
||
private Button buttonSend;
|
||
private String bookName = "";
|
||
private String userid="";
|
||
private ArrayList<Message> messageList = new ArrayList<>();
|
||
|
||
@Override
|
||
protected void onCreate(@Nullable Bundle savedInstanceState) {
|
||
super.onCreate(savedInstanceState);
|
||
setContentView(R.layout.activity_chatwithgpt);
|
||
Bmob.initialize(this, "8f9f1d1ea520b0ce4f84a6fa83a5f754");
|
||
if (getIntent() != null && getIntent().hasExtra("book_name")) {
|
||
bookName = getIntent().getStringExtra("book_name");
|
||
}
|
||
SharedPreferences sharedPreferences = getSharedPreferences("userinf", Context.MODE_PRIVATE);
|
||
String userIdFromPrefs = sharedPreferences.getString("user_id", "");
|
||
userid = userIdFromPrefs != null ? userIdFromPrefs : "";
|
||
|
||
recyclerView = findViewById(R.id.chatRecyclerView);
|
||
editTextMessage = findViewById(R.id.inputEditText);
|
||
buttonSend = findViewById(R.id.sendButton);
|
||
chatbot_title=findViewById(R.id.chatbot_title);
|
||
messageAdapter = new MessageAdapter(messageList);
|
||
recyclerView.setLayoutManager(new LinearLayoutManager(this));
|
||
recyclerView.setAdapter(messageAdapter);
|
||
init();
|
||
buttonSend.setOnClickListener(new View.OnClickListener() {
|
||
@Override
|
||
public void onClick(View v) {
|
||
String text = editTextMessage.getText().toString();
|
||
if (!text.isEmpty()) {
|
||
messageList.add(new Message(text, 1)); // User message
|
||
messageAdapter.notifyItemInserted(messageList.size() - 1);
|
||
recyclerView.scrollToPosition(messageList.size() - 1);
|
||
editTextMessage.setText("");
|
||
sendMessageToChatGPT(text);
|
||
ChatMessage_bmob chatMessage_bmob=new ChatMessage_bmob(text,1,bookName,userid);
|
||
chatMessage_bmob.save(new SaveListener<String>() {
|
||
@Override
|
||
public void done(String objectId,BmobException e) {
|
||
if(e==null){
|
||
// toast("添加数据成功，返回objectId为："+objectId);
|
||
}else{
|
||
//toast("创建数据失败：" + e.getMessage());
|
||
}
|
||
}
|
||
});
|
||
}
|
||
}
|
||
});
|
||
}
|
||
private void init(){
|
||
chatbot_title.setText(bookName);
|
||
BmobQuery<ChatMessage_bmob> bmobQuery = new BmobQuery<>();
|
||
bmobQuery.addWhereEqualTo("relatedBook", bookName);
|
||
bmobQuery.addWhereEqualTo("userid", userid);
|
||
bmobQuery.findObjects(new FindListener<ChatMessage_bmob>() {
|
||
@Override
|
||
public void done(List<ChatMessage_bmob> messages, BmobException e) {
|
||
if (e == null && messages != null) {
|
||
if (!messages.isEmpty()) {
|
||
for (ChatMessage_bmob item : messages) {
|
||
if (item.getContent() == null) {
|
||
String content = "出现未知错误!";
|
||
messageList.add(new Message(content, item.getIsuser()));
|
||
} else {
|
||
messageList.add(new Message(item.getContent(), item.getIsuser()));
|
||
}
|
||
}
|
||
} else {
|
||
// 如果没有消息，添加一条 bot 的欢迎消息
|
||
String welcomeMessage = "你好！我是你的阅读助手。我可以帮助你解答关于书籍的各种问题，比如内容概述、作者信息、主要主题和角色分析等。请随时向我提问，现在，你有什么关于《" + bookName + "》的问题吗？";
|
||
messageList.add(new Message(welcomeMessage, 0)); // 假设 0 表示系统或助手消息
|
||
ChatMessage_bmob chatMessage_bmob=new ChatMessage_bmob(welcomeMessage,0,bookName,userid);
|
||
chatMessage_bmob.save(new SaveListener<String>() {
|
||
@Override
|
||
public void done(String objectId,BmobException e) {
|
||
if(e==null){
|
||
// toast("添加数据成功，返回objectId为："+objectId);
|
||
}else{
|
||
//toast("创建数据失败：" + e.getMessage());
|
||
}
|
||
}
|
||
});
|
||
|
||
}
|
||
messageAdapter.notifyDataSetChanged();
|
||
} else {
|
||
Log.e("BookInfo", "Error loading book info: " + (e != null ? e.getMessage() : ""));
|
||
}
|
||
}
|
||
});
|
||
}
|
||
|
||
/**
|
||
* 从数据库加载历史对话并构建消息列表
|
||
*/
|
||
private List<ChatRequest.Message> loadMessagesFromDatabase(String bookName, String userId) {
|
||
List<ChatRequest.Message> messages = new ArrayList<>();
|
||
BmobQuery<ChatMessage_bmob> bmobQuery = new BmobQuery<>();
|
||
bmobQuery.addWhereEqualTo("relatedBook", bookName);
|
||
bmobQuery.addWhereEqualTo("userid", userId);
|
||
bmobQuery.order("createdAt");
|
||
bmobQuery.setLimit(10); //设置上下文最多10条历史对话
|
||
bmobQuery.findObjects(new FindListener<ChatMessage_bmob>() {
|
||
@Override
|
||
public void done(List<ChatMessage_bmob> chatMessages, BmobException e) {
|
||
if (e == null && chatMessages != null && !chatMessages.isEmpty()) {
|
||
for (ChatMessage_bmob item : chatMessages) {
|
||
// 根据消息来源判断角色，假设 0 表示用户，1 表示助手
|
||
String role = item.getIsuser() == 0 ? "user" : "assistant";
|
||
messages.add(new ChatRequest.Message(role, item.getContent()));
|
||
Log.d("test",item.getContent());
|
||
}
|
||
} else {
|
||
Log.e("BookInfo", "Error loading book info: " + (e != null ? e.getMessage() : "No error details."));
|
||
}
|
||
}
|
||
});
|
||
return messages;
|
||
}
|
||
|
||
private void sendMessageToChatGPT(String messageText) {
|
||
OkHttpClient client = new OkHttpClient.Builder()
|
||
.connectTimeout(60, TimeUnit.SECONDS)
|
||
.readTimeout(60, TimeUnit.SECONDS)
|
||
.writeTimeout(60, TimeUnit.SECONDS)
|
||
.build();
|
||
Retrofit retrofit = new Retrofit.Builder()
|
||
.client(client)
|
||
.baseUrl("https://api.openai-ch.top/")
|
||
.addConverterFactory(GsonConverterFactory.create())
|
||
.build();
|
||
ChatGPTService service = retrofit.create(ChatGPTService.class);
|
||
List<ChatRequest.Message> messages = new ArrayList<>();
|
||
String prompt = "你好！我是你的阅读助手。我可以帮助你解答关于书籍的各种问题，比如内容概述、作者信息、主要主题和角色分析等。请随时向我提问，我会用中文简体来回答你。现在，你有什么关于《" + bookName + "》的问题吗？";
|
||
messages.add(new ChatRequest.Message("system", prompt));
|
||
|
||
// messages.add(new ChatRequest.Message("system", "You are a reading assistant. The user will ask you questions about the book '" + bookName + "'."));
|
||
messages.addAll(loadMessagesFromDatabase(bookName, userid));
|
||
messages.add(new ChatRequest.Message("user", messageText)); // 添加用户消息
|
||
|
||
ChatRequest request = new ChatRequest(messages); // 使用消息列表创建请求
|
||
request.setModel("gpt-4-turbo"); // 根据需要选择不同的模型
|
||
request.setMaxTokens(200); // 设置最大 token 数，根据需要调整
|
||
request.setTemperature(0.5); // 设置适当的温度
|
||
service.postMessage(request).enqueue(new Callback<ChatResponse>() {
|
||
@Override
|
||
public void onResponse(Call<ChatResponse> call, Response<ChatResponse> response) {
|
||
if (response.isSuccessful() && response.body() != null) {
|
||
ChatResponse chatResponse = response.body();
|
||
if (chatResponse.choices != null && !chatResponse.choices.isEmpty()) {
|
||
ChatResponse.Message message = chatResponse.choices.get(0).message;
|
||
String reply = message.content; // 这是从 GPT-3.5 响应中提取的内容
|
||
runOnUiThread(() -> {
|
||
messageList.add(new Message(reply, 0));
|
||
messageAdapter.notifyItemInserted(messageList.size() - 1);
|
||
recyclerView.scrollToPosition(messageList.size() - 1);
|
||
ChatMessage_bmob chatMessage_bmob = new ChatMessage_bmob(reply, 0, bookName, userid);
|
||
chatMessage_bmob.save(new SaveListener<String>() {
|
||
@Override
|
||
public void done(String objectId, BmobException e) {
|
||
if (e == null) {
|
||
// toast("添加数据成功，返回objectId为：" + objectId);
|
||
} else {
|
||
// toast("创建数据失败：" + e.getMessage());
|
||
}
|
||
}
|
||
});
|
||
});
|
||
}
|
||
} else {
|
||
try {
|
||
String errorBody = response.errorBody() != null ? response.errorBody().string() : "Unknown error";
|
||
Log.e("ChatGPT", "Error: " + errorBody);
|
||
} catch (IOException e) {
|
||
Log.e("ChatGPT", "Error reading error body", e);
|
||
}
|
||
}
|
||
}
|
||
|
||
@Override
|
||
public void onFailure(Call<ChatResponse> call, Throwable t) {
|
||
Log.e("ChatGPT", "Failure: " + t.getMessage());
|
||
}
|
||
});
|
||
}
|
||
|
||
|
||
}
|