Compare commits
No commits in common. "871741b76453434748e3dc70ee0a3cfb0f670c77" and "30f4635072b9195fd2d3e13674c43735628b9284" have entirely different histories.
871741b764
...
30f4635072
@ -10,345 +10,207 @@ import androidx.compose.runtime.*
|
||||
import androidx.compose.ui.Alignment
|
||||
import androidx.compose.ui.Modifier
|
||||
import androidx.compose.ui.graphics.Color
|
||||
import androidx.compose.ui.text.input.PasswordVisualTransformation
|
||||
import androidx.compose.ui.unit.dp
|
||||
import androidx.compose.ui.window.Window
|
||||
import androidx.compose.ui.window.application
|
||||
import io.ktor.client.*
|
||||
import io.ktor.client.call.*
|
||||
import io.ktor.client.engine.cio.*
|
||||
import io.ktor.client.plugins.*
|
||||
import io.ktor.client.plugins.contentnegotiation.*
|
||||
import io.ktor.client.request.*
|
||||
import io.ktor.client.request.forms.MultiPartFormDataContent
|
||||
import io.ktor.client.request.forms.formData
|
||||
import io.ktor.client.statement.*
|
||||
import io.ktor.http.*
|
||||
import io.ktor.serialization.kotlinx.json.*
|
||||
import kotlinx.coroutines.Dispatchers
|
||||
import kotlinx.coroutines.launch
|
||||
import kotlinx.serialization.Serializable
|
||||
import kotlinx.serialization.encodeToString
|
||||
import kotlinx.serialization.json.Json
|
||||
import org.jsoup.Jsoup
|
||||
import org.openqa.selenium.By
|
||||
import org.openqa.selenium.WebDriverException
|
||||
import org.openqa.selenium.chrome.ChromeDriver
|
||||
import org.openqa.selenium.chrome.ChromeOptions
|
||||
import java.io.File
|
||||
import java.text.SimpleDateFormat
|
||||
import java.util.*
|
||||
import java.util.prefs.Preferences
|
||||
import kotlin.io.path.Path
|
||||
import kotlin.io.path.readText
|
||||
|
||||
// --- 데이터 클래스 정의 ---
|
||||
|
||||
@Serializable data class AnythingLLMChatRequest(val message: String)
|
||||
@Serializable data class AnythingLLMChatResponse(val textResponse: String)
|
||||
|
||||
@Serializable
|
||||
data class UploadResponse(val documents: List<Document>) {
|
||||
@Serializable
|
||||
data class Document(val id: String, val location: String)
|
||||
}
|
||||
|
||||
@Serializable
|
||||
data class DeleteDocumentsRequest(val deletes: List<String>)
|
||||
|
||||
data class SearchResult(val title: String, val url: String)
|
||||
|
||||
// ⭐️ [추가] 스크랩된 모든 정보를 담는 데이터 클래스 (JSON 저장용)
|
||||
@Serializable
|
||||
data class ScrapedData(
|
||||
val sourceUrl: String,
|
||||
val imageUrl: String?,
|
||||
val content: String
|
||||
)
|
||||
|
||||
|
||||
// --- 전역 변수 및 헬퍼 ---
|
||||
|
||||
private val httpClient = HttpClient(CIO) {
|
||||
install(ContentNegotiation) {
|
||||
json(Json { isLenient = true; ignoreUnknownKeys = true; prettyPrint = true })
|
||||
}
|
||||
install(HttpTimeout) {
|
||||
requestTimeoutMillis = 300000
|
||||
json(Json { isLenient = true; ignoreUnknownKeys = true })
|
||||
}
|
||||
}
|
||||
private val jsonParser = Json { prettyPrint = true; ignoreUnknownKeys = true }
|
||||
private var driver: ChromeDriver? = null
|
||||
private val prefs: Preferences = Preferences.userNodeForPackage(Unit::class.java)
|
||||
private const val PREF_FOLDER_PATH = "folder_path"
|
||||
private const val PREF_API_KEY = "api_key"
|
||||
private const val PREF_WORKSPACE_SLUG = "workspace_slug"
|
||||
// ⭐️ [추가] LLM 모델 이름 저장을 위한 키
|
||||
private const val PREF_MODEL_NAME = "model_name"
|
||||
|
||||
|
||||
fun logMessage(logs: MutableList<String>, message: String) {
|
||||
val timestamp = SimpleDateFormat("HH:mm:ss", Locale.getDefault()).format(Date())
|
||||
logs.add(0, "$timestamp: $message")
|
||||
}
|
||||
|
||||
// --- 브라우저 관리 ---
|
||||
private fun getChromeDriver(options: ChromeOptions): ChromeDriver {
|
||||
try {
|
||||
driver?.title
|
||||
} catch (e: WebDriverException) {
|
||||
driver = null
|
||||
}
|
||||
if (driver == null) {
|
||||
driver = ChromeDriver(options)
|
||||
}
|
||||
return driver!!
|
||||
}
|
||||
|
||||
private fun quitChromeDriver() {
|
||||
driver?.quit()
|
||||
driver = null
|
||||
}
|
||||
|
||||
// --- 핵심 기능 함수들 ---
|
||||
suspend fun fetchGoogleTrends(logs: MutableList<String>, isBrowserVisible: Boolean, keepSession: Boolean): List<String> {
|
||||
|
||||
suspend fun fetchGoogleTrends(logs: MutableList<String>, isBrowserVisible: Boolean): List<String> {
|
||||
logMessage(logs, "Google Trends 페이지 스크랩 시작...")
|
||||
val trendsUrl = "https://trends.google.co.kr/trends/trendingsearches/daily?geo=KR"
|
||||
val options = ChromeOptions().apply {
|
||||
if (!isBrowserVisible) addArguments("--headless=new")
|
||||
if (!isBrowserVisible) {
|
||||
addArguments("--headless=new")
|
||||
}
|
||||
addArguments("--disable-gpu")
|
||||
}
|
||||
val currentDriver = getChromeDriver(options)
|
||||
val driver = ChromeDriver(options)
|
||||
val keywords = mutableListOf<String>()
|
||||
return try {
|
||||
currentDriver.get(trendsUrl)
|
||||
driver.get(trendsUrl)
|
||||
Thread.sleep(2000)
|
||||
val elements = currentDriver.findElements(By.xpath("//tr[count(td)=7]/td[2]"))
|
||||
elements.mapNotNullTo(keywords) { it.text.takeIf { it.isNotBlank() } }
|
||||
val elements = driver.findElements(By.xpath("//tr[count(td)=7]/td[2]"))
|
||||
for (element in elements) {
|
||||
val keywordText = element.text
|
||||
if (keywordText.isNotBlank()) {
|
||||
keywords.add(keywordText)
|
||||
}
|
||||
}
|
||||
logMessage(logs, "✅ Google Trends 키워드 ${keywords.size}개 스크랩 완료.")
|
||||
keywords
|
||||
} catch (e: Exception) {
|
||||
logMessage(logs, "❌ Google Trends 스크랩 오류: ${e.message}")
|
||||
quitChromeDriver()
|
||||
e.printStackTrace()
|
||||
emptyList()
|
||||
} finally {
|
||||
if (!keepSession) quitChromeDriver()
|
||||
driver.quit()
|
||||
}
|
||||
}
|
||||
|
||||
suspend fun searchOnGoogle(keyword: String, logs: MutableList<String>, isBrowserVisible: Boolean, keepSession: Boolean): List<SearchResult> {
|
||||
suspend fun searchOnGoogle(keyword: String, logs: MutableList<String>, isBrowserVisible: Boolean): List<SearchResult> {
|
||||
logMessage(logs, "'$keyword' 키워드로 Google 검색 시작...")
|
||||
val options = ChromeOptions().apply {
|
||||
if (!isBrowserVisible) addArguments("--headless=new")
|
||||
if (!isBrowserVisible) {
|
||||
addArguments("--headless=new")
|
||||
}
|
||||
addArguments("--disable-gpu")
|
||||
}
|
||||
val currentDriver = getChromeDriver(options)
|
||||
val driver = ChromeDriver(options)
|
||||
val results = mutableListOf<SearchResult>()
|
||||
try {
|
||||
currentDriver.get("https://www.google.com/search?q=$keyword")
|
||||
driver.get("https://www.google.com/search?q=$keyword")
|
||||
Thread.sleep(2000)
|
||||
val resultElements = currentDriver.findElements(By.cssSelector("div[data-rpos]"))
|
||||
val resultElements = driver.findElements(By.cssSelector("div[data-rpos]"))
|
||||
for (element in resultElements.take(10)) {
|
||||
try {
|
||||
val title = element.findElement(By.cssSelector("h3")).text
|
||||
val url = element.findElement(By.cssSelector("a")).getAttribute("href")
|
||||
if (title.isNotBlank() && url.isNotBlank()) results.add(SearchResult(title, url))
|
||||
} catch (e: Exception) { /* 개별 오류 무시 */
|
||||
val titleElement = element.findElement(By.cssSelector("h3"))
|
||||
val linkElement = element.findElement(By.cssSelector("a"))
|
||||
val title = titleElement.text
|
||||
val url = linkElement.getAttribute("href")
|
||||
if (title.isNotBlank() && url.isNotBlank()) {
|
||||
results.add(SearchResult(title, url))
|
||||
}
|
||||
} catch (e: Exception) {
|
||||
// 개별 검색 결과 파싱 오류는 무시
|
||||
}
|
||||
}
|
||||
logMessage(logs, "✅ '$keyword' 검색 결과 ${results.size}개 수집 완료.")
|
||||
} catch (e: Exception) {
|
||||
logMessage(logs, "❌ Google 검색 중 오류: ${e.message}")
|
||||
quitChromeDriver()
|
||||
e.printStackTrace()
|
||||
} finally {
|
||||
if (!keepSession) quitChromeDriver()
|
||||
driver.quit()
|
||||
}
|
||||
return results
|
||||
}
|
||||
|
||||
// ⭐️ [수정] 반환 타입을 ScrapedData로 변경하고 이미지 URL 추출 로직 추가
|
||||
suspend fun scrapeArticleByUrl(url: String, logs: MutableList<String>, isBrowserVisible: Boolean, keepSession: Boolean): ScrapedData? {
|
||||
suspend fun scrapeArticleByUrl(url: String, logs: MutableList<String>, isBrowserVisible: Boolean): String {
|
||||
logMessage(logs, "URL 스크랩 시작: $url")
|
||||
val options = ChromeOptions().apply {
|
||||
if (!isBrowserVisible) addArguments("--headless=new")
|
||||
if (!isBrowserVisible) {
|
||||
addArguments("--headless=new")
|
||||
}
|
||||
addArguments("--disable-gpu")
|
||||
}
|
||||
val currentDriver = getChromeDriver(options)
|
||||
val driver = ChromeDriver(options)
|
||||
return try {
|
||||
currentDriver.get(url)
|
||||
driver.get(url)
|
||||
Thread.sleep(2000)
|
||||
val doc = Jsoup.parse(currentDriver.pageSource)
|
||||
val finalHtml = driver.pageSource
|
||||
val doc = Jsoup.parse(finalHtml)
|
||||
val articleContent = doc.select("article, .article-body, #article_body, .news-article-body-view").text()
|
||||
// OG 태그에서 대표 이미지 URL 추출
|
||||
val imageUrl = doc.select("meta[property=og:image]").attr("content")
|
||||
|
||||
if (articleContent.isBlank()) {
|
||||
logMessage(logs, "⚠️ 기사 본문을 찾을 수 없습니다.")
|
||||
null
|
||||
"기사 본문을 찾을 수 없습니다."
|
||||
} else {
|
||||
logMessage(logs, "✅ URL 스크랩 완료. (이미지: ${if (imageUrl.isNotBlank()) "있음" else "없음"})")
|
||||
ScrapedData(
|
||||
sourceUrl = url,
|
||||
imageUrl = imageUrl.ifBlank { null },
|
||||
content = articleContent
|
||||
)
|
||||
logMessage(logs, "✅ URL 스크랩 완료. (총 ${articleContent.length}자)")
|
||||
articleContent
|
||||
}
|
||||
} catch (e: Exception) {
|
||||
logMessage(logs, "❌ URL 스크랩 중 오류: ${e.message}")
|
||||
quitChromeDriver()
|
||||
null
|
||||
e.printStackTrace()
|
||||
"페이지 스크랩 중 오류 발생: ${e.message}"
|
||||
} finally {
|
||||
if (!keepSession) quitChromeDriver()
|
||||
driver.quit()
|
||||
}
|
||||
}
|
||||
|
||||
suspend fun generateBlogPostWithLocalLLM(fileNames: List<String>, userDirection: String, logs: MutableList<String>): String {
|
||||
logMessage(logs, "LLM 블로그 글 생성 요청 (${fileNames.size}개 파일 기반)...")
|
||||
val apiKey = System.getenv("ANYTHINGLLM_API_KEY") ?: "YOUR_API_KEY_HERE"
|
||||
val fileNamesString = fileNames.joinToString(", ")
|
||||
|
||||
suspend fun uploadFilesToLLM(files: List<File>, logs: MutableList<String>, apiKey: String, workspaceSlug: String): List<String> {
|
||||
logMessage(logs, "LLM에 파일 ${files.size}개 순차 업로드 시작...")
|
||||
val uploadedDocIds = mutableListOf<String>()
|
||||
|
||||
for (file in files) {
|
||||
logMessage(logs, " - '${file.name}' 업로드 중...")
|
||||
try {
|
||||
val response: HttpResponse = httpClient.post("http://localhost:3001/api/v1/document/upload") {
|
||||
header("Authorization", "Bearer $apiKey")
|
||||
setBody(MultiPartFormDataContent(
|
||||
formData {
|
||||
append("addToWorkspaces", workspaceSlug)
|
||||
// ⭐️ [수정] JSON 파일의 내용을 텍스트로 읽어 전송
|
||||
val scrapedData = jsonParser.decodeFromString<ScrapedData>(file.readText())
|
||||
append("file", scrapedData.content.toByteArray(), Headers.build {
|
||||
append(HttpHeaders.ContentType, ContentType.Text.Plain)
|
||||
append(HttpHeaders.ContentDisposition, "filename=\"${file.name}\"")
|
||||
})
|
||||
}
|
||||
))
|
||||
}
|
||||
|
||||
if (response.status.isSuccess()) {
|
||||
val uploadResponse = response.body<UploadResponse>()
|
||||
val docId = uploadResponse.documents.firstOrNull()?.id
|
||||
if (docId != null) {
|
||||
uploadedDocIds.add(docId)
|
||||
logMessage(logs, " ✅ '${file.name}' 업로드 성공 (ID: $docId).")
|
||||
} else {
|
||||
logMessage(logs, " ❌ '${file.name}' 업로드 응답에 문서 정보가 없습니다.")
|
||||
}
|
||||
} else {
|
||||
logMessage(logs, " ❌ '${file.name}' 업로드 실패: ${response.status} - ${response.bodyAsText()}")
|
||||
}
|
||||
} catch (e: Exception) {
|
||||
logMessage(logs, " ❌ '${file.name}' 업로드 중 오류: ${e.message}")
|
||||
}
|
||||
}
|
||||
|
||||
logMessage(logs, "총 ${files.size}개 중 ${uploadedDocIds.size}개 파일 업로드 완료.")
|
||||
return uploadedDocIds
|
||||
}
|
||||
|
||||
// ⭐️ [수정] ScrapedData 리스트와 modelName을 파라미터로 받도록 변경
|
||||
suspend fun generateBlogPostWithLocalLLM(scrapedDataList: List<ScrapedData>, userDirection: String, modelName: String, logs: MutableList<String>, apiKey: String, workspaceSlug: String): String {
|
||||
logMessage(logs, "LLM 블로그 글 생성 요청...")
|
||||
|
||||
// ⭐️ [수정] LLM에게 전달할 참고자료 형식 변경
|
||||
val referencesText = scrapedDataList.mapIndexed { index, data ->
|
||||
"""
|
||||
[참고자료 ${index + 1}]
|
||||
- 원문 출처: ${data.sourceUrl}
|
||||
- 대표 이미지: ${data.imageUrl ?: "없음"}
|
||||
- 내용 요약: ${data.content.take(500)}...
|
||||
""".trimIndent()
|
||||
}.joinToString("\n\n")
|
||||
|
||||
// ⭐️ [수정] 최종 프롬프트 수정
|
||||
val finalPrompt = """
|
||||
당신은 최신 정보를 종합하여 SEO에 최적화된 블로그 글을 작성하는 전문 작가입니다.
|
||||
아래 '참고자료'와 '요청사항'을 바탕으로, 독자들이 흥미를 느낄만한 멋진 블로그 글을 작성해주세요.
|
||||
내 지식 베이스에 있는 다음 문서들을 종합적으로 참고해서 아래 '요청사항'에 맞춰 SEO에 최적화된 블로그 글을 작성해줘. 제목도 2~3개 추천해줘.
|
||||
|
||||
--- 참고자료 ---
|
||||
$referencesText
|
||||
--- 참고 문서 ---
|
||||
$fileNamesString
|
||||
|
||||
--- 요청사항 ---
|
||||
1. 사용자 요청: "$userDirection"
|
||||
2. 참고자료의 내용을 종합적으로 활용하여 하나의 완성된 글을 작성해주세요.
|
||||
3. 글의 시작 부분에 독자의 시선을 사로잡을 만한 제목을 2~3개 추천해주세요.
|
||||
4. 대표 이미지가 있는 경우, 글의 적절한 위치에 마크다운 형식 ``으로 이미지를 삽입해주세요.
|
||||
5. 글의 마지막에는 아래 형식에 맞춰 출처 정보를 반드시 포함해주세요.
|
||||
|
||||
---
|
||||
- 원문 출처:
|
||||
${scrapedDataList.map { "- ${it.sourceUrl}" }.joinToString("\n")}
|
||||
- 이미지 출처: ${scrapedDataList.firstNotNullOfOrNull { it.imageUrl } ?: "없음"}
|
||||
- 이 글은 ${modelName} 모델을 활용하여 작성되었습니다.
|
||||
|
||||
$userDirection
|
||||
""".trimIndent()
|
||||
|
||||
val requestBody = AnythingLLMChatRequest(message = finalPrompt)
|
||||
try {
|
||||
val response: HttpResponse = httpClient.post("http://localhost:3001/api/v1/workspace/$workspaceSlug/chat") {
|
||||
return try {
|
||||
val response: AnythingLLMChatResponse = httpClient.post("http://localhost:3001/api/v1/workspace/my/chat") {
|
||||
header("Authorization", "Bearer $apiKey")
|
||||
contentType(ContentType.Application.Json)
|
||||
setBody(requestBody)
|
||||
}
|
||||
|
||||
val responseBodyText = response.bodyAsText()
|
||||
logMessage(logs, "LLM 응답 수신: $responseBodyText")
|
||||
|
||||
val chatResponse = jsonParser.decodeFromString<AnythingLLMChatResponse>(responseBodyText)
|
||||
|
||||
}.body()
|
||||
logMessage(logs, "✅ LLM 블로그 글 생성 완료.")
|
||||
return chatResponse.textResponse
|
||||
|
||||
response.textResponse
|
||||
} catch (e: Exception) {
|
||||
logMessage(logs, "❌ LLM 호출 중 오류: ${e.message}")
|
||||
return "블로그 글 생성 실패: ${e.message}"
|
||||
e.printStackTrace()
|
||||
"블로그 글 생성 실패: ${e.message}"
|
||||
}
|
||||
}
|
||||
|
||||
suspend fun cleanupLLMWorkspace(docIds: List<String>, logs: MutableList<String>, apiKey: String) {
|
||||
if (docIds.isEmpty()) return
|
||||
logMessage(logs, "LLM 워크스페이스 정리 시작 (파일 ${docIds.size}개 삭제)...")
|
||||
fun saveContentToFile(keyword: String, content: String, logs: MutableList<String>) {
|
||||
try {
|
||||
val response: HttpResponse = httpClient.post("http://localhost:3001/api/v1/document/delete") {
|
||||
header("Authorization", "Bearer $apiKey")
|
||||
contentType(ContentType.Application.Json)
|
||||
setBody(DeleteDocumentsRequest(deletes = docIds))
|
||||
}
|
||||
if (response.status.isSuccess()) {
|
||||
logMessage(logs, "✅ LLM 워크스페이스 정리 완료.")
|
||||
} else {
|
||||
logMessage(logs, "❌ LLM 워크스페이스 정리 실패: ${response.status} - ${response.bodyAsText()}")
|
||||
}
|
||||
} catch (e: Exception) {
|
||||
logMessage(logs, "❌ LLM 워크스페이스 정리 중 오류: ${e.message}")
|
||||
}
|
||||
}
|
||||
|
||||
// ⭐️ [수정] ScrapedData를 JSON 파일로 저장하는 함수
|
||||
fun saveDataToJsonFile(keyword: String, data: ScrapedData, logs: MutableList<String>, folderPath: String) {
|
||||
try {
|
||||
val directory = File(folderPath)
|
||||
val directory = File("/Users/jibumhan/autoblog_content")
|
||||
if (!directory.exists()) directory.mkdirs()
|
||||
val sanitizedKeyword = keyword.replace(Regex("[^A-Za-z0-9ㄱ-ㅎㅏ-ㅣ가-힣]"), "")
|
||||
// ⭐️ [수정] 확장자를 .json으로 변경
|
||||
val fileName = "${sanitizedKeyword}_${System.currentTimeMillis()}.json"
|
||||
val fileName = "${sanitizedKeyword}_${System.currentTimeMillis()}.txt"
|
||||
val file = File(directory, fileName)
|
||||
file.writeText(jsonParser.encodeToString(data))
|
||||
logMessage(logs, "✅ '${file.path}'에 스크랩 데이터 저장 완료.")
|
||||
file.writeText(content)
|
||||
logMessage(logs, "✅ '${file.path}'에 스크랩 내용 저장 완료.")
|
||||
} catch (e: Exception) {
|
||||
logMessage(logs, "❌ 파일 저장 중 오류: ${e.message}")
|
||||
e.printStackTrace()
|
||||
}
|
||||
}
|
||||
|
||||
// ⭐️ [수정] .json 파일을 읽어오는 함수
|
||||
fun loadScrapedJsonFiles(logs: MutableList<String>, folderPath: String): List<File> {
|
||||
logMessage(logs, "'$folderPath'에서 파일 목록 로딩...")
|
||||
fun loadScrapedFiles(logs: MutableList<String>): List<File> {
|
||||
logMessage(logs, "스크랩된 파일 목록 로딩...")
|
||||
return try {
|
||||
val directory = File(folderPath)
|
||||
val directory = File("scraped_articles")
|
||||
if (!directory.exists() || !directory.isDirectory) {
|
||||
logMessage(logs, "⚠️ '$folderPath' 폴더를 찾을 수 없습니다.")
|
||||
logMessage(logs, "⚠️ 'scraped_articles' 폴더를 찾을 수 없습니다.")
|
||||
return emptyList()
|
||||
}
|
||||
// ⭐️ [수정] .json 파일만 대상으로 함
|
||||
val files = directory.listFiles { _, name -> name.endsWith(".json") }
|
||||
val files = directory.listFiles { _, name -> name.endsWith(".txt") }
|
||||
?.sortedByDescending { it.lastModified() }
|
||||
?: emptyList()
|
||||
logMessage(logs, "✅ 파일 ${files.size}개 로딩 완료.")
|
||||
@ -359,13 +221,16 @@ fun loadScrapedJsonFiles(logs: MutableList<String>, folderPath: String): List<Fi
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
// --- UI 컴포넌트 ---
|
||||
|
||||
@Composable
|
||||
fun App() {
|
||||
var tabIndex by remember { mutableStateOf(0) }
|
||||
val tabs = listOf("워크플로우", "통신 로그", "블로그 결과")
|
||||
val coroutineScope = rememberCoroutineScope()
|
||||
|
||||
// --- 상태 관리 ---
|
||||
var keywords by remember { mutableStateOf<List<String>>(emptyList()) }
|
||||
var searchResults by remember { mutableStateOf<List<SearchResult>>(emptyList()) }
|
||||
var blogPostResult by remember { mutableStateOf("LLM으로부터 생성된 블로그 글이 여기에 표시됩니다.") }
|
||||
@ -373,48 +238,30 @@ fun App() {
|
||||
var isLoading by remember { mutableStateOf(false) }
|
||||
var selectedKeyword by remember { mutableStateOf("") }
|
||||
var userPrompt by remember { mutableStateOf("친근하고 유용한 정보 전달 스타일로 작성해줘.") }
|
||||
var isBrowserVisible by remember { mutableStateOf(true) } // 브라우저 가시성 상태
|
||||
|
||||
var isBrowserVisible by remember { mutableStateOf(true) }
|
||||
var keepBrowserSession by remember { mutableStateOf(false) }
|
||||
var scrapedFolderPath by remember { mutableStateOf(prefs.get(PREF_FOLDER_PATH, "scraped_articles")) }
|
||||
var apiKey by remember { mutableStateOf(prefs.get(PREF_API_KEY, "")) }
|
||||
var workspaceSlug by remember { mutableStateOf(prefs.get(PREF_WORKSPACE_SLUG, "my")) }
|
||||
// ⭐️ [추가] 모델 이름 상태 변수
|
||||
var modelName by remember { mutableStateOf(prefs.get(PREF_MODEL_NAME, "gpt-4o")) }
|
||||
|
||||
// 파일 관리 상태
|
||||
var scrapedFiles by remember { mutableStateOf<List<File>>(emptyList()) }
|
||||
var selectedFiles by remember { mutableStateOf<Set<File>>(emptySet()) }
|
||||
var viewedFileContent by remember { mutableStateOf("파일을 선택하면 내용이 여기에 표시됩니다.") }
|
||||
|
||||
LaunchedEffect(scrapedFolderPath) {
|
||||
scrapedFiles = loadScrapedJsonFiles(logMessages, scrapedFolderPath)
|
||||
// 앱 시작 시 파일 목록 로드
|
||||
LaunchedEffect(Unit) {
|
||||
scrapedFiles = loadScrapedFiles(logMessages)
|
||||
}
|
||||
LaunchedEffect(scrapedFolderPath) { prefs.put(PREF_FOLDER_PATH, scrapedFolderPath) }
|
||||
LaunchedEffect(apiKey) { prefs.put(PREF_API_KEY, apiKey) }
|
||||
LaunchedEffect(workspaceSlug) { prefs.put(PREF_WORKSPACE_SLUG, workspaceSlug) }
|
||||
// ⭐️ [추가] 모델 이름 변경 시 자동 저장
|
||||
LaunchedEffect(modelName) { prefs.put(PREF_MODEL_NAME, modelName) }
|
||||
|
||||
MaterialTheme {
|
||||
Column(modifier = Modifier.fillMaxSize()) {
|
||||
Column(modifier = Modifier.padding(8.dp).border(1.dp, Color.LightGray).padding(8.dp)) {
|
||||
Row(verticalAlignment = Alignment.CenterVertically) {
|
||||
Checkbox(checked = isBrowserVisible, onCheckedChange = { isBrowserVisible = it })
|
||||
Text("브라우저 화면 보기", modifier = Modifier.clickable { isBrowserVisible = !isBrowserVisible }.weight(1f))
|
||||
Checkbox(checked = keepBrowserSession, onCheckedChange = { keepBrowserSession = it })
|
||||
Text("브라우저 세션 유지", modifier = Modifier.clickable { keepBrowserSession = !keepBrowserSession }.weight(1f))
|
||||
}
|
||||
Spacer(Modifier.height(8.dp))
|
||||
OutlinedTextField(value = scrapedFolderPath, onValueChange = { scrapedFolderPath = it }, label = { Text("스크랩 저장 폴더 경로") }, modifier = Modifier.fillMaxWidth(), singleLine = true)
|
||||
Spacer(Modifier.height(4.dp))
|
||||
Row(Modifier.fillMaxWidth()) {
|
||||
OutlinedTextField(value = apiKey, onValueChange = { apiKey = it }, label = { Text("AnythingLLM API Key") }, modifier = Modifier.weight(1f), singleLine = true, visualTransformation = PasswordVisualTransformation())
|
||||
Spacer(Modifier.width(4.dp))
|
||||
OutlinedTextField(value = workspaceSlug, onValueChange = { workspaceSlug = it }, label = { Text("Workspace Slug") }, modifier = Modifier.weight(1f), singleLine = true)
|
||||
}
|
||||
Spacer(Modifier.height(4.dp))
|
||||
// ⭐️ [추가] 모델 이름 입력 필드
|
||||
OutlinedTextField(value = modelName, onValueChange = { modelName = it }, label = { Text("LLM 모델 이름") }, modifier = Modifier.fillMaxWidth(), singleLine = true)
|
||||
// --- 전역 설정 (체크박스) ---
|
||||
Row(
|
||||
modifier = Modifier.fillMaxWidth().padding(horizontal = 8.dp, vertical = 4.dp),
|
||||
verticalAlignment = Alignment.CenterVertically
|
||||
) {
|
||||
Checkbox(
|
||||
checked = isBrowserVisible,
|
||||
onCheckedChange = { isBrowserVisible = it }
|
||||
)
|
||||
Text("브라우저 화면 보기", modifier = Modifier.clickable { isBrowserVisible = !isBrowserVisible })
|
||||
}
|
||||
Divider()
|
||||
|
||||
@ -423,15 +270,21 @@ fun App() {
|
||||
Tab(text = { Text(title) }, selected = tabIndex == index, onClick = { tabIndex = index })
|
||||
}
|
||||
}
|
||||
|
||||
when (tabIndex) {
|
||||
0 -> WorkflowTab(
|
||||
isLoading = isLoading, keywords = keywords, searchResults = searchResults,
|
||||
scrapedFiles = scrapedFiles, selectedFiles = selectedFiles, viewedFileContent = viewedFileContent,
|
||||
userPrompt = userPrompt, onUserPromptChange = { userPrompt = it },
|
||||
isLoading = isLoading,
|
||||
keywords = keywords,
|
||||
searchResults = searchResults,
|
||||
scrapedFiles = scrapedFiles,
|
||||
selectedFiles = selectedFiles,
|
||||
viewedFileContent = viewedFileContent,
|
||||
userPrompt = userPrompt,
|
||||
onUserPromptChange = { userPrompt = it },
|
||||
onFetchTrends = {
|
||||
coroutineScope.launch(Dispatchers.IO) {
|
||||
isLoading = true
|
||||
keywords = fetchGoogleTrends(logMessages, isBrowserVisible, keepBrowserSession)
|
||||
keywords = fetchGoogleTrends(logMessages, isBrowserVisible)
|
||||
isLoading = false
|
||||
}
|
||||
},
|
||||
@ -439,41 +292,37 @@ fun App() {
|
||||
selectedKeyword = keyword
|
||||
coroutineScope.launch(Dispatchers.IO) {
|
||||
isLoading = true
|
||||
searchResults = searchOnGoogle(keyword, logMessages, isBrowserVisible, keepBrowserSession)
|
||||
searchResults = searchOnGoogle(keyword, logMessages, isBrowserVisible)
|
||||
isLoading = false
|
||||
}
|
||||
},
|
||||
onSearchResultSelect = { result ->
|
||||
coroutineScope.launch(Dispatchers.IO) {
|
||||
isLoading = true
|
||||
// ⭐️ [수정] ScrapedData를 받아 JSON으로 저장
|
||||
val data = scrapeArticleByUrl(result.url, logMessages, isBrowserVisible, keepBrowserSession)
|
||||
if (data != null) {
|
||||
saveDataToJsonFile(selectedKeyword, data, logMessages, scrapedFolderPath)
|
||||
scrapedFiles = loadScrapedJsonFiles(logMessages, scrapedFolderPath)
|
||||
val content = scrapeArticleByUrl(result.url, logMessages, isBrowserVisible)
|
||||
if (!content.contains("오류 발생")) {
|
||||
saveContentToFile(selectedKeyword, content, logMessages)
|
||||
scrapedFiles = loadScrapedFiles(logMessages) // 저장 후 목록 새로고침
|
||||
}
|
||||
isLoading = false
|
||||
}
|
||||
},
|
||||
onRefreshFiles = {
|
||||
coroutineScope.launch(Dispatchers.IO) {
|
||||
scrapedFiles = loadScrapedJsonFiles(logMessages, scrapedFolderPath)
|
||||
scrapedFiles = loadScrapedFiles(logMessages)
|
||||
}
|
||||
},
|
||||
onFileSelectToggle = { file, isSelected ->
|
||||
selectedFiles = if (isSelected) selectedFiles + file else selectedFiles - file
|
||||
selectedFiles = if (isSelected) {
|
||||
selectedFiles + file
|
||||
} else {
|
||||
selectedFiles - file
|
||||
}
|
||||
},
|
||||
onFileView = { file ->
|
||||
coroutineScope.launch(Dispatchers.IO) {
|
||||
try {
|
||||
// ⭐️ [수정] JSON을 읽어 본문 내용만 표시
|
||||
val data = jsonParser.decodeFromString<ScrapedData>(Path(file.absolutePath).readText(Charsets.UTF_8))
|
||||
viewedFileContent = """
|
||||
[원문] ${data.sourceUrl}
|
||||
[이미지] ${data.imageUrl ?: "없음"}
|
||||
--------------------
|
||||
${data.content}
|
||||
""".trimIndent()
|
||||
viewedFileContent = Path(file.absolutePath).readText(Charsets.UTF_8)
|
||||
} catch (e: Exception) {
|
||||
logMessage(logMessages, "❌ 파일 읽기 오류: ${e.message}")
|
||||
viewedFileContent = "파일을 읽는 중 오류가 발생했습니다."
|
||||
@ -481,32 +330,14 @@ fun App() {
|
||||
}
|
||||
},
|
||||
onGeneratePost = {
|
||||
if (apiKey.isBlank() || workspaceSlug.isBlank() || modelName.isBlank()) {
|
||||
logMessage(logMessages, "⚠️ API 키, 슬러그, 모델 이름을 모두 입력해주세요.")
|
||||
return@WorkflowTab
|
||||
}
|
||||
if (selectedFiles.isNotEmpty()) {
|
||||
coroutineScope.launch(Dispatchers.IO) {
|
||||
isLoading = true
|
||||
var uploadedDocIds: List<String> = emptyList()
|
||||
try {
|
||||
// ⭐️ [수정] 선택된 JSON 파일들을 ScrapedData로 변환
|
||||
val selectedData = selectedFiles.map { jsonParser.decodeFromString<ScrapedData>(it.readText()) }
|
||||
|
||||
uploadedDocIds = uploadFilesToLLM(selectedFiles.toList(), logMessages, apiKey, workspaceSlug)
|
||||
if (uploadedDocIds.isNotEmpty()) {
|
||||
// ⭐️ [수정] ScrapedData 리스트와 모델 이름을 전달
|
||||
blogPostResult = generateBlogPostWithLocalLLM(selectedData, userPrompt, modelName, logMessages, apiKey, workspaceSlug)
|
||||
tabIndex = 2
|
||||
} else {
|
||||
logMessage(logMessages, "⚠️ 파일 업로드에 실패하여 글 생성을 중단합니다.")
|
||||
blogPostResult = "파일 업로드 실패"
|
||||
}
|
||||
} finally {
|
||||
cleanupLLMWorkspace(uploadedDocIds, logMessages, apiKey)
|
||||
val fileNames = selectedFiles.map { it.name }
|
||||
blogPostResult = generateBlogPostWithLocalLLM(fileNames, userPrompt, logMessages)
|
||||
tabIndex = 2 // 결과 탭으로 이동
|
||||
isLoading = false
|
||||
}
|
||||
}
|
||||
} else {
|
||||
logMessage(logMessages, "⚠️ 블로그 글을 생성할 파일을 선택해주세요.")
|
||||
}
|
||||
@ -521,23 +352,36 @@ fun App() {
|
||||
|
||||
@Composable
|
||||
fun WorkflowTab(
|
||||
isLoading: Boolean, keywords: List<String>, searchResults: List<SearchResult>,
|
||||
scrapedFiles: List<File>, selectedFiles: Set<File>, viewedFileContent: String,
|
||||
userPrompt: String, onUserPromptChange: (String) -> Unit, onFetchTrends: () -> Unit,
|
||||
onKeywordSelect: (String) -> Unit, onSearchResultSelect: (SearchResult) -> Unit,
|
||||
onRefreshFiles: () -> Unit, onFileSelectToggle: (File, Boolean) -> Unit,
|
||||
onFileView: (File) -> Unit, onGeneratePost: () -> Unit
|
||||
isLoading: Boolean,
|
||||
keywords: List<String>,
|
||||
searchResults: List<SearchResult>,
|
||||
scrapedFiles: List<File>,
|
||||
selectedFiles: Set<File>,
|
||||
viewedFileContent: String,
|
||||
userPrompt: String,
|
||||
onUserPromptChange: (String) -> Unit,
|
||||
onFetchTrends: () -> Unit,
|
||||
onKeywordSelect: (String) -> Unit,
|
||||
onSearchResultSelect: (SearchResult) -> Unit,
|
||||
onRefreshFiles: () -> Unit,
|
||||
onFileSelectToggle: (File, Boolean) -> Unit,
|
||||
onFileView: (File) -> Unit,
|
||||
onGeneratePost: () -> Unit
|
||||
) {
|
||||
Box(modifier = Modifier.fillMaxSize()) {
|
||||
Row(modifier = Modifier.fillMaxSize()) {
|
||||
// 1열: 트렌드 키워드
|
||||
Column(modifier = Modifier.weight(1f).border(1.dp, Color.LightGray).padding(4.dp)) {
|
||||
Button(onClick = onFetchTrends, modifier = Modifier.fillMaxWidth(), enabled = !isLoading) { Text("트렌드 가져오기") }
|
||||
Button(onClick = onFetchTrends, modifier = Modifier.fillMaxWidth(), enabled = !isLoading) {
|
||||
Text("트렌드 가져오기")
|
||||
}
|
||||
LazyColumn(modifier = Modifier.fillMaxSize()) {
|
||||
items(keywords) { keyword ->
|
||||
Text(keyword, modifier = Modifier.fillMaxWidth().clickable { if (!isLoading) onKeywordSelect(keyword) }.padding(8.dp))
|
||||
}
|
||||
}
|
||||
}
|
||||
// 2열: 검색 결과
|
||||
Column(modifier = Modifier.weight(1.5f).border(1.dp, Color.LightGray).padding(4.dp)) {
|
||||
Text("검색 결과", style = MaterialTheme.typography.h6, modifier = Modifier.padding(4.dp))
|
||||
LazyColumn(modifier = Modifier.fillMaxSize()) {
|
||||
@ -549,6 +393,7 @@ fun WorkflowTab(
|
||||
}
|
||||
}
|
||||
}
|
||||
// 3열: 스크랩된 파일 목록
|
||||
Column(modifier = Modifier.weight(1.5f).border(1.dp, Color.LightGray).padding(4.dp)) {
|
||||
Row(verticalAlignment = Alignment.CenterVertically) {
|
||||
Text("저장된 파일", style = MaterialTheme.typography.h6, modifier = Modifier.weight(1f).padding(4.dp))
|
||||
@ -556,19 +401,36 @@ fun WorkflowTab(
|
||||
}
|
||||
LazyColumn(modifier = Modifier.fillMaxSize()) {
|
||||
items(scrapedFiles) { file ->
|
||||
Row(modifier = Modifier.fillMaxWidth().clickable { if (!isLoading) onFileView(file) }.padding(vertical = 4.dp), verticalAlignment = Alignment.CenterVertically) {
|
||||
Checkbox(checked = file in selectedFiles, onCheckedChange = { isChecked -> onFileSelectToggle(file, isChecked) }, enabled = !isLoading)
|
||||
Row(
|
||||
modifier = Modifier.fillMaxWidth().clickable { if (!isLoading) onFileView(file) }.padding(vertical = 4.dp),
|
||||
verticalAlignment = Alignment.CenterVertically
|
||||
) {
|
||||
Checkbox(
|
||||
checked = file in selectedFiles,
|
||||
onCheckedChange = { isChecked -> onFileSelectToggle(file, isChecked) },
|
||||
enabled = !isLoading
|
||||
)
|
||||
Text(file.name, modifier = Modifier.padding(start = 4.dp), maxLines = 1)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
// 4열: 내용 확인 및 생성
|
||||
Column(modifier = Modifier.weight(2f).border(1.dp, Color.LightGray).padding(8.dp)) {
|
||||
Text("파일 내용", style = MaterialTheme.typography.h6)
|
||||
Text(text = viewedFileContent, modifier = Modifier.weight(1f).fillMaxWidth().verticalScroll(rememberScrollState()).border(1.dp, Color.LightGray).padding(4.dp), style = MaterialTheme.typography.body2)
|
||||
Text(
|
||||
text = viewedFileContent,
|
||||
modifier = Modifier.weight(1f).fillMaxWidth().verticalScroll(rememberScrollState()).border(1.dp, Color.LightGray).padding(4.dp),
|
||||
style = MaterialTheme.typography.body2
|
||||
)
|
||||
Spacer(Modifier.height(8.dp))
|
||||
Text("LLM 요청사항", style = MaterialTheme.typography.h6)
|
||||
TextField(value = userPrompt, onValueChange = onUserPromptChange, modifier = Modifier.fillMaxWidth().height(100.dp), placeholder = { Text("예: 선택된 파일들을 종합해서...") })
|
||||
TextField(
|
||||
value = userPrompt,
|
||||
onValueChange = onUserPromptChange,
|
||||
modifier = Modifier.fillMaxWidth().height(100.dp),
|
||||
placeholder = { Text("예: 선택된 파일들을 종합해서...") }
|
||||
)
|
||||
Spacer(Modifier.height(8.dp))
|
||||
Button(onClick = onGeneratePost, enabled = !isLoading && selectedFiles.isNotEmpty(), modifier = Modifier.fillMaxWidth()) {
|
||||
Text("블로그 글 생성하기 (${selectedFiles.size}개 파일)")
|
||||
@ -583,7 +445,12 @@ fun WorkflowTab(
|
||||
|
||||
@Composable
|
||||
fun LogTab(logs: List<String>) {
|
||||
TextField(value = logs.joinToString("\n"), onValueChange = {}, readOnly = true, modifier = Modifier.fillMaxSize().padding(8.dp))
|
||||
TextField(
|
||||
value = logs.joinToString("\n"),
|
||||
onValueChange = {},
|
||||
readOnly = true,
|
||||
modifier = Modifier.fillMaxSize().padding(8.dp)
|
||||
)
|
||||
}
|
||||
|
||||
@Composable
|
||||
@ -594,14 +461,14 @@ fun ResultTab(result: String) {
|
||||
}
|
||||
|
||||
// --- 애플리케이션 시작점 ---
|
||||
|
||||
fun main() = application {
|
||||
Window(
|
||||
onCloseRequest = {
|
||||
quitChromeDriver()
|
||||
httpClient.close()
|
||||
exitApplication()
|
||||
},
|
||||
title = "자동 블로그 포스팅 도우미 v4.0 (Final)"
|
||||
title = "자동 블로그 포스팅 도우미 v2.6"
|
||||
) {
|
||||
App()
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user