Update app.py

app.py

@@ -4,8 +4,13 @@ from transformers import pipeline
 from sentence_transformers import SentenceTransformer, util
 import PyPDF2
 
-# Set up logging
-logging.basicConfig(filename='support_bot_log.txt', level=logging.INFO)
+# Set up logging with immediate flushing
+logger = logging.getLogger()
+logger.setLevel(logging.INFO)
+handler = logging.FileHandler('support_bot_log.txt', mode='a', buffering=1)  # Line-buffered for immediate writes
+formatter = logging.Formatter('%(asctime)s - %(message)s')
+handler.setFormatter(formatter)
+logger.addHandler(handler)
 
 # Load models
 qa_model = pipeline("question-answering", model="distilbert-base-uncased-distilled-squad")
@@ -50,18 +55,22 @@ def find_relevant_section(query, sections, section_embeddings):
     logging.info(f"No good keyword match found. Returning default fallback response.")
     return "I don’t have enough information to answer that."
 
-# Process the uploaded file
+# Process the uploaded file with detailed logging
 def process_file(file, state):
     if file is None:
+        logging.info("No file uploaded.")
         return [("Bot", "Please upload a file.")], state
     
     file_path = file.name
     if file_path.lower().endswith(".pdf"):
+        logging.info(f"Uploaded PDF file: {file_path}")
         text = extract_text_from_pdf(file_path)
     elif file_path.lower().endswith(".txt"):
+        logging.info(f"Uploaded TXT file: {file_path}")
         with open(file_path, 'r', encoding='utf-8') as f:
             text = f.read()
     else:
+        logging.error(f"Unsupported file format: {file_path}")
         return [("Bot", "Unsupported file format. Please upload a PDF or TXT file.")], state
     
     sections = text.split('\n\n')
@@ -80,6 +89,7 @@ def process_file(file, state):
 def handle_input(user_input, state):
     if state['mode'] == 'waiting_for_upload':
         state['chat_history'].append(("Bot", "Please upload a file first."))
+        logging.info("User attempted to interact without uploading a file.")
     elif state['mode'] == 'waiting_for_query':
         query = user_input
         state['current_query'] = query
@@ -140,6 +150,7 @@ with gr.Blocks() as demo:
     chat = gr.Chatbot()
     user_input = gr.Textbox(label="Your query or feedback")
     submit_btn = gr.Button("Submit")
+    log_file = gr.File(label="Download Log File", value="support_bot_log.txt")  # Added for log download
 
     # Process file upload
     file_upload.upload(process_file, inputs=[file_upload, state], outputs=[chat, state])