Update app.py

app.py

@@ -394,8 +394,8 @@ def get_short_document_name(filename: str) -> str:
     short_names = {
         '20241211_Econ_VRProposals_Budget2025_OnePagers': 'Budget 2025 One-Pagers',
         '20250813_Budget2026_Proposal_ExpandingIndustrialLand_En_F': 'Industrial Land Expansion (EN)',
-        '20250813_Budget2026Proposal_MaternityLeaveBenefit_Raj_D01': 'Maternity Leave Benefits ',
-        '20250813_Budget2026Proposal_RemovalOfTaxationOnEPF_Raj_F': 'EPF Tax Removal',
+        '20250813_Budget2026Proposal_MaternityLeaveBenefit_Raj_D01': 'Maternity Leave Benefits (Raj)',
+        '20250813_Budget2026Proposal_RemovalOfTaxationOnEPF_Raj_F': 'EPF Tax Removal (Raj)',
         '20250825_Budget2026Proposal_MaternityLeaveBenefit_Sin_F': 'Maternity Leave Benefits (Sinhala)',
         '20250825_Budget2026Proposal_MaternityLeaveBenefit_Tam_F': 'Maternity Leave Benefits (Tamil)',
         '20250825_Budget2026Proposal_RemovalOfTaxationOnEPF_Sin_Final': 'EPF Tax Removal (Sinhala)',
@@ -859,16 +859,20 @@ def get_available_pdfs_endpoint():
         
         # Create list with both full names and short names
         pdf_list = []
+        short_names = []
         for pdf in available_pdfs:
+            short_name = get_short_document_name(pdf)
             pdf_list.append({
                 "filename": pdf,
-                "short_name": get_short_document_name(pdf),
+                "short_name": short_name,
                 "type": "PDF" if pdf.endswith('.pdf') else "DOCX"
             })
+            short_names.append(short_name)
         
         return jsonify({
             "available_pdfs": available_pdfs,
             "pdf_list": pdf_list,
+            "short_names": short_names,  # Simple array for easy frontend use
             "count": len(available_pdfs),
             "pdf_directory": "Budget_Proposals copy-2/assets/pdfs"
         })
@@ -895,6 +899,25 @@ def get_document_names():
         logger.error(f"Error getting document names: {e}")
         return jsonify({"error": str(e)}), 500
 
+@app.route('/api/chat/short-document-names', methods=['GET'])
+def get_short_document_names():
+    """Get just the short document names as a simple array for frontend display"""
+    try:
+        available_pdfs = get_available_pdfs()
+        
+        # Create simple array of short names
+        short_names = []
+        for pdf in available_pdfs:
+            short_names.append(get_short_document_name(pdf))
+        
+        return jsonify({
+            "short_names": short_names,
+            "count": len(short_names)
+        })
+    except Exception as e:
+        logger.error(f"Error getting short document names: {e}")
+        return jsonify({"error": str(e)}), 500
+
 @app.route('/api/chat/detect-language', methods=['POST'])
 def detect_language():
     """Test language detection functionality"""
@@ -943,6 +966,7 @@ def home():
             "GET /api/chat/suggestions": "Get suggested questions (multilingual)",
             "GET /api/chat/available-pdfs": "Get available PDF files with short names",
             "GET /api/chat/document-names": "Get document name mapping (full to short names)",
+            "GET /api/chat/short-document-names": "Get simple array of short document names",
             "POST /api/chat/detect-language": "Test language detection"
         },
         "status": "running"