UI imrpovements

README.md

@@ -75,7 +75,7 @@ PDF Upload → LlamaIndex Parser → Vector Index → OpenAI Analysis → Cited
 
 ```bash
 # Clone the repository
-git clone <your-repo-url>
+git clone https://huggingface.co/spaces/sukhrobnurali/financial-document-analyzer
 cd investment-screener
 
 # Install dependencies

app.py

@@ -54,12 +54,51 @@ st.markdown("""
         display: inline-block;
         margin: 0.5rem 0;
     }
-    .citation {
-        background-color: #f8f9fa;
-        border-left: 4px solid #007bff;
-        padding: 1rem;
-        margin: 0.5rem 0;
-        border-radius: 0.25rem;
+    div[data-testid="stExpander"] {
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
+        border-radius: 12px;
+        border: none;
+        box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+        margin: 0.75rem 0;
+    }
+    div[data-testid="stExpander"]:hover {
+        box-shadow: 0 6px 12px rgba(0, 0, 0, 0.15);
+    }
+    div[data-testid="stExpander"] summary {
+        background: transparent !important;
+        color: white !important;
+        font-weight: 500;
+        padding: 1rem 1.25rem;
+    }
+    div[data-testid="stExpander"] summary:hover {
+        background: transparent !important;
+    }
+    div[data-testid="stExpander"][open] summary {
+        background: transparent !important;
+    }
+    div[data-testid="stExpander"] div[role="button"] {
+        background: transparent !important;
+    }
+    div[data-testid="stExpander"] div[role="button"] p {
+        color: white !important;
+        font-size: 0.95rem;
+    }
+    div[data-testid="stExpander"] > div > div {
+        padding: 0 1.25rem 1.25rem 1.25rem;
+        color: white;
+    }
+    .citation-preview {
+        color: white;
+        font-size: 0.9rem;
+        line-height: 1.6;
+        opacity: 0.95;
+        margin-top: 0.5rem;
+    }
+    .citation-full {
+        color: white;
+        font-size: 0.9rem;
+        line-height: 1.6;
+        opacity: 0.95;
     }
     .metric-card {
         background-color: #ffffff;
@@ -67,6 +106,7 @@ st.markdown("""
         border-radius: 0.5rem;
         border: 1px solid #e0e0e0;
         margin: 1rem 0;
+        color: #212529;
     }
 </style>
 """, unsafe_allow_html=True)
@@ -110,9 +150,6 @@ def display_analysis_result(result, criteria_name):
         st.markdown("### Summary")
         st.info(result['summary'])
 
-    # Detailed metrics
-    st.markdown("### Detailed Analysis")
-
     # Remove metadata fields for display
     metrics = {k: v for k, v in result.items()
                if k not in ['overall_pass', 'summary', 'citations', 'source_nodes_count', 'parse_error', 'raw_response']}
@@ -129,11 +166,6 @@ def display_analysis_result(result, criteria_name):
         for citation in result['citations'][:5]:  # Show top 5 citations
             display_citation(citation)
 
-    # Debug: Show raw response if parse error
-    if result.get('parse_error'):
-        with st.expander("Raw LLM Response (Debug)"):
-            st.text(result.get('raw_response', 'No response'))
-
 
 def display_metric_card(metric_name, metric_data):
     """Display a single metric card with citation."""
@@ -174,13 +206,18 @@ def display_metric_card(metric_name, metric_data):
 
 
 def display_citation(citation):
-    """Display a citation box."""
-    st.markdown(f"""
-    <div class="citation">
-        <strong>Page {citation['page']}</strong> (Relevance: {citation['score']:.2%})<br>
-        <small>{citation['text_preview']}</small>
-    </div>
-    """, unsafe_allow_html=True)
+    """Display a modern expandable citation card using Streamlit."""
+    # Create expander with page, score, and preview in the header
+    header = f"📄 Page {citation['page']} • ⭐ {citation['score']:.0%} Match\n\n{citation['text_preview']}"
+
+    with st.expander(header, expanded=False):
+        # Show only full text when expanded (no duplicate preview)
+        if citation.get('is_truncated', False):
+            st.markdown(f"<div class='citation-full'>{citation['full_text']}</div>",
+                       unsafe_allow_html=True)
+        else:
+            # If not truncated, preview and full text are the same, so show nothing extra
+            st.caption("(Full text shown above)")
 
 
 def main():

document_processor.py

@@ -108,6 +108,8 @@ class InvestmentDocumentProcessor:
                 "page": page,
                 "score": node.score,
                 "text_preview": node.node.text[:200] + "..." if len(node.node.text) > 200 else node.node.text,
+                "full_text": node.node.text,
+                "is_truncated": len(node.node.text) > 200,
                 "file_name": node.node.metadata.get('file_name', 'Unknown')
             }
             citations.append(citation)