JirasakJo commited on
Commit
6a6cbe8
·
verified ·
1 Parent(s): 8b2112d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -6
app.py CHANGED
@@ -230,23 +230,30 @@ def submit():
230
  st.error("❌ ไม่สามารถเชื่อมต่อกับระบบได้ กรุณาลองใหม่อีกครั้ง")
231
  return
232
 
 
 
 
 
 
 
233
  user_query = st.session_state.query_input
234
  st.session_state.chat_history.append(("user", user_query))
235
 
236
- # Display loading message
237
  st.session_state.chat_history.append(("assistant", {
238
  "answer": "🔍 กำลังค้นหาคำตอบ...",
239
  "documents": [],
240
  "query_info": {}
241
  }))
242
-
243
- st.session_state.query_input = "" # Clear input after submission
244
 
245
  try:
246
- # Process the query
247
  result = st.session_state.pipeline.process_query(user_query)
 
248
 
249
- # Replace loading message with actual answer
250
  st.session_state.chat_history[-1] = ("assistant", {
251
  "answer": result["answer"],
252
  "documents": result["documents"],
@@ -257,6 +264,7 @@ def submit():
257
  add_to_qa_history(user_query, result["answer"])
258
 
259
  except Exception as e:
 
260
  st.session_state.chat_history[-1] = ("assistant", {
261
  "answer": f"❌ เกิดข้อผิดพลาด: {str(e)}",
262
  "documents": [],
@@ -264,7 +272,10 @@ def submit():
264
  })
265
  st.error(f"Query processing error: {e}")
266
 
267
- st.rerun()
 
 
 
268
 
269
  def main():
270
  # Page config
 
230
  st.error("❌ ไม่สามารถเชื่อมต่อกับระบบได้ กรุณาลองใหม่อีกครั้ง")
231
  return
232
 
233
+ # Prevent duplicate submissions
234
+ if "processing_query" in st.session_state and st.session_state.processing_query:
235
+ return
236
+
237
+ st.session_state.processing_query = True
238
+
239
  user_query = st.session_state.query_input
240
  st.session_state.chat_history.append(("user", user_query))
241
 
242
+ # Show loading message
243
  st.session_state.chat_history.append(("assistant", {
244
  "answer": "🔍 กำลังค้นหาคำตอบ...",
245
  "documents": [],
246
  "query_info": {}
247
  }))
248
+
249
+ st.session_state.query_input = "" # Clear input
250
 
251
  try:
252
+ print("Processing query:", user_query) # Debugging
253
  result = st.session_state.pipeline.process_query(user_query)
254
+ print("Query processed successfully") # Debugging
255
 
256
+ # Update the answer
257
  st.session_state.chat_history[-1] = ("assistant", {
258
  "answer": result["answer"],
259
  "documents": result["documents"],
 
264
  add_to_qa_history(user_query, result["answer"])
265
 
266
  except Exception as e:
267
+ print("Error processing query:", str(e)) # Debugging
268
  st.session_state.chat_history[-1] = ("assistant", {
269
  "answer": f"❌ เกิดข้อผิดพลาด: {str(e)}",
270
  "documents": [],
 
272
  })
273
  st.error(f"Query processing error: {e}")
274
 
275
+ finally:
276
+ st.session_state.processing_query = False # Reset flag
277
+
278
+ st.rerun() # Refresh UI
279
 
280
  def main():
281
  # Page config