Skip to content

Commit e8bde0b

Browse files
feat: add src_model name info to xlsx
1 parent a045197 commit e8bde0b

File tree

2 files changed

+10
-4
lines changed

2 files changed

+10
-4
lines changed

src/components/TextBox/index.tsx

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ export default function TextBox( {modelName: model_name, providerKey, index}: IT
4343
untranslated: global_text().untranslated,
4444
translated: "Waiting for text...",
4545
editing: false,
46+
model_name
4647
})
4748
const [ dragging, setDragging ] = createSignal(false)
4849
const handler = getHandler(providerKey, model_name)

webview/pywebview/xlsx.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,13 @@ class TextDTO:
1616
def __init__(self, DTO: Dict[str, str]):
1717
self.window_title = DTO["window_title"]
1818
self.originalText = DTO["originalText"]
19+
self.src_model = DTO['src_model'] #if 'src_model' in DTO else None
1920
self.translatedText = DTO["translatedText"] if "translatedText" in DTO else None
20-
self.speakerName = DTO['speakerName'] if 'speakerName' in DTO else None
21+
self._speakerName = DTO['speakerName'] if 'speakerName' in DTO else None
22+
23+
@property
24+
def speakerName(self):
25+
return self._speakerName if self._speakerName is not None else ""
2126

2227

2328
def loadFile(name: str):
@@ -51,7 +56,7 @@ def get_history(lastRowNumber: int):
5156
if rowNumber > 0:
5257
cell = worksheet().cell(row= rowNumber, column=2)
5358
if cell.value is not None and cell.value != "":
54-
speaker_name = f"[{cell.comment.text}]: " if cell.comment is not None else ""
59+
speaker_name = f"[{cell.comment.text}]: " if cell.comment is not None and len(cell.comment.text) else ""
5560
history.append(speaker_name + cell.value)
5661
else: break
5762

@@ -79,15 +84,15 @@ def SaveText(self, textDTO: Dict[str, str]):
7984
if entry is not None and worksheet() is not None:
8085
cell = worksheet().cell(row= entry, column=2)
8186
cell.value = textDTO.translatedText
82-
if textDTO.speakerName is not None: cell.comment = Comment(textDTO.speakerName, textDTO.speakerName)
87+
cell.comment = Comment(textDTO.speakerName, textDTO.src_model)
8388

8489
elif worksheet() is not None:
8590
lastRow = worksheet().max_row+1 # same variable for both calls to avoid the "stairs" bug
8691
cellA = worksheet().cell(row= lastRow, column=1)
8792
cellB = worksheet().cell(row= lastRow, column=2)
8893
cellA.value = textDTO.originalText
8994
cellB.value = textDTO.translatedText
90-
if textDTO.speakerName is not None: cellB.comment = Comment(textDTO.speakerName, textDTO.speakerName)
95+
cellB.comment = Comment(textDTO.speakerName, textDTO.src_model)
9196

9297
else: return
9398

0 commit comments

Comments
 (0)