androidlover5842/TrisolarisServer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Centralize document prompts
All checks were successful
build-and-deploy / build-deploy (push) Successful in 31s
Details

Browse Source
This commit is contained in:
androidlover5842
2026-01-31 02:05:11 +05:30
parent d4a9a06725
commit e0643ce695
2 changed files with 109 additions and 81 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/main/kotlin/com/android/trisolarisserver/controller/DocumentPrompts.kt Normal file
View File

@@ -0,0 +1,13 @@
package com.android.trisolarisserver.controller
object DocumentPrompts {
val NAME = "name" to "NAME? Reply only the name or NONE."
val DOB = "dob" to "DOB? Reply only date or NONE."
val ID_NUMBER = "idNumber" to "ID NUMBER? Reply only number or NONE."
val ADDRESS = "address" to "POSTAL ADDRESS ONLY (street/area/city/state). Ignore IDs, UUIDs, and codes. Reply only address or NONE."
val PIN_CODE = "pinCode" to "POSTAL PIN CODE? Reply only pin or NONE."
val CITY = "city" to "CITY? Reply only city or NONE."
val GENDER = "gender" to "GENDER? Reply only MALE/FEMALE/OTHER or NONE."
val NATIONALITY = "nationality" to "NATIONALITY? Reply only nationality or NONE."
val VEHICLE_NUMBER = "vehicleNumber" to "VEHICLE NUMBER? Reply only number or NONE."
}

177
src/main/kotlin/com/android/trisolarisserver/controller/GuestDocuments.kt
View File

