43a18ba070
* Add regen button and repharasing question on regen * Stop appending regen messages to history, allow only one * Add dynamic conversation state * Allow reasoning pipeline to manipulate state --------- Co-authored-by: albert <albert@cinnamon.is> Co-authored-by: Duc Nguyen (john) <trungduc1992@gmail.com>
78 lines
2.4 KiB
Python
78 lines
2.4 KiB
Python
from typing import Optional
|
|
|
|
import gradio as gr
|
|
from ktem.app import BasePage
|
|
from ktem.db.models import IssueReport, engine
|
|
from sqlmodel import Session
|
|
|
|
|
|
class ReportIssue(BasePage):
|
|
def __init__(self, app):
|
|
self._app = app
|
|
self.on_building_ui()
|
|
|
|
def on_building_ui(self):
|
|
with gr.Accordion(label="Report", open=False):
|
|
self.correctness = gr.Radio(
|
|
choices=[
|
|
("The answer is correct", "correct"),
|
|
("The answer is incorrect", "incorrect"),
|
|
],
|
|
label="Correctness:",
|
|
)
|
|
self.issues = gr.CheckboxGroup(
|
|
choices=[
|
|
("The answer is offensive", "offensive"),
|
|
("The evidence is incorrect", "wrong-evidence"),
|
|
],
|
|
label="Other issue:",
|
|
)
|
|
self.more_detail = gr.Textbox(
|
|
placeholder="More detail (e.g. how wrong is it, what is the "
|
|
"correct answer, etc...)",
|
|
container=False,
|
|
lines=3,
|
|
)
|
|
gr.Markdown(
|
|
"This will send the current chat and the user settings to "
|
|
"help with investigation"
|
|
)
|
|
self.report_btn = gr.Button("Report")
|
|
|
|
def report(
|
|
self,
|
|
correctness: str,
|
|
issues: list[str],
|
|
more_detail: str,
|
|
conv_id: str,
|
|
chat_history: list,
|
|
settings: dict,
|
|
user_id: Optional[int],
|
|
chat_state: dict,
|
|
*selecteds
|
|
):
|
|
selecteds_ = {}
|
|
for index in self._app.index_manager.indices:
|
|
if index.selector != -1:
|
|
selecteds_[str(index.id)] = selecteds[index.selector]
|
|
|
|
with Session(engine) as session:
|
|
issue = IssueReport(
|
|
issues={
|
|
"correctness": correctness,
|
|
"issues": issues,
|
|
"more_detail": more_detail,
|
|
},
|
|
chat={
|
|
"conv_id": conv_id,
|
|
"chat_history": chat_history,
|
|
"chat_state": chat_state,
|
|
"selecteds": selecteds_,
|
|
},
|
|
settings=settings,
|
|
user=user_id,
|
|
)
|
|
session.add(issue)
|
|
session.commit()
|
|
gr.Info("Thank you for your feedback")
|