Upload folder using huggingface_hub

hate_speech_demo.py

@@ -36,6 +36,7 @@ function openPolicyPopup() {{
 </script>
 """
 
+# Replace your process_retrieval_text function with this updated version
 def process_retrieval_text(retrieval_text, user_input):
     """
     Process the retrieval text by identifying proper document boundaries
@@ -69,11 +70,15 @@ def process_retrieval_text(retrieval_text, user_input):
                 # Format with clear section headers and better spacing
                 formatted_html = f"""
                 <div class='doc-section'>
-                    <h3 class="doc-number">Evidence Document {i+1}</h3>
-                    <div class="doc-info">
-                        <p><strong>Document Title:</strong> {doc_name}</p>
-                        <p><strong>Section:</strong> Page {page}</p>
+                    <h3 class="doc-number">Evidence Section {i+1}</h3>
+                    
+                    <div class="doc-section-info">
+                        <p><strong>Section Title:</strong> {doc_name}</p>
+                        <div class="subsection-info">
+                            <p><strong>Subsection:</strong> Page {page}</p>
+                        </div>
                     </div>
+                    
                     <div class="doc-content-container">
                         <h4>Content:</h4>
                         <div class='doc-content'>{content}</div>
@@ -110,7 +115,7 @@ def process_retrieval_text(retrieval_text, user_input):
         # Format each chunk with better section styling
         chunks = [f"""
         <div class='doc-section'>
-            <h3 class="doc-title">Evidence Document {i+1}</h3>
+            <h3 class="doc-title">Evidence Section {i+1}</h3>
             <div class='doc-content'>{chunk.strip()}</div>
         </div>
         """ for i, chunk in enumerate(chunks)]
@@ -158,15 +163,28 @@ def process_retrieval_text(retrieval_text, user_input):
         font-size: 18px;
     }
     
-    .doc-info {
+    .doc-section-info {
         margin: 10px 0;
         padding: 8px;
         background: #f5f5f5;
         border-radius: 4px;
     }
     
-    .doc-info p {
+    .doc-section-info p {
         margin: 5px 0;
+        font-size: 16px;
+    }
+    
+    .subsection-info {
+        margin-left: 15px;
+        padding-left: 10px;
+        border-left: 2px solid #ddd;
+        margin-top: 5px;
+    }
+    
+    .subsection-info p {
+        font-size: 14px;
+        color: #555;
     }
     
     .doc-content-container {