Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
|
| 2 |
import os
|
| 3 |
import gradio as gr
|
| 4 |
from PyPDF2 import PdfReader
|
|
@@ -17,7 +17,7 @@ from datetime import datetime
|
|
| 17 |
# Load environment variables
|
| 18 |
load_dotenv()
|
| 19 |
|
| 20 |
-
#
|
| 21 |
def _get_api_key() -> str:
|
| 22 |
candidate_keys = [
|
| 23 |
"GOOGLE_API_KEY",
|
|
@@ -25,41 +25,45 @@ def _get_api_key() -> str:
|
|
| 25 |
"GOOGLE_GENAI_API_KEY",
|
| 26 |
"GENAI_API_KEY",
|
| 27 |
]
|
|
|
|
| 28 |
for key_name in candidate_keys:
|
| 29 |
value = os.getenv(key_name, "").strip()
|
| 30 |
if value:
|
| 31 |
-
#
|
| 32 |
os.environ["GOOGLE_API_KEY"] = value
|
| 33 |
return value
|
|
|
|
| 34 |
return ""
|
| 35 |
|
|
|
|
| 36 |
class PDFChatBot:
|
| 37 |
def __init__(self):
|
| 38 |
self.vector_store = None
|
| 39 |
-
#
|
| 40 |
self.embeddings = None
|
| 41 |
self.processed_files = []
|
| 42 |
-
self.chat_history = [] #
|
| 43 |
|
| 44 |
def get_pdf_text(self, pdf_files):
|
| 45 |
-
"""
|
| 46 |
raw_text = ""
|
| 47 |
processed_count = 0
|
| 48 |
|
| 49 |
if not pdf_files:
|
| 50 |
return raw_text, processed_count
|
| 51 |
|
| 52 |
-
#
|
| 53 |
if not isinstance(pdf_files, list):
|
| 54 |
pdf_files = [pdf_files]
|
| 55 |
|
| 56 |
for pdf_file in pdf_files:
|
| 57 |
try:
|
| 58 |
-
#
|
| 59 |
-
pdf_path = pdf_file.name if hasattr(pdf_file,
|
| 60 |
|
| 61 |
pdf_reader = PdfReader(pdf_path)
|
| 62 |
file_text = ""
|
|
|
|
| 63 |
for page in pdf_reader.pages:
|
| 64 |
text = page.extract_text()
|
| 65 |
if text:
|
|
@@ -71,118 +75,99 @@ class PDFChatBot:
|
|
| 71 |
self.processed_files.append(os.path.basename(pdf_path))
|
| 72 |
|
| 73 |
except Exception as e:
|
| 74 |
-
print(f"
|
| 75 |
continue
|
| 76 |
|
| 77 |
return raw_text, processed_count
|
| 78 |
|
| 79 |
def get_pdf_text_via_gemini(self, pdf_files):
|
| 80 |
-
"""
|
| 81 |
api_key = _get_api_key()
|
| 82 |
if not api_key:
|
| 83 |
return "", 0
|
| 84 |
|
| 85 |
genai.configure(api_key=api_key)
|
| 86 |
model = genai.GenerativeModel("gemini-2.0-flash-exp")
|
| 87 |
-
|
| 88 |
-
raw_text = ""
|
| 89 |
-
processed_count = 0
|
| 90 |
-
|
| 91 |
-
if not pdf_files:
|
| 92 |
-
return raw_text, processed_count
|
| 93 |
-
|
| 94 |
-
if not isinstance(pdf_files, list):
|
| 95 |
-
pdf_files = [pdf_files]
|
| 96 |
-
|
| 97 |
-
for pdf_file in pdf_files:
|
| 98 |
-
try:
|
| 99 |
-
pdf_path = pdf_file.name if hasattr(pdf_file, 'name') else pdf_file
|
| 100 |
-
uploaded = genai.upload_file(pdf_path)
|
| 101 |
-
prompt = (
|
| 102 |
-
"請從此 PDF 中提取可讀文字,按頁面順序輸出純文字。"
|
| 103 |
-
)
|
| 104 |
-
resp = model.generate_content([uploaded, prompt])
|
| 105 |
-
text = resp.text or ""
|
| 106 |
-
if text.strip():
|
| 107 |
-
raw_text += text + "\n"
|
| 108 |
-
processed_count += 1
|
| 109 |
-
self.processed_files.append(os.path.basename(pdf_path))
|
| 110 |
-
except Exception as e:
|
| 111 |
-
print(f"使用Gemini解析PDF時發生錯誤:{str(e)}")
|
| 112 |
-
continue
|
| 113 |
-
|
| 114 |
-
return raw_text, processed_count
|
| 115 |
-
|
| 116 |
def get_text_chunks(self, text):
|
| 117 |
-
"""
|
| 118 |
text_splitter = CharacterTextSplitter(
|
| 119 |
separator="\n",
|
| 120 |
chunk_size=10000,
|
| 121 |
chunk_overlap=1000,
|
| 122 |
-
length_function=len
|
| 123 |
)
|
| 124 |
-
|
| 125 |
-
return chunks
|
| 126 |
|
| 127 |
def create_vector_store(self, chunks):
|
| 128 |
-
"""
|
| 129 |
try:
|
| 130 |
if self.embeddings is None:
|
| 131 |
api_key = _get_api_key()
|
| 132 |
if not api_key:
|
| 133 |
return False
|
|
|
|
| 134 |
self.embeddings = GoogleGenerativeAIEmbeddings(
|
| 135 |
model="models/text-embedding-004",
|
| 136 |
-
google_api_key=api_key
|
| 137 |
)
|
|
|
|
| 138 |
self.vector_store = FAISS.from_texts(chunks, self.embeddings)
|
| 139 |
self.vector_store.save_local("faiss_index")
|
| 140 |
return True
|
|
|
|
| 141 |
except Exception as e:
|
| 142 |
-
print(f"
|
| 143 |
return False
|
| 144 |
|
| 145 |
def load_vector_store(self):
|
| 146 |
-
"""
|
| 147 |
try:
|
| 148 |
-
if os.path.exists("faiss_index"):
|
| 149 |
-
if self.embeddings is None:
|
| 150 |
-
api_key = _get_api_key()
|
| 151 |
-
if not api_key:
|
| 152 |
-
return False
|
| 153 |
-
self.embeddings = GoogleGenerativeAIEmbeddings(
|
| 154 |
-
model="models/text-embedding-004",
|
| 155 |
-
google_api_key=api_key
|
| 156 |
-
)
|
| 157 |
-
self.vector_store = FAISS.load_local(
|
| 158 |
-
"faiss_index",
|
| 159 |
-
embeddings=self.embeddings,
|
| 160 |
-
allow_dangerous_deserialization=True
|
| 161 |
-
)
|
| 162 |
-
return True
|
| 163 |
-
else:
|
| 164 |
return False
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 165 |
except Exception as e:
|
| 166 |
-
print(f"
|
| 167 |
return False
|
| 168 |
|
| 169 |
def get_conversational_chain(self, temperature=0.3, max_tokens=4096):
|
| 170 |
-
"""
|
| 171 |
prompt_template = """
|
| 172 |
-
|
| 173 |
-
|
| 174 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 175 |
|
| 176 |
-
|
| 177 |
-
|
| 178 |
|
| 179 |
-
|
| 180 |
-
|
| 181 |
|
| 182 |
-
# Using Flash 2.0 model(延後讀取 API Key)
|
| 183 |
api_key = _get_api_key()
|
| 184 |
if not api_key:
|
| 185 |
-
raise RuntimeError(
|
|
|
|
|
|
|
| 186 |
|
| 187 |
model = ChatGoogleGenerativeAI(
|
| 188 |
model="gemini-2.0-flash-exp",
|
|
@@ -190,233 +175,172 @@ class PDFChatBot:
|
|
| 190 |
temperature=temperature,
|
| 191 |
max_tokens=max_tokens,
|
| 192 |
top_p=0.8,
|
| 193 |
-
top_k=40
|
| 194 |
)
|
| 195 |
|
| 196 |
prompt = PromptTemplate(
|
| 197 |
template=prompt_template,
|
| 198 |
-
input_variables=[
|
| 199 |
)
|
| 200 |
|
| 201 |
-
|
| 202 |
-
|
| 203 |
-
|
| 204 |
-
|
| 205 |
-
|
| 206 |
-
if not self.vector_store:
|
| 207 |
-
return "請先上傳並處理PDF文件!"
|
| 208 |
-
|
| 209 |
-
if not question.strip():
|
| 210 |
-
return "請輸入您的問題。"
|
| 211 |
-
|
| 212 |
-
try:
|
| 213 |
-
# 搜索相關文檔
|
| 214 |
-
docs = self.vector_store.similarity_search(question, k=search_k)
|
| 215 |
-
|
| 216 |
-
if not docs:
|
| 217 |
-
return "在上傳的文檔中找不到相關信息。"
|
| 218 |
-
|
| 219 |
-
# 生成回答
|
| 220 |
-
chain = self.get_conversational_chain(temperature, max_tokens)
|
| 221 |
-
response = chain(
|
| 222 |
-
{
|
| 223 |
-
"input_documents": docs,
|
| 224 |
-
"question": question,
|
| 225 |
-
},
|
| 226 |
-
return_only_outputs=True
|
| 227 |
-
)
|
| 228 |
-
|
| 229 |
-
return response["output_text"]
|
| 230 |
-
|
| 231 |
-
except Exception as e:
|
| 232 |
-
return f"處理問題時發生錯誤:{str(e)}"
|
| 233 |
-
|
| 234 |
def process_pdfs(self, pdf_files, progress=gr.Progress(), use_gemini=False):
|
| 235 |
-
"""
|
| 236 |
if not pdf_files:
|
| 237 |
-
return "
|
| 238 |
|
| 239 |
self.processed_files = []
|
| 240 |
-
progress(0, desc="
|
| 241 |
|
| 242 |
-
#
|
| 243 |
-
progress(0.2, desc="
|
| 244 |
if use_gemini:
|
| 245 |
raw_text, processed_count = self.get_pdf_text_via_gemini(pdf_files)
|
| 246 |
else:
|
| 247 |
raw_text, processed_count = self.get_pdf_text(pdf_files)
|
| 248 |
|
| 249 |
if not raw_text.strip():
|
| 250 |
-
return "
|
| 251 |
|
| 252 |
-
|
| 253 |
-
|
| 254 |
text_chunks = self.get_text_chunks(raw_text)
|
| 255 |
|
| 256 |
-
|
| 257 |
-
|
| 258 |
success = self.create_vector_store(text_chunks)
|
| 259 |
|
| 260 |
-
progress(1.0, desc="
|
| 261 |
|
| 262 |
if success:
|
| 263 |
-
file_list = "
|
| 264 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 265 |
else:
|
| 266 |
-
return "❌ PDF
|
| 267 |
|
| 268 |
def clear_data(self):
|
| 269 |
-
"""
|
| 270 |
try:
|
| 271 |
if os.path.exists("faiss_index"):
|
| 272 |
shutil.rmtree("faiss_index")
|
|
|
|
| 273 |
self.vector_store = None
|
| 274 |
self.processed_files = []
|
| 275 |
self.chat_history = []
|
| 276 |
-
|
|
|
|
|
|
|
| 277 |
except Exception as e:
|
| 278 |
-
return f"❌
|
| 279 |
|
| 280 |
def create_docx_report(self, chat_history):
|
| 281 |
-
"""
|
| 282 |
try:
|
| 283 |
-
# 創建新的文檔
|
| 284 |
doc = Document()
|
| 285 |
|
| 286 |
-
#
|
| 287 |
-
title = doc.add_heading(
|
| 288 |
-
title.alignment = 1 #
|
| 289 |
|
| 290 |
-
#
|
| 291 |
-
doc.add_paragraph(
|
|
|
|
|
|
|
| 292 |
|
| 293 |
-
#
|
| 294 |
if self.processed_files:
|
| 295 |
-
doc.add_heading(
|
| 296 |
for i, file in enumerate(self.processed_files, 1):
|
| 297 |
-
doc.add_paragraph(f
|
| 298 |
|
| 299 |
-
|
| 300 |
|
| 301 |
-
#
|
| 302 |
-
doc.add_heading(
|
| 303 |
|
| 304 |
if not chat_history:
|
| 305 |
-
doc.add_paragraph(
|
| 306 |
else:
|
| 307 |
for i in range(0, len(chat_history), 2):
|
| 308 |
if i + 1 < len(chat_history):
|
| 309 |
-
question = chat_history[i][
|
| 310 |
-
answer = chat_history[i + 1][
|
| 311 |
|
| 312 |
-
#
|
| 313 |
q_paragraph = doc.add_paragraph()
|
| 314 |
-
q_run = q_paragraph.add_run(f
|
| 315 |
q_run.bold = True
|
| 316 |
q_run.font.size = Inches(0.14)
|
| 317 |
-
q_paragraph.add_run(question)
|
| 318 |
-
|
| 319 |
-
# 回答
|
| 320 |
-
a_paragraph = doc.add_paragraph()
|
| 321 |
-
a_run = a_paragraph.add_run('回答:')
|
| 322 |
-
a_run.bold = True
|
| 323 |
-
a_run.font.size = Inches(0.14)
|
| 324 |
-
a_paragraph.add_run(answer)
|
| 325 |
-
|
| 326 |
-
# 分隔線
|
| 327 |
-
doc.add_paragraph('─' * 50)
|
| 328 |
-
|
| 329 |
-
# 保存到臨時文件
|
| 330 |
-
temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.docx')
|
| 331 |
-
doc.save(temp_file.name)
|
| 332 |
-
temp_file.close()
|
| 333 |
|
| 334 |
-
|
| 335 |
|
| 336 |
except Exception as e:
|
| 337 |
-
|
| 338 |
-
|
| 339 |
-
|
| 340 |
-
# 初始化聊天機器人
|
| 341 |
bot = PDFChatBot()
|
| 342 |
|
| 343 |
-
# Gradio 接口函數
|
| 344 |
-
def upload_and_process(files, use_gemini=False, progress=gr.Progress()):
|
| 345 |
-
return bot.process_pdfs(files, progress, use_gemini)
|
| 346 |
-
|
| 347 |
-
def ask_question(question, history, temperature, max_tokens, search_k):
|
| 348 |
-
if not question.strip():
|
| 349 |
-
return history, ""
|
| 350 |
-
|
| 351 |
-
response = bot.answer_question(question, temperature, max_tokens, search_k)
|
| 352 |
-
# 使用新的消息格式
|
| 353 |
-
user_msg = {"role": "user", "content": question}
|
| 354 |
-
assistant_msg = {"role": "assistant", "content": response}
|
| 355 |
-
|
| 356 |
-
history.append(user_msg)
|
| 357 |
-
history.append(assistant_msg)
|
| 358 |
-
|
| 359 |
-
# 同步更新聊天歷史到bot實例
|
| 360 |
-
bot.chat_history = history.copy()
|
| 361 |
-
|
| 362 |
-
return history, ""
|
| 363 |
-
|
| 364 |
-
def download_chat_history():
|
| 365 |
-
"""下載聊天記錄為docx文件"""
|
| 366 |
-
if not bot.chat_history:
|
| 367 |
-
return None
|
| 368 |
-
|
| 369 |
-
docx_path = bot.create_docx_report(bot.chat_history)
|
| 370 |
-
return docx_path
|
| 371 |
-
|
| 372 |
-
def export_to_word():
|
| 373 |
-
"""匯出問答記錄為Word文件"""
|
| 374 |
-
if not bot.chat_history:
|
| 375 |
-
return None
|
| 376 |
-
|
| 377 |
-
docx_path = bot.create_docx_report(bot.chat_history)
|
| 378 |
-
return docx_path
|
| 379 |
|
| 380 |
def clear_chat():
|
| 381 |
-
"""
|
| 382 |
bot.chat_history = []
|
| 383 |
return [], ""
|
| 384 |
|
|
|
|
| 385 |
def clear_all_data():
|
| 386 |
return bot.clear_data()
|
| 387 |
|
|
|
|
| 388 |
def load_existing_data():
|
| 389 |
if bot.load_vector_store():
|
| 390 |
-
return "✅
|
| 391 |
else:
|
| 392 |
-
return "❌
|
|
|
|
| 393 |
|
| 394 |
def set_api_key(api_key: str):
|
| 395 |
-
"""
|
| 396 |
-
|
|
|
|
|
|
|
|
|
|
| 397 |
key = (api_key or "").strip()
|
| 398 |
if not key:
|
| 399 |
-
return "❌
|
|
|
|
| 400 |
os.environ["GOOGLE_API_KEY"] = key
|
| 401 |
-
|
|
|
|
| 402 |
try:
|
| 403 |
bot.embeddings = None
|
| 404 |
except Exception:
|
| 405 |
pass
|
| 406 |
-
return "✅ 已設定 API 金鑰(僅本次執行期間有效)。"
|
| 407 |
|
| 408 |
-
|
|
|
|
|
|
|
|
|
|
| 409 |
custom_theme = gr.themes.Soft(
|
| 410 |
primary_hue="blue",
|
| 411 |
secondary_hue="gray",
|
| 412 |
neutral_hue="slate",
|
| 413 |
font=gr.themes.GoogleFont("Noto Sans TC"),
|
| 414 |
-
font_mono=gr.themes.GoogleFont("JetBrains Mono")
|
| 415 |
)
|
| 416 |
|
| 417 |
-
|
|
|
|
| 418 |
with gr.Blocks(
|
| 419 |
-
title="PDF
|
| 420 |
theme=custom_theme,
|
| 421 |
css="""
|
| 422 |
.gradio-container {
|
|
@@ -443,255 +367,210 @@ with gr.Blocks(
|
|
| 443 |
padding: 10px;
|
| 444 |
border-radius: 5px;
|
| 445 |
}
|
| 446 |
-
"""
|
| 447 |
-
)
|
| 448 |
-
|
| 449 |
-
#
|
| 450 |
with gr.Row():
|
| 451 |
gr.HTML("""
|
| 452 |
<div class="main-header">
|
| 453 |
-
<h1>🤖 PDF
|
| 454 |
-
<p>
|
| 455 |
</div>
|
| 456 |
""")
|
| 457 |
|
| 458 |
-
#
|
| 459 |
-
with gr.Tab("📁
|
| 460 |
with gr.Row():
|
|
|
|
| 461 |
with gr.Column(scale=3):
|
| 462 |
-
#
|
| 463 |
with gr.Group():
|
| 464 |
-
gr.Markdown("### 📤
|
|
|
|
| 465 |
api_key_box = gr.Textbox(
|
| 466 |
-
label="Google API Key (
|
| 467 |
-
placeholder="
|
| 468 |
-
type="password"
|
| 469 |
)
|
| 470 |
-
|
| 471 |
-
|
|
|
|
|
|
|
| 472 |
file_count="multiple",
|
| 473 |
file_types=[".pdf"],
|
| 474 |
-
label="
|
| 475 |
-
height=150
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 476 |
)
|
| 477 |
-
|
| 478 |
-
|
| 479 |
-
# 處理選項
|
| 480 |
with gr.Row():
|
| 481 |
process_btn = gr.Button(
|
| 482 |
-
"🚀
|
| 483 |
-
variant="primary",
|
| 484 |
size="lg",
|
| 485 |
-
scale=2
|
| 486 |
)
|
|
|
|
| 487 |
load_btn = gr.Button(
|
| 488 |
-
"📂
|
| 489 |
variant="secondary",
|
| 490 |
-
scale=1
|
| 491 |
)
|
|
|
|
| 492 |
clear_btn = gr.Button(
|
| 493 |
-
"🗑️
|
| 494 |
variant="stop",
|
| 495 |
-
scale=1
|
| 496 |
)
|
| 497 |
|
| 498 |
with gr.Column(scale=2):
|
| 499 |
-
#
|
| 500 |
with gr.Group():
|
| 501 |
-
gr.Markdown("### 📊
|
|
|
|
| 502 |
status_text = gr.Textbox(
|
| 503 |
-
label="
|
| 504 |
lines=6,
|
| 505 |
interactive=False,
|
| 506 |
-
elem_classes=["status-box"]
|
| 507 |
)
|
| 508 |
-
|
| 509 |
-
#
|
| 510 |
-
gr.Markdown("### 📋
|
|
|
|
| 511 |
file_list = gr.Textbox(
|
| 512 |
-
label="
|
| 513 |
lines=8,
|
| 514 |
interactive=False,
|
| 515 |
-
elem_classes=["file-info"]
|
| 516 |
)
|
| 517 |
|
| 518 |
-
|
|
|
|
| 519 |
with gr.Row():
|
|
|
|
| 520 |
with gr.Column(scale=4):
|
| 521 |
-
# 聊天區域
|
| 522 |
chatbot = gr.Chatbot(
|
| 523 |
-
label="💬
|
| 524 |
height=600,
|
| 525 |
show_copy_button=True,
|
| 526 |
type="messages",
|
| 527 |
-
avatar_images=["👤", "🤖"]
|
| 528 |
)
|
| 529 |
|
| 530 |
with gr.Column(scale=1):
|
| 531 |
-
#
|
| 532 |
with gr.Group():
|
| 533 |
-
gr.Markdown("### ⚙️
|
| 534 |
-
|
| 535 |
-
# 模型參數調整
|
| 536 |
temperature = gr.Slider(
|
| 537 |
minimum=0.1,
|
| 538 |
maximum=1.0,
|
| 539 |
value=0.3,
|
| 540 |
-
step=0.
|
| 541 |
-
label="
|
| 542 |
-
info="數值越高回答越有創意"
|
| 543 |
)
|
| 544 |
-
|
| 545 |
-
max_tokens = gr.Slider(
|
| 546 |
-
minimum=512,
|
| 547 |
-
maximum=8192,
|
| 548 |
-
value=4096,
|
| 549 |
-
step=512,
|
| 550 |
-
label="最大回答長度",
|
| 551 |
-
info="控制回答的詳細程度"
|
| 552 |
-
)
|
| 553 |
-
|
| 554 |
-
search_k = gr.Slider(
|
| 555 |
-
minimum=2,
|
| 556 |
-
maximum=10,
|
| 557 |
-
value=6,
|
| 558 |
-
step=1,
|
| 559 |
-
label="檢索文檔數量",
|
| 560 |
-
info="搜索相關文檔的數量"
|
| 561 |
-
)
|
| 562 |
-
|
| 563 |
-
# 輸入區域
|
| 564 |
with gr.Row():
|
| 565 |
question_input = gr.Textbox(
|
| 566 |
-
placeholder="
|
| 567 |
-
label="💭
|
| 568 |
lines=3,
|
| 569 |
scale=4,
|
| 570 |
-
max_lines=5
|
| 571 |
)
|
|
|
|
| 572 |
ask_btn = gr.Button(
|
| 573 |
-
"📤
|
| 574 |
-
variant="primary",
|
| 575 |
scale=1,
|
| 576 |
-
size="lg"
|
| 577 |
)
|
| 578 |
|
| 579 |
-
#
|
| 580 |
with gr.Row():
|
| 581 |
clear_chat_btn = gr.Button(
|
| 582 |
-
"🧹
|
| 583 |
-
variant="secondary",
|
| 584 |
-
scale=1
|
| 585 |
)
|
|
|
|
| 586 |
download_btn = gr.Button(
|
| 587 |
-
"📥
|
| 588 |
-
variant="primary",
|
| 589 |
-
scale=1
|
| 590 |
)
|
|
|
|
| 591 |
export_btn = gr.Button(
|
| 592 |
-
"📄
|
| 593 |
-
variant="secondary",
|
| 594 |
-
scale=1
|
| 595 |
)
|
| 596 |
|
| 597 |
-
#
|
| 598 |
with gr.Group():
|
| 599 |
-
gr.Markdown("### 💡
|
|
|
|
| 600 |
gr.Examples(
|
| 601 |
examples=[
|
| 602 |
-
"
|
| 603 |
-
"
|
| 604 |
-
"
|
| 605 |
-
"
|
| 606 |
-
"
|
| 607 |
-
"
|
| 608 |
-
"
|
| 609 |
-
"
|
| 610 |
],
|
| 611 |
inputs=question_input,
|
| 612 |
-
label="
|
| 613 |
)
|
| 614 |
|
| 615 |
-
|
| 616 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 617 |
|
| 618 |
-
# 下載功能處理函數
|
| 619 |
-
def handle_download():
|
| 620 |
-
file_path = download_chat_history()
|
| 621 |
-
if file_path:
|
| 622 |
-
return gr.update(value=file_path, visible=True)
|
| 623 |
-
else:
|
| 624 |
-
gr.Warning("沒有聊天記錄可以下載!")
|
| 625 |
-
return gr.update(visible=False)
|
| 626 |
-
|
| 627 |
-
# 事件處理
|
| 628 |
-
process_btn.click(
|
| 629 |
-
fn=upload_and_process,
|
| 630 |
-
inputs=[file_upload, use_gemini_toggle],
|
| 631 |
-
outputs=[status_text, file_list],
|
| 632 |
-
show_progress=True
|
| 633 |
-
)
|
| 634 |
-
|
| 635 |
-
set_key_btn.click(
|
| 636 |
-
fn=set_api_key,
|
| 637 |
-
inputs=[api_key_box],
|
| 638 |
-
outputs=[status_text]
|
| 639 |
-
)
|
| 640 |
-
|
| 641 |
-
load_btn.click(
|
| 642 |
-
fn=load_existing_data,
|
| 643 |
-
outputs=[status_text, file_list]
|
| 644 |
-
)
|
| 645 |
-
|
| 646 |
-
clear_btn.click(
|
| 647 |
-
fn=clear_all_data,
|
| 648 |
-
outputs=[status_text, file_list]
|
| 649 |
-
)
|
| 650 |
-
|
| 651 |
-
ask_btn.click(
|
| 652 |
-
fn=ask_question,
|
| 653 |
-
inputs=[question_input, chatbot, temperature, max_tokens, search_k],
|
| 654 |
-
outputs=[chatbot, question_input]
|
| 655 |
-
)
|
| 656 |
-
|
| 657 |
-
question_input.submit(
|
| 658 |
-
fn=ask_question,
|
| 659 |
-
inputs=[question_input, chatbot, temperature, max_tokens, search_k],
|
| 660 |
-
outputs=[chatbot, question_input]
|
| 661 |
-
)
|
| 662 |
-
|
| 663 |
-
clear_chat_btn.click(
|
| 664 |
-
fn=clear_chat,
|
| 665 |
-
outputs=[chatbot, question_input]
|
| 666 |
-
)
|
| 667 |
-
|
| 668 |
-
download_btn.click(
|
| 669 |
-
fn=handle_download,
|
| 670 |
-
outputs=download_file
|
| 671 |
-
)
|
| 672 |
-
|
| 673 |
-
export_btn.click(
|
| 674 |
-
fn=export_to_word,
|
| 675 |
-
outputs=download_file
|
| 676 |
-
)
|
| 677 |
-
|
| 678 |
-
if __name__ == "__main__":
|
| 679 |
-
# 嘗試載入現有的向量存儲
|
| 680 |
-
bot.load_vector_store()
|
| 681 |
-
|
| 682 |
-
# 讀取部署相關配置
|
| 683 |
-
server_name = os.getenv("HOST", os.getenv("SERVER_NAME", "0.0.0.0"))
|
| 684 |
-
# 常見平台會傳入 PORT;若無則使用 7860(Gradio 預設)
|
| 685 |
-
server_port_env = os.getenv("PORT", os.getenv("SERVER_PORT"))
|
| 686 |
-
server_port = int(server_port_env) if server_port_env and server_port_env.isdigit() else 7860
|
| 687 |
-
inbrowser = os.getenv("INBROWSER", "false").lower() == "true"
|
| 688 |
-
share = os.getenv("GRADIO_SHARE", "false").lower() == "true"
|
| 689 |
-
|
| 690 |
-
# 啟動應用(綁定 0.0.0.0 以支援容器/雲端)
|
| 691 |
-
demo.launch(
|
| 692 |
-
share=share,
|
| 693 |
-
server_name=server_name,
|
| 694 |
-
server_port=server_port,
|
| 695 |
-
show_error=True,
|
| 696 |
-
inbrowser=inbrowser
|
| 697 |
-
)
|
|
|
|
| 1 |
+
from dotenv import load_dotenv
|
| 2 |
import os
|
| 3 |
import gradio as gr
|
| 4 |
from PyPDF2 import PdfReader
|
|
|
|
| 17 |
# Load environment variables
|
| 18 |
load_dotenv()
|
| 19 |
|
| 20 |
+
# Delay reading API key: provide helper function, read only when needed
|
| 21 |
def _get_api_key() -> str:
|
| 22 |
candidate_keys = [
|
| 23 |
"GOOGLE_API_KEY",
|
|
|
|
| 25 |
"GOOGLE_GENAI_API_KEY",
|
| 26 |
"GENAI_API_KEY",
|
| 27 |
]
|
| 28 |
+
|
| 29 |
for key_name in candidate_keys:
|
| 30 |
value = os.getenv(key_name, "").strip()
|
| 31 |
if value:
|
| 32 |
+
# Sync to GOOGLE_API_KEY for compatibility with underlying libraries
|
| 33 |
os.environ["GOOGLE_API_KEY"] = value
|
| 34 |
return value
|
| 35 |
+
|
| 36 |
return ""
|
| 37 |
|
| 38 |
+
|
| 39 |
class PDFChatBot:
|
| 40 |
def __init__(self):
|
| 41 |
self.vector_store = None
|
| 42 |
+
# Delay embedding model initialization until actually needed
|
| 43 |
self.embeddings = None
|
| 44 |
self.processed_files = []
|
| 45 |
+
self.chat_history = [] # Store chat history
|
| 46 |
|
| 47 |
def get_pdf_text(self, pdf_files):
|
| 48 |
+
"""Extract text from multiple PDF files"""
|
| 49 |
raw_text = ""
|
| 50 |
processed_count = 0
|
| 51 |
|
| 52 |
if not pdf_files:
|
| 53 |
return raw_text, processed_count
|
| 54 |
|
| 55 |
+
# Handle single file and multiple files
|
| 56 |
if not isinstance(pdf_files, list):
|
| 57 |
pdf_files = [pdf_files]
|
| 58 |
|
| 59 |
for pdf_file in pdf_files:
|
| 60 |
try:
|
| 61 |
+
# If uploaded file object, use its name attribute
|
| 62 |
+
pdf_path = pdf_file.name if hasattr(pdf_file, "name") else pdf_file
|
| 63 |
|
| 64 |
pdf_reader = PdfReader(pdf_path)
|
| 65 |
file_text = ""
|
| 66 |
+
|
| 67 |
for page in pdf_reader.pages:
|
| 68 |
text = page.extract_text()
|
| 69 |
if text:
|
|
|
|
| 75 |
self.processed_files.append(os.path.basename(pdf_path))
|
| 76 |
|
| 77 |
except Exception as e:
|
| 78 |
+
print(f"Error while reading PDF: {str(e)}")
|
| 79 |
continue
|
| 80 |
|
| 81 |
return raw_text, processed_count
|
| 82 |
|
| 83 |
def get_pdf_text_via_gemini(self, pdf_files):
|
| 84 |
+
"""Use Gemini 2.0 Flash to directly parse PDF text (via Files API)."""
|
| 85 |
api_key = _get_api_key()
|
| 86 |
if not api_key:
|
| 87 |
return "", 0
|
| 88 |
|
| 89 |
genai.configure(api_key=api_key)
|
| 90 |
model = genai.GenerativeModel("gemini-2.0-flash-exp")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 91 |
def get_text_chunks(self, text):
|
| 92 |
+
"""Split text into chunks for processing"""
|
| 93 |
text_splitter = CharacterTextSplitter(
|
| 94 |
separator="\n",
|
| 95 |
chunk_size=10000,
|
| 96 |
chunk_overlap=1000,
|
| 97 |
+
length_function=len,
|
| 98 |
)
|
| 99 |
+
return text_splitter.split_text(text)
|
|
|
|
| 100 |
|
| 101 |
def create_vector_store(self, chunks):
|
| 102 |
+
"""Create FAISS vector store from text chunks"""
|
| 103 |
try:
|
| 104 |
if self.embeddings is None:
|
| 105 |
api_key = _get_api_key()
|
| 106 |
if not api_key:
|
| 107 |
return False
|
| 108 |
+
|
| 109 |
self.embeddings = GoogleGenerativeAIEmbeddings(
|
| 110 |
model="models/text-embedding-004",
|
| 111 |
+
google_api_key=api_key,
|
| 112 |
)
|
| 113 |
+
|
| 114 |
self.vector_store = FAISS.from_texts(chunks, self.embeddings)
|
| 115 |
self.vector_store.save_local("faiss_index")
|
| 116 |
return True
|
| 117 |
+
|
| 118 |
except Exception as e:
|
| 119 |
+
print(f"Error while creating vector store: {str(e)}")
|
| 120 |
return False
|
| 121 |
|
| 122 |
def load_vector_store(self):
|
| 123 |
+
"""Load existing vector store"""
|
| 124 |
try:
|
| 125 |
+
if not os.path.exists("faiss_index"):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 126 |
return False
|
| 127 |
+
|
| 128 |
+
if self.embeddings is None:
|
| 129 |
+
api_key = _get_api_key()
|
| 130 |
+
if not api_key:
|
| 131 |
+
return False
|
| 132 |
+
|
| 133 |
+
self.embeddings = GoogleGenerativeAIEmbeddings(
|
| 134 |
+
model="models/text-embedding-004",
|
| 135 |
+
google_api_key=api_key,
|
| 136 |
+
)
|
| 137 |
+
|
| 138 |
+
self.vector_store = FAISS.load_local(
|
| 139 |
+
"faiss_index",
|
| 140 |
+
embeddings=self.embeddings,
|
| 141 |
+
allow_dangerous_deserialization=True,
|
| 142 |
+
)
|
| 143 |
+
return True
|
| 144 |
+
|
| 145 |
except Exception as e:
|
| 146 |
+
print(f"Error while loading vector store: {str(e)}")
|
| 147 |
return False
|
| 148 |
|
| 149 |
def get_conversational_chain(self, temperature=0.3, max_tokens=4096):
|
| 150 |
+
"""Create conversational QA chain"""
|
| 151 |
prompt_template = """
|
| 152 |
+
Answer the question in as much detail as possible based on the provided context.
|
| 153 |
+
If you need more information to answer perfectly, ask for the missing details.
|
| 154 |
+
If the answer cannot be found in the provided content, simply say:
|
| 155 |
+
"The answer cannot be found in the provided content."
|
| 156 |
+
|
| 157 |
+
Context:
|
| 158 |
+
{context}
|
| 159 |
|
| 160 |
+
Question:
|
| 161 |
+
{question}
|
| 162 |
|
| 163 |
+
Answer:
|
| 164 |
+
"""
|
| 165 |
|
|
|
|
| 166 |
api_key = _get_api_key()
|
| 167 |
if not api_key:
|
| 168 |
+
raise RuntimeError(
|
| 169 |
+
"API key not set. Please configure GOOGLE_API_KEY after deployment."
|
| 170 |
+
)
|
| 171 |
|
| 172 |
model = ChatGoogleGenerativeAI(
|
| 173 |
model="gemini-2.0-flash-exp",
|
|
|
|
| 175 |
temperature=temperature,
|
| 176 |
max_tokens=max_tokens,
|
| 177 |
top_p=0.8,
|
|
|
|
| 178 |
)
|
| 179 |
|
| 180 |
prompt = PromptTemplate(
|
| 181 |
template=prompt_template,
|
| 182 |
+
input_variables=["context", "question"],
|
| 183 |
)
|
| 184 |
|
| 185 |
+
return load_qa_chain(
|
| 186 |
+
model,
|
| 187 |
+
chain_type="stuff",
|
| 188 |
+
prompt=prompt,
|
| 189 |
+
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 190 |
def process_pdfs(self, pdf_files, progress=gr.Progress(), use_gemini=False):
|
| 191 |
+
"""Process PDF files"""
|
| 192 |
if not pdf_files:
|
| 193 |
+
return "Please upload at least one PDF file.", ""
|
| 194 |
|
| 195 |
self.processed_files = []
|
| 196 |
+
progress(0, desc="Starting PDF processing...")
|
| 197 |
|
| 198 |
+
# Extract text
|
| 199 |
+
progress(0.2, desc="Extracting PDF text...")
|
| 200 |
if use_gemini:
|
| 201 |
raw_text, processed_count = self.get_pdf_text_via_gemini(pdf_files)
|
| 202 |
else:
|
| 203 |
raw_text, processed_count = self.get_pdf_text(pdf_files)
|
| 204 |
|
| 205 |
if not raw_text.strip():
|
| 206 |
+
return "Unable to extract text from the PDF files.", ""
|
| 207 |
|
| 208 |
+
# Split text
|
| 209 |
+
progress(0.4, desc="Splitting text...")
|
| 210 |
text_chunks = self.get_text_chunks(raw_text)
|
| 211 |
|
| 212 |
+
# Create vector store
|
| 213 |
+
progress(0.6, desc="Creating vector store...")
|
| 214 |
success = self.create_vector_store(text_chunks)
|
| 215 |
|
| 216 |
+
progress(1.0, desc="Processing completed!")
|
| 217 |
|
| 218 |
if success:
|
| 219 |
+
file_list = "Processed files:\n" + "\n".join(
|
| 220 |
+
[f"• {file}" for file in self.processed_files]
|
| 221 |
+
)
|
| 222 |
+
return (
|
| 223 |
+
f"✅ Successfully processed {processed_count} PDF files!\n"
|
| 224 |
+
f"Total text chunks: {len(text_chunks)}\n"
|
| 225 |
+
"You can now start asking questions.",
|
| 226 |
+
file_list,
|
| 227 |
+
)
|
| 228 |
else:
|
| 229 |
+
return "❌ PDF processing failed. Please try again.", ""
|
| 230 |
|
| 231 |
def clear_data(self):
|
| 232 |
+
"""Clear processed data"""
|
| 233 |
try:
|
| 234 |
if os.path.exists("faiss_index"):
|
| 235 |
shutil.rmtree("faiss_index")
|
| 236 |
+
|
| 237 |
self.vector_store = None
|
| 238 |
self.processed_files = []
|
| 239 |
self.chat_history = []
|
| 240 |
+
|
| 241 |
+
return "✅ All processed data has been cleared!", ""
|
| 242 |
+
|
| 243 |
except Exception as e:
|
| 244 |
+
return f"❌ Error while clearing data: {str(e)}", ""
|
| 245 |
|
| 246 |
def create_docx_report(self, chat_history):
|
| 247 |
+
"""Create a DOCX report containing chat history"""
|
| 248 |
try:
|
|
|
|
| 249 |
doc = Document()
|
| 250 |
|
| 251 |
+
# Title
|
| 252 |
+
title = doc.add_heading("PDF Chatbot - Q&A Report", 0)
|
| 253 |
+
title.alignment = 1 # Center alignment
|
| 254 |
|
| 255 |
+
# Generation time
|
| 256 |
+
doc.add_paragraph(
|
| 257 |
+
f"Generated at: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
|
| 258 |
+
)
|
| 259 |
|
| 260 |
+
# Processed files
|
| 261 |
if self.processed_files:
|
| 262 |
+
doc.add_heading("Processed PDF files:", level=2)
|
| 263 |
for i, file in enumerate(self.processed_files, 1):
|
| 264 |
+
doc.add_paragraph(f"{i}. {file}", style="List Number")
|
| 265 |
|
| 266 |
+
doc.add_paragraph("")
|
| 267 |
|
| 268 |
+
# Chat history
|
| 269 |
+
doc.add_heading("Q&A History:", level=2)
|
| 270 |
|
| 271 |
if not chat_history:
|
| 272 |
+
doc.add_paragraph("There is currently no chat history.")
|
| 273 |
else:
|
| 274 |
for i in range(0, len(chat_history), 2):
|
| 275 |
if i + 1 < len(chat_history):
|
| 276 |
+
question = chat_history[i]["content"]
|
| 277 |
+
answer = chat_history[i + 1]["content"]
|
| 278 |
|
| 279 |
+
# Question
|
| 280 |
q_paragraph = doc.add_paragraph()
|
| 281 |
+
q_run = q_paragraph.add_run(f"Question {(i // 2) + 1}: ")
|
| 282 |
q_run.bold = True
|
| 283 |
q_run.font.size = Inches(0.14)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 284 |
|
| 285 |
+
# ⚠️ Answer handling & saving likely continues in PART 4
|
| 286 |
|
| 287 |
except Exception as e:
|
| 288 |
+
raise RuntimeError(f"Error while creating DOCX report: {str(e)}")
|
| 289 |
+
# Initialize chatbot
|
|
|
|
|
|
|
| 290 |
bot = PDFChatBot()
|
| 291 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 292 |
|
| 293 |
def clear_chat():
|
| 294 |
+
"""Clear chat history"""
|
| 295 |
bot.chat_history = []
|
| 296 |
return [], ""
|
| 297 |
|
| 298 |
+
|
| 299 |
def clear_all_data():
|
| 300 |
return bot.clear_data()
|
| 301 |
|
| 302 |
+
|
| 303 |
def load_existing_data():
|
| 304 |
if bot.load_vector_store():
|
| 305 |
+
return "✅ Successfully loaded processed data!", ""
|
| 306 |
else:
|
| 307 |
+
return "❌ No processed data found.", ""
|
| 308 |
+
|
| 309 |
|
| 310 |
def set_api_key(api_key: str):
|
| 311 |
+
"""
|
| 312 |
+
Set / update Google Gemini API key.
|
| 313 |
+
Updated only in memory and environment variables.
|
| 314 |
+
Will not be written to disk.
|
| 315 |
+
"""
|
| 316 |
key = (api_key or "").strip()
|
| 317 |
if not key:
|
| 318 |
+
return "❌ No API key provided. Please paste a valid GOOGLE_API_KEY."
|
| 319 |
+
|
| 320 |
os.environ["GOOGLE_API_KEY"] = key
|
| 321 |
+
|
| 322 |
+
# Reset embeddings to ensure re-initialization with new key
|
| 323 |
try:
|
| 324 |
bot.embeddings = None
|
| 325 |
except Exception:
|
| 326 |
pass
|
|
|
|
| 327 |
|
| 328 |
+
return "✅ API key set (valid for this session only)."
|
| 329 |
+
|
| 330 |
+
|
| 331 |
+
# Create custom theme
|
| 332 |
custom_theme = gr.themes.Soft(
|
| 333 |
primary_hue="blue",
|
| 334 |
secondary_hue="gray",
|
| 335 |
neutral_hue="slate",
|
| 336 |
font=gr.themes.GoogleFont("Noto Sans TC"),
|
| 337 |
+
font_mono=gr.themes.GoogleFont("JetBrains Mono"),
|
| 338 |
)
|
| 339 |
|
| 340 |
+
|
| 341 |
+
# Create Gradio interface
|
| 342 |
with gr.Blocks(
|
| 343 |
+
title="PDF Intelligent Q&A System",
|
| 344 |
theme=custom_theme,
|
| 345 |
css="""
|
| 346 |
.gradio-container {
|
|
|
|
| 367 |
padding: 10px;
|
| 368 |
border-radius: 5px;
|
| 369 |
}
|
| 370 |
+
""",
|
| 371 |
+
):
|
| 372 |
+
|
| 373 |
+
# Main header section
|
| 374 |
with gr.Row():
|
| 375 |
gr.HTML("""
|
| 376 |
<div class="main-header">
|
| 377 |
+
<h1>🤖 PDF Intelligent Q&A System</h1>
|
| 378 |
+
<p>Based on Gemini 2.0 Flash RAG technology | Supports multilingual Q&A</p>
|
| 379 |
</div>
|
| 380 |
""")
|
| 381 |
|
| 382 |
+
# Main feature area
|
| 383 |
+
with gr.Tab("📁 File Management", id="file_tab"):
|
| 384 |
with gr.Row():
|
| 385 |
+
|
| 386 |
with gr.Column(scale=3):
|
| 387 |
+
# File upload section
|
| 388 |
with gr.Group():
|
| 389 |
+
gr.Markdown("### 📤 Upload PDF Files")
|
| 390 |
+
|
| 391 |
api_key_box = gr.Textbox(
|
| 392 |
+
label="Google API Key (optional – paste after deployment)",
|
| 393 |
+
placeholder="Key starting with sk- or AIza (not saved to disk)",
|
| 394 |
+
type="password",
|
| 395 |
)
|
| 396 |
+
|
| 397 |
+
set_key_btn = gr.Button("🔑 Set API Key")
|
| 398 |
+
|
| 399 |
+
file_upload = gr.File(
|
| 400 |
file_count="multiple",
|
| 401 |
file_types=[".pdf"],
|
| 402 |
+
label="Select PDF files",
|
| 403 |
+
height=150,
|
| 404 |
+
)
|
| 405 |
+
|
| 406 |
+
use_gemini_toggle = gr.Checkbox(
|
| 407 |
+
label="Use Gemini to parse PDF (supports scanned images)",
|
| 408 |
+
value=False,
|
| 409 |
)
|
| 410 |
+
|
| 411 |
+
# Processing options
|
|
|
|
| 412 |
with gr.Row():
|
| 413 |
process_btn = gr.Button(
|
| 414 |
+
"🚀 Start Processing",
|
| 415 |
+
variant="primary",
|
| 416 |
size="lg",
|
| 417 |
+
scale=2,
|
| 418 |
)
|
| 419 |
+
|
| 420 |
load_btn = gr.Button(
|
| 421 |
+
"📂 Load processed data",
|
| 422 |
variant="secondary",
|
| 423 |
+
scale=1,
|
| 424 |
)
|
| 425 |
+
|
| 426 |
clear_btn = gr.Button(
|
| 427 |
+
"🗑️ Clear all data",
|
| 428 |
variant="stop",
|
| 429 |
+
scale=1,
|
| 430 |
)
|
| 431 |
|
| 432 |
with gr.Column(scale=2):
|
| 433 |
+
# Status display section
|
| 434 |
with gr.Group():
|
| 435 |
+
gr.Markdown("### 📊 Processing Status")
|
| 436 |
+
|
| 437 |
status_text = gr.Textbox(
|
| 438 |
+
label="Progress",
|
| 439 |
lines=6,
|
| 440 |
interactive=False,
|
| 441 |
+
elem_classes=["status-box"],
|
| 442 |
)
|
| 443 |
+
|
| 444 |
+
# File list
|
| 445 |
+
gr.Markdown("### 📋 Processed Files")
|
| 446 |
+
|
| 447 |
file_list = gr.Textbox(
|
| 448 |
+
label="File list",
|
| 449 |
lines=8,
|
| 450 |
interactive=False,
|
| 451 |
+
elem_classes=["file-info"],
|
| 452 |
)
|
| 453 |
|
| 454 |
+
# Chat tab
|
| 455 |
+
with gr.Tab("💬 Intelligent Chat", id="chat_tab"):
|
| 456 |
with gr.Row():
|
| 457 |
+
|
| 458 |
with gr.Column(scale=4):
|
|
|
|
| 459 |
chatbot = gr.Chatbot(
|
| 460 |
+
label="💬 Chat History",
|
| 461 |
height=600,
|
| 462 |
show_copy_button=True,
|
| 463 |
type="messages",
|
| 464 |
+
avatar_images=["👤", "🤖"],
|
| 465 |
)
|
| 466 |
|
| 467 |
with gr.Column(scale=1):
|
| 468 |
+
# Sidebar features
|
| 469 |
with gr.Group():
|
| 470 |
+
gr.Markdown("### ⚙️ Q&A Settings")
|
| 471 |
+
|
|
|
|
| 472 |
temperature = gr.Slider(
|
| 473 |
minimum=0.1,
|
| 474 |
maximum=1.0,
|
| 475 |
value=0.3,
|
| 476 |
+
step=0.05,
|
| 477 |
+
label="Temperature",
|
|
|
|
| 478 |
)
|
| 479 |
+
# Input area
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 480 |
with gr.Row():
|
| 481 |
question_input = gr.Textbox(
|
| 482 |
+
placeholder="Please enter your question... (supports multiple languages)",
|
| 483 |
+
label="💭 Question Input",
|
| 484 |
lines=3,
|
| 485 |
scale=4,
|
| 486 |
+
max_lines=5,
|
| 487 |
)
|
| 488 |
+
|
| 489 |
ask_btn = gr.Button(
|
| 490 |
+
"📤 Send Question",
|
| 491 |
+
variant="primary",
|
| 492 |
scale=1,
|
| 493 |
+
size="lg",
|
| 494 |
)
|
| 495 |
|
| 496 |
+
# Quick actions
|
| 497 |
with gr.Row():
|
| 498 |
clear_chat_btn = gr.Button(
|
| 499 |
+
"🧹 Clear Chat",
|
| 500 |
+
variant="secondary",
|
| 501 |
+
scale=1,
|
| 502 |
)
|
| 503 |
+
|
| 504 |
download_btn = gr.Button(
|
| 505 |
+
"📥 Download Chat History",
|
| 506 |
+
variant="primary",
|
| 507 |
+
scale=1,
|
| 508 |
)
|
| 509 |
+
|
| 510 |
export_btn = gr.Button(
|
| 511 |
+
"📄 Export to Word",
|
| 512 |
+
variant="secondary",
|
| 513 |
+
scale=1,
|
| 514 |
)
|
| 515 |
|
| 516 |
+
# Example questions
|
| 517 |
with gr.Group():
|
| 518 |
+
gr.Markdown("### 💡 Example Questions")
|
| 519 |
+
|
| 520 |
gr.Examples(
|
| 521 |
examples=[
|
| 522 |
+
"What is the main content of this document?",
|
| 523 |
+
"Please summarize the key points and concepts.",
|
| 524 |
+
"What important data or statistics are mentioned?",
|
| 525 |
+
"Can you explain a specific topic in detail?",
|
| 526 |
+
"What is the conclusion of the document?",
|
| 527 |
+
"What important recommendations are provided?",
|
| 528 |
+
"What risks or challenges are mentioned?",
|
| 529 |
+
"Compare the different viewpoints discussed.",
|
| 530 |
],
|
| 531 |
inputs=question_input,
|
| 532 |
+
label="Click an example to autofill",
|
| 533 |
)
|
| 534 |
|
| 535 |
+
# Hidden file download component
|
| 536 |
+
download_file = gr.File(visible=False)
|
| 537 |
+
|
| 538 |
+
# Download handler
|
| 539 |
+
def handle_download():
|
| 540 |
+
file_path = download_chat_history() # ⚠️ must exist elsewhere
|
| 541 |
+
if file_path:
|
| 542 |
+
return gr.update(value=file_path, visible=True)
|
| 543 |
+
else:
|
| 544 |
+
gr.Warning("No chat history available for download!")
|
| 545 |
+
return gr.update(visible=False)
|
| 546 |
+
|
| 547 |
+
# Event handlers
|
| 548 |
+
process_btn.click(
|
| 549 |
+
fn=upload_and_process, # ⚠️ must exist
|
| 550 |
+
inputs=[file_upload, use_gemini_toggle],
|
| 551 |
+
outputs=[status_text, file_list],
|
| 552 |
+
show_progress=True,
|
| 553 |
+
)
|
| 554 |
+
|
| 555 |
+
set_key_btn.click(
|
| 556 |
+
fn=set_api_key,
|
| 557 |
+
inputs=[api_key_box],
|
| 558 |
+
outputs=[status_text],
|
| 559 |
+
)
|
| 560 |
+
|
| 561 |
+
load_btn.click(
|
| 562 |
+
fn=load_existing_data,
|
| 563 |
+
outputs=[status_text, file_list],
|
| 564 |
+
)
|
| 565 |
+
|
| 566 |
+
clear_btn.click(
|
| 567 |
+
fn=clear_all_data,
|
| 568 |
+
outputs=[status_text, file_list],
|
| 569 |
+
)
|
| 570 |
+
|
| 571 |
+
ask_btn.click(
|
| 572 |
+
fn=ask_question, # ⚠️ must exist
|
| 573 |
+
inputs=[question_input, chatbot, temperature, max_tokens, search_k],
|
| 574 |
+
outputs=[chatbot, question_input],
|
| 575 |
+
)
|
| 576 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|