Update app.py

app.py

@@ -1,4 +1,5 @@
-# app.py
+works app.py
+
 from __future__ import annotations
 import os
 import io
@@ -13,10 +14,8 @@ from datetime import datetime
 import regex as re2
 import re
 
-# --- BACKEND IMPORTS ---
 from langchain_cohere import ChatCohere
 
-# --- LOCAL MODULE IMPORTS ---
 from settings import (
     GENERAL_CONVERSATION_PROMPT,
     COHERE_MODEL_PRIMARY, COHERE_TIMEOUT_S, USE_OPEN_FALLBACKS
@@ -25,12 +24,10 @@ from audit_log import log_event
 from privacy import safety_filter, refusal_reply
 from llm_router import cohere_chat, _co_client, cohere_embed
 
-# --- UTILITY FUNCTIONS ---
-
 def load_markdown_text(filepath: str) -> str:
-    """Safely loads text content from a markdown file."""
     try:
-        with open(filepath, 'r', encoding='utf-8') as f: return f.read()
+        with open(filepath, 'r', encoding='utf-8') as f:
+            return f.read()
     except FileNotFoundError:
         return f"**Error:** Document `{os.path.basename(filepath)}` not found."
 
@@ -38,11 +35,7 @@ def _sanitize_text(s: str) -> str:
     if not isinstance(s, str): return s
     return re2.sub(r'[\p{C}--[\n\t]]+', '', s)
 
-# --- THE "ANALYST-WRITER" PIPELINE ---
-
 def _create_python_script(user_scenario: str, schema_context: str) -> str:
-    """Asks the AI to write a Python script that outputs raw, structured JSON."""
-    
     EXPERT_ANALYTICAL_GUIDELINES = """
 --- EXPERT ANALYTICAL GUIDELINES ---
 When writing your script, you MUST follow these expert business rules:
@@ -81,7 +74,6 @@ Now, write the complete Python script that performs the analysis and prints a si
     return "print(json.dumps({'error': 'Failed to generate a valid Python script.'}))"
 
 def _generate_long_report(prompt: str) -> str:
-    """Calls the Cohere API directly with a much higher max_tokens limit."""
     try:
         client = _co_client()
         if not client: return "Error: Cohere client not initialized."
@@ -96,7 +88,6 @@ def _generate_long_report(prompt: str) -> str:
         return f"Error during final report generation: {e}"
 
 def _generate_final_report(user_scenario: str, raw_data_json: str) -> str:
-    """Asks the AI to act as a consultant and write a polished report from the raw data."""
     prompt_for_writer = f"""
 You are an expert management consultant and data analyst.
 A data science script has run to extract key findings. You have the user's original request and the raw JSON data.
@@ -123,7 +114,6 @@ def _append_msg(h: List[Dict[str, str]], r: str, c: str) -> List[Dict[str, str]]
     return (h or []) + [{"role": r, "content": c}]
 
 def ping_cohere() -> str:
-    """Lightweight health check."""
     try:
         cli = _co_client()
         if not cli: return "Cohere client not initialized."
@@ -132,7 +122,6 @@ def ping_cohere() -> str:
     except Exception as e: return f"Cohere ping failed: {e}"
 
 def handle(user_msg: str, files: list, yield_update) -> str:
-    """Orchestrates the 'Analyst-Writer' pipeline."""
     try:
         safe_in, blocked_in, reason_in = safety_filter(user_msg, mode="input")
         if blocked_in: return refusal_reply(reason_in)
@@ -218,8 +207,6 @@ with gr.Blocks(theme="soft", css="style.css") as demo:
         privacy_link = gr.Button("Privacy Policy", variant="link")
         terms_link = gr.Button("Terms of Service", variant="link")
 
-    # --- DEFINITIVE FIX: ALL UI LOGIC IS NOW CORRECTLY INDENTED ---
-
     def run_analysis_wrapper(prompt, files, chat_history_list, history_state_list):
         if not prompt:
             gr.Warning("Please enter a prompt.")
@@ -251,7 +238,6 @@ with gr.Blocks(theme="soft", css="style.css") as demo:
     def view_history(selection, history_state_list):
         if not selection or not history_state_list:
             return ""
-        # FIX 1: Correctly extract the ID (the timestamp) from the selection string
         selected_id = selection.split(" - ")
         selected_assessment = next((item for item in history_state_list if item["id"] == selected_id), None)
         
@@ -279,7 +265,6 @@ with gr.Blocks(theme="soft", css="style.css") as demo:
         inputs=[history_dropdown, assessment_history],
         outputs=[history_display]
     )
-    # FIX 2: The 'assessment_history' state is no longer cleared.
     clear_btn.click(
         lambda: (None, None, []), 
         outputs=[prompt_input, files_input, chat_history_output]