@@ -238,8 +238,8 @@ class GuestDocuments(
val hasAddress = isYes(results["hasAddress"]) val hasAddress = isYes(results["hasAddress"])
if (hasAddress) { if (hasAddress) {
val addressQuestions = linkedMapOf( val addressQuestions = linkedMapOf(
"pinCode" to "POSTAL ADDRESS PIN CODE (6 digit)? Reply only pin or NONE.", DocumentPrompts.PIN_CODE,
"address" to "POSTAL ADDRESS ONLY (street/area/city/state). Ignore IDs, UUIDs, and codes. Reply only address or NONE." DocumentPrompts.ADDRESS
) )
for ((key, question) in addressQuestions) { for ((key, question) in addressQuestions) {
results[key] = llamaClient.ask(imageUrl, question) results[key] = llamaClient.ask(imageUrl, question)
@@ -249,10 +249,10 @@ class GuestDocuments(
val hasGender = isYes(results["hasGenderMentioned"]) val hasGender = isYes(results["hasGenderMentioned"])
if (hasDob && hasGender) { if (hasDob && hasGender) {
val aadharFrontQuestions = linkedMapOf( val aadharFrontQuestions = linkedMapOf(
"name" to "NAME? Reply only the name or NONE.", DocumentPrompts.NAME,
"dob" to "DOB? Reply only date or NONE.", DocumentPrompts.DOB,
"idNumber" to "ID NUMBER? Reply only number or NONE.", DocumentPrompts.ID_NUMBER,
"gender" to "GENDER? Reply only MALE/FEMALE/OTHER or NONE." DocumentPrompts.GENDER
) )
for ((key, question) in aadharFrontQuestions) { for ((key, question) in aadharFrontQuestions) {
results[key] = llamaClient.ask(imageUrl, question) results[key] = llamaClient.ask(imageUrl, question)
@@ -260,17 +260,6 @@ class GuestDocuments(
} }
} else { } else {
val detectionQuestions = linkedMapOf(
"hasDrivingLicence" to "CONTAINS DRIVING LICENCE? Answer YES or NO only.",
"hasTransportDept" to "CONTAINS TRANSPORT DEPARTMENT? Answer YES or NO only.",
"hasElectionCommission" to "CONTAINS ELECTION COMMISSION OF INDIA? Answer YES or NO only.",
"hasIncomeTaxDept" to "CONTAINS INCOME TAX DEPARTMENT? Answer YES or NO only.",
"hasPassport" to "CONTAINS PASSPORT? Answer YES or NO only."
)
for ((key, question) in detectionQuestions) {
results[key] = llamaClient.ask(imageUrl, question)
}
if (isYes(results["isVehiclePhoto"])) { if (isYes(results["isVehiclePhoto"])) {
val vehicleQuestions = linkedMapOf( val vehicleQuestions = linkedMapOf(
"vehicleNumber" to "VEHICLE NUMBER PLATE? Reply only number or NONE." "vehicleNumber" to "VEHICLE NUMBER PLATE? Reply only number or NONE."
@@ -279,77 +268,103 @@ class GuestDocuments(
results[key] = llamaClient.ask(imageUrl, question) results[key] = llamaClient.ask(imageUrl, question)
} }
} else { } else {
results["hasDrivingLicence"] = llamaClient.ask(
imageUrl,
"CONTAINS DRIVING LICENCE? Answer YES or NO only."
)
results["hasTransportDept"] = llamaClient.ask(
imageUrl,
"CONTAINS TRANSPORT DEPARTMENT? Answer YES or NO only."
)
val isDriving = isYes(results["hasDrivingLicence"]) || isYes(results["hasTransportDept"]) val isDriving = isYes(results["hasDrivingLicence"]) || isYes(results["hasTransportDept"])
if (isDriving) { if (isDriving) {
val drivingQuestions = linkedMapOf( val drivingQuestions = linkedMapOf(
"name" to "NAME? Reply only the name or NONE.", DocumentPrompts.NAME,
"dob" to "DOB? Reply only date or NONE.", DocumentPrompts.DOB,
"idNumber" to "DL NUMBER? Reply only number or NONE.", "idNumber" to "DL NUMBER? Reply only number or NONE.",
"address" to "POSTAL ADDRESS ONLY (street/area/city/state). Ignore IDs, UUIDs, and codes. Reply only address or NONE.", DocumentPrompts.ADDRESS,
"pinCode" to "PIN CODE? Reply only pin or NONE.", DocumentPrompts.PIN_CODE,
"city" to "CITY? Reply only city or NONE.", DocumentPrompts.CITY,
"gender" to "GENDER? Reply only MALE/FEMALE/OTHER or NONE.", DocumentPrompts.GENDER,
"nationality" to "NATIONALITY? Reply only nationality or NONE." DocumentPrompts.NATIONALITY
) )
for ((key, question) in drivingQuestions) { for ((key, question) in drivingQuestions) {
results[key] = llamaClient.ask(imageUrl, question) results[key] = llamaClient.ask(imageUrl, question)
} }
} else if (isYes(results["hasElectionCommission"])) {
val voterQuestions = linkedMapOf(
"name" to "NAME? Reply only the name or NONE.",
"dob" to "DOB? Reply only date or NONE.",
"idNumber" to "VOTER ID NUMBER? Reply only number or NONE.",
"address" to "POSTAL ADDRESS ONLY (street/area/city/state). Ignore IDs, UUIDs, and codes. Reply only address or NONE.",
"pinCode" to "PIN CODE? Reply only pin or NONE.",
"city" to "CITY? Reply only city or NONE.",
"gender" to "GENDER? Reply only MALE/FEMALE/OTHER or NONE.",
"nationality" to "NATIONALITY? Reply only nationality or NONE."
)
for ((key, question) in voterQuestions) {
results[key] = llamaClient.ask(imageUrl, question)
}
} else if (isYes(results["hasIncomeTaxDept"])) {
val panQuestions = linkedMapOf(
"name" to "NAME? Reply only the name or NONE.",
"dob" to "DOB? Reply only date or NONE.",
"idNumber" to "PAN NUMBER? Reply only number or NONE.",
"address" to "POSTAL ADDRESS ONLY (street/area/city/state). Ignore IDs, UUIDs, and codes. Reply only address or NONE.",
"pinCode" to "PIN CODE? Reply only pin or NONE.",
"city" to "CITY? Reply only city or NONE.",
"gender" to "GENDER? Reply only MALE/FEMALE/OTHER or NONE.",
"nationality" to "NATIONALITY? Reply only nationality or NONE."
)
for ((key, question) in panQuestions) {
results[key] = llamaClient.ask(imageUrl, question)
}
} else if (isYes(results["hasPassport"])) {
val passportQuestions = linkedMapOf(
"name" to "NAME? Reply only the name or NONE.",
"dob" to "DOB? Reply only date or NONE.",
"idNumber" to "PASSPORT NUMBER? Reply only number or NONE.",
"address" to "POSTAL ADDRESS ONLY (street/area/city/state). Ignore IDs, UUIDs, and codes. Reply only address or NONE.",
"pinCode" to "PIN CODE? Reply only pin or NONE.",
"city" to "CITY? Reply only city or NONE.",
"gender" to "GENDER? Reply only MALE/FEMALE/OTHER or NONE.",
"nationality" to "NATIONALITY? Reply only nationality or NONE."
)
for ((key, question) in passportQuestions) {
results[key] = llamaClient.ask(imageUrl, question)
}
} else { } else {
val generalQuestions = linkedMapOf( results["hasElectionCommission"] = llamaClient.ask(
"name" to "NAME? Reply only the name or NONE.", imageUrl,
"dob" to "DOB? Reply only date or NONE.", "CONTAINS ELECTION COMMISSION OF INDIA? Answer YES or NO only."
"idNumber" to "ID NUMBER? Reply only number or NONE.",
"address" to "POSTAL ADDRESS ONLY (street/area/city/state). Ignore IDs, UUIDs, and codes. Reply only address or NONE.",
"vehicleNumber" to "VEHICLE NUMBER? Reply only number or NONE.",
"pinCode" to "PIN CODE? Reply only pin or NONE.",
"city" to "CITY? Reply only city or NONE.",
"gender" to "GENDER? Reply only MALE/FEMALE/OTHER or NONE.",
"nationality" to "NATIONALITY? Reply only nationality or NONE."
) )
for ((key, question) in generalQuestions) { if (isYes(results["hasElectionCommission"])) {
results[key] = llamaClient.ask(imageUrl, question) val voterQuestions = linkedMapOf(
DocumentPrompts.NAME,
DocumentPrompts.DOB,
"idNumber" to "VOTER ID NUMBER? Reply only number or NONE.",
DocumentPrompts.ADDRESS,
DocumentPrompts.PIN_CODE,
DocumentPrompts.CITY,
DocumentPrompts.GENDER,
DocumentPrompts.NATIONALITY
)
for ((key, question) in voterQuestions) {
results[key] = llamaClient.ask(imageUrl, question)
}
} else {
results["hasIncomeTaxDept"] = llamaClient.ask(
imageUrl,
"CONTAINS INCOME TAX DEPARTMENT? Answer YES or NO only."
)
if (isYes(results["hasIncomeTaxDept"])) {
val panQuestions = linkedMapOf(
DocumentPrompts.NAME,
DocumentPrompts.DOB,
"idNumber" to "PAN NUMBER? Reply only number or NONE.",
DocumentPrompts.ADDRESS,
DocumentPrompts.PIN_CODE,
DocumentPrompts.CITY,
DocumentPrompts.GENDER,
DocumentPrompts.NATIONALITY
)
for ((key, question) in panQuestions) {
results[key] = llamaClient.ask(imageUrl, question)
}
} else {
results["hasPassport"] = llamaClient.ask(
imageUrl,
"CONTAINS PASSPORT? Answer YES or NO only."
)
if (isYes(results["hasPassport"])) {
val passportQuestions = linkedMapOf(
DocumentPrompts.NAME,
DocumentPrompts.DOB,
"idNumber" to "PASSPORT NUMBER? Reply only number or NONE.",
DocumentPrompts.ADDRESS,
DocumentPrompts.PIN_CODE,
DocumentPrompts.CITY,
DocumentPrompts.GENDER,
DocumentPrompts.NATIONALITY
)
for ((key, question) in passportQuestions) {
results[key] = llamaClient.ask(imageUrl, question)
}
} else {
val generalQuestions = linkedMapOf(
DocumentPrompts.NAME,
DocumentPrompts.DOB,
DocumentPrompts.ID_NUMBER,
DocumentPrompts.ADDRESS,
DocumentPrompts.VEHICLE_NUMBER,
DocumentPrompts.PIN_CODE,
DocumentPrompts.CITY,
DocumentPrompts.GENDER,
DocumentPrompts.NATIONALITY
)
for ((key, question) in generalQuestions) {
results[key] = llamaClient.ask(imageUrl, question)
}
}
}
} }
} }
} }