Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import json
 import os
 from datetime import datetime, timedelta
+import subprocess
 from huggingface_hub import HfApi
 from pathlib import Path
 from calendar_rag import (
@@ -120,11 +121,9 @@ def load_qa_history():
         st.error(f"Error loading QA history: {str(e)}")
         return []
 
-
 def save_qa_history(history_entry):
     """Save QA history entry to local JSON file and push to Hugging Face"""
     try:
-        
         history_file = Path("qa_history.json")
         
         # Initialize or load existing history
@@ -270,7 +269,7 @@ def main():
                 key="query_input"
             )
 
-            # Button layout
+            # Button layout inside main() function
             col1, col2, col3 = st.columns([1, 1, 4])
         
             with col1:
@@ -288,13 +287,12 @@ def main():
                     use_container_width=True,
                     key="clear_history_button"
                 )
-
             # Process query
             if send_query and query:
                 if st.session_state.pipeline is None:
                     st.error("❌ ไม่สามารถเชื่อมต่อกับระบบได้ กรุณาลองใหม่อีกครั้ง")
                     return
-
+                    
                 add_to_history("user", query)
                     
                 try:
@@ -314,17 +312,19 @@ def main():
                         with st.expander("🔍 รายละเอียดการวิเคราะห์คำถาม", expanded=False):
                             st.json(result["query_info"])
                         
+                        st.rerun()
+                        
                 except Exception as e:
                     st.error(f"❌ เกิดข้อผิดพลาด: {str(e)}")
             
             elif send_query and not query:
                 st.warning("⚠️ กรุณาระบุคำถาม")
-            
-            # Clear history
-            if clear_history:
-                st.session_state.chat_history = []
-                save_qa_history([])  # Clear saved history
-                st.rerun()
+
+        # Handle clear history action
+        if clear_history:
+            st.session_state.chat_history = []
+            clear_qa_history()
+            st.rerun()
     
     with info_col:
         # System information
@@ -366,11 +366,5 @@ def main():
             "พร้อมใช้งาน" if st.session_state.pipeline else "ไม่พร้อมใช้งาน"
         ), unsafe_allow_html=True)
 
-        # Handle clear history action
-        if clear_history:
-            st.session_state.chat_history = []
-            clear_qa_history()
-            st.rerun()
-
 if __name__ == "__main__":
     main()