Upload folder using huggingface_hub

hate_speech_demo.py

@@ -533,75 +533,118 @@ def rate_user_input(user_input):
     - OpenAI Moderation
     - Google Perspective API
     """
-    # Initialize APIs with hardcoded keys
-    contextual_api = ContextualAPIUtils(api_key=ORACLE_API_KEY)
-    together_client = Together(api_key=TOGETHER_API_KEY)
-    openai_client = openai.OpenAI(api_key=OPENAI_API_KEY)
-    
-    # Get ratings from all four services
-    llama_rating, llama_safety = get_llama_guard_rating(together_client, user_input, user_input)
-    contextual_rating, contextual_retrieval, contextual_safety = get_contextual_rating(contextual_api, user_input, user_input)
-    openai_rating, openai_safety = get_openai_moderation(openai_client, user_input, user_input)
-    perspective_rating, perspective_safety = get_perspective_rating(user_input)
-    
-    # Format responses carefully to avoid random line breaks
-    llama_rating = re.sub(r'\.(?=\s+[A-Z])', '.\n', llama_rating)
-    contextual_rating = re.sub(r'\.(?=\s+[A-Z])', '.\n', contextual_rating)
-    
-    # Process retrieval text to highlight keywords with better formatting
-    processed_retrieval = process_retrieval_text(contextual_retrieval, user_input)
-    
-    # Format results with HTML styling
-    llama_html = f"""<div class="rating-box secondary-box {llama_safety}-rating">{llama_rating}</div>"""
-    openai_html = f"""<div class="rating-box secondary-box {openai_safety}-rating">{openai_rating}</div>"""
-    perspective_html = f"""<div class="rating-box secondary-box {perspective_safety}-rating">{perspective_rating}</div>"""
-    
-    # Create the knowledge section (initially hidden) and button
-    knowledge_html = ""
-    knowledge_button = ""
-    
-    if processed_retrieval and processed_retrieval != "No retrieval text found.":
-        # Create unique ID for this instance
-        import uuid
-        popup_id = f"knowledge-popup-{uuid.uuid4().hex[:8]}"
-        
-        # Create the popup div (initially hidden)
-        knowledge_html = f"""
-        <div id="{popup_id}" class="knowledge-popup" style="display: none;">
-            <div class="knowledge-popup-header">Retrieved Knowledge</div>
-            <button class="knowledge-popup-close" 
-                    onclick="this.parentElement.style.display='none'; 
-                             document.getElementById('btn-{popup_id}').style.display='inline-block'; 
-                             return false;">
-                Close
-            </button>
-            <div class="knowledge-popup-content">
-                {processed_retrieval}
+    try:
+        # Initialize APIs with hardcoded keys
+        contextual_api = ContextualAPIUtils(api_key=ORACLE_API_KEY)
+        together_client = Together(api_key=TOGETHER_API_KEY)
+        openai_client = openai.OpenAI(api_key=OPENAI_API_KEY)
+        
+        # Skip empty input
+        if not user_input or user_input.strip() == "":
+            return (
+                '<div class="rating-box contextual-box empty-rating">Please enter content to rate</div>',
+                '<div class="rating-box secondary-box empty-rating">Please enter content to rate</div>',
+                '<div class="rating-box secondary-box empty-rating">Please enter content to rate</div>',
+                '<div class="rating-box secondary-box empty-rating">Please enter content to rate</div>',
+                ""
+            )
+        
+        # Get ratings
+        try:
+            llama_rating, llama_safety = get_llama_guard_rating(together_client, user_input, user_input)
+        except Exception as e:
+            llama_rating = f"Error getting LlamaGuard rating: {str(e)}"
+            llama_safety = "unsafe"
+            
+        try:
+            contextual_rating, contextual_retrieval, contextual_safety = get_contextual_rating(contextual_api, user_input, user_input)
+        except Exception as e:
+            contextual_rating = f"Error getting Contextual rating: {str(e)}"
+            contextual_retrieval = ""
+            contextual_safety = "unsafe"
+            
+        try:
+            openai_rating, openai_safety = get_openai_moderation(openai_client, user_input, user_input)
+        except Exception as e:
+            openai_rating = f"Error getting OpenAI rating: {str(e)}"
+            openai_safety = "unsafe"
+            
+        try:
+            perspective_rating, perspective_safety = get_perspective_rating(user_input)
+        except Exception as e:
+            perspective_rating = f"Error getting Perspective API rating: {str(e)}"
+            perspective_safety = "unsafe"
+        
+        # Format responses carefully to avoid random line breaks
+        llama_rating = re.sub(r'\.(?=\s+[A-Z])', '.\n', llama_rating)
+        contextual_rating = re.sub(r'\.(?=\s+[A-Z])', '.\n', contextual_rating)
+        
+        # Process retrieval text to highlight keywords with better formatting
+        if contextual_retrieval:
+            try:
+                processed_retrieval = process_retrieval_text(contextual_retrieval, user_input)
+            except Exception as e:
+                processed_retrieval = f"Error processing retrieval text: {str(e)}"
+        else:
+            processed_retrieval = "No retrieval text found."
+        
+        # Format results with HTML styling
+        llama_html = f"""<div class="rating-box secondary-box {llama_safety}-rating">{llama_rating}</div>"""
+        openai_html = f"""<div class="rating-box secondary-box {openai_safety}-rating">{openai_rating}</div>"""
+        perspective_html = f"""<div class="rating-box secondary-box {perspective_safety}-rating">{perspective_rating}</div>"""
+        
+        # Create the knowledge section (initially hidden) and button
+        knowledge_html = ""
+        knowledge_button = ""
+        
+        if processed_retrieval and processed_retrieval != "No retrieval text found.":
+            # Create unique ID for this instance
+            import uuid
+            popup_id = f"knowledge-popup-{uuid.uuid4().hex[:8]}"
+            
+            # Create the popup div (initially hidden)
+            knowledge_html = f"""
+            <div id="{popup_id}" class="knowledge-popup" style="display: none;">
+                <div class="knowledge-popup-header">Retrieved Knowledge</div>
+                <button class="knowledge-popup-close" 
+                        onclick="this.parentElement.style.display='none'; 
+                                 document.getElementById('btn-{popup_id}').style.display='inline-block'; 
+                                 return false;">
+                    Close
+                </button>
+                <div class="knowledge-popup-content">
+                    {processed_retrieval}
+                </div>
             </div>
-        </div>
-        """
+            """
+            
+            # Create a toggle button (positioned BELOW the contextual results)
+            knowledge_button = f"""
+            <div style="margin-top: 10px; margin-bottom: 5px;">
+                <a href="#" id="btn-{popup_id}" class="knowledge-button" 
+                   onclick="document.getElementById('{popup_id}').style.display='block'; this.style.display='none'; return false;">
+                   Show Retrieved Knowledge
+                </a>
+            </div>
+            """
         
-        # Create a toggle button (positioned BELOW the contextual results)
-        knowledge_button = f"""
-        <div style="margin-top: 10px; margin-bottom: 5px;">
-            <a href="#" id="btn-{popup_id}" class="knowledge-button" 
-               onclick="document.getElementById('{popup_id}').style.display='block'; this.style.display='none'; return false;">
-               Show Retrieved Knowledge
-            </a>
+        # Format contextual results with HTML styling
+        contextual_html = f"""
+        <div class="rating-box contextual-box {contextual_safety}-rating">
+            <button class="copy-button" onclick="navigator.clipboard.writeText(this.parentElement.innerText.replace('Copy', ''))">Copy</button>
+            {contextual_rating}
         </div>
+        {knowledge_button}
+        {knowledge_html}
         """
