mainfile_with copy

app.py

@@ -1,5 +1,6 @@
 import os
 import sys
+import html
 
 # ============================================================================
 # CRITICAL: Set API key BEFORE importing Together
@@ -16,16 +17,25 @@ os.environ["TOGETHER_API_KEY"] = API_KEY
 
 # NOW import Together and other dependencies
 import streamlit as st
+import streamlit.components.v1
 from together import Together
 from langchain_community.vectorstores import Chroma
 from langchain_huggingface import HuggingFaceEmbeddings
+from huggingface_hub import snapshot_download
 
 # ============================================================================
 # CONFIGURATION
 # ============================================================================
 
+# Your HuggingFace dataset repository containing all vector stores
+DATASET_REPO = "Sbnos/vstoryies"
+
 # Vector store configurations
 VECTOR_STORES = {
+    "General Medicine": {
+        "collection_name": "oxfordmed",
+        "persist_directory": "oxfordmedbookdir"
+    },
     "Paediatrics": {
         "collection_name": "paedia",
         "persist_directory": "nelsonpaedia"
@@ -75,7 +85,7 @@ VECTOR_STORES = {
 # Model configurations
 EMBED_MODEL = "BAAI/bge-base-en"
 LLM_MODEL = "Qwen/Qwen3-Next-80B-A3B-Instruct"
-RETRIEVAL_K = 20
+RETRIEVAL_K = 26
 
 # ============================================================================
 # PAGE CONFIG
@@ -98,6 +108,27 @@ except Exception as e:
     st.error(f"❌ Failed to initialize Together client: {e}")
     st.stop()
 
+# Download all vector stores from HuggingFace dataset on first run
+@st.cache_resource
+def download_all_vectorstores():
+    """Download all vector stores from HuggingFace dataset repository"""
+    if not any(os.path.exists(config["persist_directory"]) for config in VECTOR_STORES.values()):
+        with st.spinner("📥 Downloading vector stores from HuggingFace (one-time setup)..."):
+            try:
+                snapshot_download(
+                    repo_id=DATASET_REPO,
+                    repo_type="dataset",
+                    local_dir=".",
+                    allow_patterns=["*"]
+                )
+                st.success("✅ Vector stores downloaded successfully!")
+            except Exception as e:
+                st.error(f"❌ Failed to download vector stores: {e}")
+                st.stop()
+
+# Download vector stores if needed
+download_all_vectorstores()
+
 # Initialize embeddings
 @st.cache_resource
 def get_embeddings():
@@ -230,11 +261,46 @@ for i, message in enumerate(st.session_state.chat_history):
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
         
-        # Add copy button for assistant messages
+        # Add one-click copy button for assistant messages
         if message["role"] == "assistant":
-            # Create expandable section with copyable text
-            with st.expander("📋 Click to copy response"):
-                st.code(message["content"], language=None)
+            escaped_content = html.escape(message["content"])
+            
+            copy_html = f"""
+                <div style="margin-top: 10px;">
+                    <button onclick="copyText_{i}()" style="
+                        background: transparent;
+                        border: 1px solid rgba(250, 250, 250, 0.2);
+                        border-radius: 4px;
+                        cursor: pointer;
+                        font-size: 1.2rem;
+                        padding: 0.25rem 0.5rem;
+                        opacity: 0.7;
+                        transition: opacity 0.2s;
+                    " onmouseover="this.style.opacity='1'" onmouseout="this.style.opacity='0.7'" title="Copy to clipboard">
+                        📋
+                    </button>
+                    <span id="status_{i}" style="margin-left: 10px; color: green; display: none;">✓ Copied!</span>
+                    <textarea id="text_{i}" style="position: absolute; left: -9999px;">{escaped_content}</textarea>
+                </div>
+                
+                <script>
+                function copyText_{i}() {{
+                    const textarea = document.getElementById('text_{i}');
+                    textarea.select();
+                    document.execCommand('copy');
+                    
+                    // Try modern API as fallback
+                    if (navigator.clipboard) {{
+                        navigator.clipboard.writeText(textarea.value);
+                    }}
+                    
+                    const status = document.getElementById('status_{i}');
+                    status.style.display = 'inline';
+                    setTimeout(() => status.style.display = 'none', 2000);
+                }}
+                </script>
+            """
+            st.components.v1.html(copy_html, height=50)
 
 # Chat input
 user_input = st.chat_input("Ask me anything about medical topics...")
@@ -280,9 +346,45 @@ if user_input:
         
         response_placeholder.markdown(full_response)
         
-        # Add copy section with expandable code block
-        with st.expander("📋 Click to copy response"):
-            st.code(full_response, language=None)
+        # Add one-click copy button for the new response
+        escaped_content_new = html.escape(full_response)
+        
+        copy_html_new = f"""
+            <div style="margin-top: 10px;">
+                <button onclick="copyText_new()" style="
+                    background: transparent;
+                    border: 1px solid rgba(250, 250, 250, 0.2);
+                    border-radius: 4px;
+                    cursor: pointer;
+                    font-size: 1.2rem;
+                    padding: 0.25rem 0.5rem;
+                    opacity: 0.7;
+                    transition: opacity 0.2s;
+                " onmouseover="this.style.opacity='1'" onmouseout="this.style.opacity='0.7'" title="Copy to clipboard">
+                    📋
+                </button>
+                <span id="status_new" style="margin-left: 10px; color: green; display: none;">✓ Copied!</span>
+                <textarea id="text_new" style="position: absolute; left: -9999px;">{escaped_content_new}</textarea>
+            </div>
+            
+            <script>
+            function copyText_new() {{
+                const textarea = document.getElementById('text_new');
+                textarea.select();
+                document.execCommand('copy');
+                
+                // Try modern API as fallback
+                if (navigator.clipboard) {{
+                    navigator.clipboard.writeText(textarea.value);
+                }}
+                
+                const status = document.getElementById('status_new');
+                status.style.display = 'inline';
+                setTimeout(() => status.style.display = 'none', 2000);
+            }}
+            </script>
+        """
+        st.components.v1.html(copy_html_new, height=50)
     
     # Save assistant response
     st.session_state.chat_history.append({