Upload app.py with huggingface_hub

app.py

@@ -17,6 +17,8 @@ FEEDBACK_FILE = Path(__file__).parent / "feedback_data.json"
 HF_DATASET_REPO = "Vrda/im-error-check-data"
 HF_DATASET_FILE = "feedback_data.json"
 
+_DEEPSEEK_RESULTS: dict[str, dict] = {}
+
 # -------------------------------------------------------------------------
 # Feedback persistence (local + HF Hub sync)
 # -------------------------------------------------------------------------
@@ -101,27 +103,32 @@ st.set_page_config(
 st.markdown("""
 <style>
     .error-card {
-        background: #fff5f5; border-left: 4px solid #e53e3e;
+        background: #2d1f1f; border-left: 4px solid #e53e3e;
         border-radius: 8px; padding: 0.8rem 1rem; margin: 0.5rem 0;
+        color: #fde8e8;
     }
+    .error-card strong { color: #feb2b2; }
+    .error-card em { color: #fbd38d; }
     .suggestion-card {
-        background: #f0fff4; border-left: 4px solid #38a169;
+        background: #1a2e1a; border-left: 4px solid #38a169;
         border-radius: 8px; padding: 0.8rem 1rem; margin: 0.5rem 0;
+        color: #c6f6d5;
     }
+    .suggestion-card strong { color: #9ae6b4; }
     .model-header-a {
-        background: #ebf8ff; border-left: 4px solid #3182ce;
+        background: #1a2332; border-left: 4px solid #63b3ed;
         border-radius: 8px; padding: 0.6rem 1rem; margin-bottom: 0.5rem;
     }
     .model-header-b {
-        background: #faf5ff; border-left: 4px solid #805ad5;
+        background: #2d1f3d; border-left: 4px solid #b794f4;
         border-radius: 8px; padding: 0.6rem 1rem; margin-bottom: 0.5rem;
     }
-    .severity-critical { color: #c53030; font-weight: bold; }
-    .severity-high { color: #dd6b20; font-weight: bold; }
-    .severity-medium { color: #d69e2e; }
-    .severity-low { color: #38a169; }
+    .severity-critical { color: #fc8181; font-weight: bold; }
+    .severity-high { color: #f6ad55; font-weight: bold; }
+    .severity-medium { color: #f6e05e; }
+    .severity-low { color: #68d391; }
     .category-badge {
-        display: inline-block; background: #edf2f7; color: #4a5568;
+        display: inline-block; background: #2d3748; color: #e2e8f0;
         padding: 2px 8px; border-radius: 12px; font-size: 0.8em; margin-right: 4px;
     }
 </style>
@@ -245,10 +252,14 @@ st.button("Analyze", type="primary", on_click=trigger_analysis)
 # Run analysis (progressive: show GPT-OSS first, DeepSeek when ready)
 # -------------------------------------------------------------------------
 
-def _run_deepseek_background(english_text: str):
-    """Background thread: calls DeepSeek and stores result in session state."""
+def _run_deepseek_background(session_key: str, english_text: str):
+    """Background thread: calls DeepSeek and stores result in module-level dict."""
     result = call_model_a(english_text)
-    st.session_state._deepseek_container["result"] = result
+    _DEEPSEEK_RESULTS[session_key] = result
+
+if "session_key" not in st.session_state:
+    import uuid
+    st.session_state.session_key = str(uuid.uuid4())
 
 if st.session_state.run_analysis and st.session_state.input_text.strip():
     st.session_state.run_analysis = False
@@ -257,6 +268,9 @@ if st.session_state.run_analysis and st.session_state.input_text.strip():
     st.session_state.total_elapsed = 0
     st.session_state._analysis_start = time.time()
 
+    skey = st.session_state.session_key
+    _DEEPSEEK_RESULTS.pop(skey, None)
+
     with st.spinner("Translating discharge letter..."):
         t0 = time.time()
         st.session_state.translated_text = translate_to_english(st.session_state.input_text)
@@ -264,10 +278,9 @@ if st.session_state.run_analysis and st.session_state.input_text.strip():
 
     english = st.session_state.translated_text
 
-    container = {"result": None}
-    st.session_state._deepseek_container = container
-    thread = threading.Thread(target=_run_deepseek_background, args=(english,), daemon=True)
-    st.session_state._deepseek_thread = thread
+    thread = threading.Thread(
+        target=_run_deepseek_background, args=(skey, english), daemon=True
+    )
     thread.start()
 
     with st.spinner("GPT-OSS-120B responding (~5s)..."):
@@ -394,20 +407,18 @@ if has_any_result:
         else:
             @st.fragment(run_every=5)
             def _poll_deepseek():
-                if (
-                    hasattr(st.session_state, "_deepseek_container")
-                    and st.session_state._deepseek_container["result"] is not None
-                    and st.session_state.model_a_result is None
-                ):
-                    st.session_state.model_a_result = st.session_state._deepseek_container["result"]
+                skey = st.session_state.session_key
+                if skey in _DEEPSEEK_RESULTS:
+                    st.session_state.model_a_result = _DEEPSEEK_RESULTS.pop(skey)
                     st.session_state.total_elapsed = round(
                         time.time() - st.session_state._analysis_start, 2
                     )
                     st.rerun()
+                elapsed = round(time.time() - st.session_state._analysis_start)
                 st.markdown(
-                    '<div style="background:#f7fafc; border:2px dashed #cbd5e0; '
-                    'border-radius:8px; padding:2rem; text-align:center; color:#718096;">'
-                    "<strong>DeepSeek Reasoner</strong> is still processing...<br>"
+                    '<div style="background:#1e293b; border:2px dashed #475569; '
+                    'border-radius:8px; padding:2rem; text-align:center; color:#e2e8f0;">'
+                    f"<strong>DeepSeek Reasoner</strong> is still processing... ({elapsed}s)<br>"
                     "This typically takes 60-90 seconds.<br>"
                     "Review and rate GPT-OSS results below while you wait."
                     "</div>",