-    
-    # Format contextual results with HTML styling
-    contextual_html = f"""
-    <div class="rating-box contextual-box {contextual_safety}-rating">
-        <button class="copy-button" onclick="navigator.clipboard.writeText(this.parentElement.innerText.replace('Copy', ''))">Copy</button>
-        {contextual_rating}
-    </div>
-    {knowledge_button}
-    {knowledge_html}
-    """
-    
-    return contextual_html, llama_html, openai_html, perspective_html, ""
+        
+        return contextual_html, llama_html, openai_html, perspective_html, ""
+        
+    except Exception as e:
+        # Global error handler to prevent the entire function from failing
+        error_message = f"An error occurred: {str(e)}"
+        error_html = f'<div class="rating-box empty-rating">{error_message}</div>'
+        return error_html, error_html, error_html, error_html, ""
 
 def random_test_case():
     try:
@@ -718,10 +761,12 @@ def create_gradio_app():
         color: #000000 !important;
     }
     
-    /* Perspective API styling */
-    .perspective-icon {
-        vertical-align: middle;
-        margin-right: 5px;
+    /* Adjusted secondary boxes */
+    .secondary-box {
+        min-height: 120px;
+        font-size: 0.9em;
+        overflow-y: auto;
+        max-height: 300px;
     }
     """
     
@@ -852,6 +897,8 @@ def create_gradio_app():
                 # Hidden placeholder for retrieved knowledge
                 retrieved_knowledge = gr.HTML('', visible=False)
                 
+                # All three other systems in one row
+                gr.Markdown("### Other Rating Systems")
                 with gr.Row():
                     with gr.Column():
                         # LlamaGuard section with permanent model card link
@@ -877,14 +924,11 @@ def create_gradio_app():
                         </div>
                         """)
                         openai_results = gr.HTML('<div class="rating-box secondary-box empty-rating">Rating will appear here</div>')
-                
-                # Add Perspective API section
-                with gr.Row():
                     with gr.Column():
-                        # Perspective API section with model card link
+                        # Perspective API section with info link
                         gr.HTML("""
                         <div>
-                            <h3 class="result-header">🔍 Google Perspective API</h3>
+                            <h3 class="result-header">🔍 Perspective API</h3>
                             <div style="margin-top: -10px; margin-bottom: 10px;">
                                 <a href="https://developers.perspectiveapi.com/s/about-the-api" 
                                    target="_blank" class="knowledge-button">View API info</a>