Update loading_animations.py

loading_animations.py

@@ -1,146 +1,52 @@
 """
 Loading animations for Gradio chatbot interface.
-Contains functions to generate animated thinking indicators.
+Contains functions to generate animated thinking indicators with just pulsing dots.
 """
 
-import random
-
-def create_thinking_indicator(message="Thinking"):
+def create_thinking_indicator():
     """
-    Creates an HTML thinking indicator with animated dots.
-    
-    Args:
-        message (str): The text to show before the dots (default: "Thinking")
+    Creates an HTML thinking indicator with just animated dots.
     
     Returns:
-        str: HTML string with animated thinking indicator
+        str: HTML string with animated dots only
     """
-    return f'''<div class="thinking-indicator">
-        {message}<div class="dots-container">
+    return '''<div class="thinking-indicator">
+        <div class="dots-container">
             <span class="dot"></span>
             <span class="dot"></span>
             <span class="dot"></span>
         </div>
     </div>'''
 
-def get_random_thinking_message():
-    """
-    Returns a random thinking message for variety.
-    
-    Returns:
-        str: Random thinking message
-    """
-    messages = [
-        "Thinking",
-        "Processing", 
-        "Analyzing",
-        "Working on it",
-        "Let me think",
-        "Calculating",
-        "Considering",
-        "Evaluating"
-    ]
-    return random.choice(messages)
-
-def create_random_thinking_indicator():
+def create_custom_dot_indicator(dot_count=3):
     """
-    Creates a thinking indicator with a random message.
-    
-    Returns:
-        str: HTML string with animated thinking indicator
-    """
-    return create_thinking_indicator(get_random_thinking_message())
-
-def get_thinking_indicator_by_query(query):
-    """
-    Creates a context-appropriate thinking indicator based on the user query.
+    Creates a thinking indicator with specified number of dots.
     
     Args:
-        query (str): The user's query text
-    
-    Returns:
-        str: HTML string with context-appropriate thinking indicator
-    """
-    query_lower = query.lower()
-    
-    if any(word in query_lower for word in ["math", "calculate", "equation", "formula"]):
-        return create_thinking_indicator("Calculating")
-    elif any(word in query_lower for word in ["graph", "chart", "plot", "visualization"]):
-        return create_thinking_indicator("Creating visualization")
-    elif any(word in query_lower for word in ["explain", "define", "what is"]):
-        return create_thinking_indicator("Analyzing")
-    elif any(word in query_lower for word in ["solve", "problem", "solution"]):
-        return create_thinking_indicator("Solving")
-    elif any(word in query_lower for word in ["code", "program", "algorithm"]):
-        return create_thinking_indicator("Processing")
-    else:
-        return create_random_thinking_indicator()
-
-def create_custom_thinking_indicator(message, dot_count=3):
-    """
-    Creates a custom thinking indicator with specified message and dot count.
-    
-    Args:
-        message (str): Custom message to display
         dot_count (int): Number of animated dots (default: 3)
     
     Returns:
-        str: HTML string with custom thinking indicator
+        str: HTML string with custom number of dots
     """
     dots = ''.join(['<span class="dot"></span>' for _ in range(dot_count)])
     
     return f'''<div class="thinking-indicator">
-        {message}<div class="dots-container">
+        <div class="dots-container">
             {dots}
         </div>
     </div>'''
 
-def create_processing_stages():
-    """
-    Returns a list of processing stage messages for longer operations.
-    
-    Returns:
-        list: List of progressive thinking messages
-    """
-    return [
-        "Reading your question",
-        "Analyzing the request", 
-        "Gathering information",
-        "Processing response",
-        "Finalizing answer"
-    ]
-
-def get_stage_thinking_indicator(stage_index):
+# Main function to use in the chatbot
+def get_thinking_dots():
     """
-    Creates thinking indicators for multi-stage processing.
-    
-    Args:
-        stage_index (int): Current stage (0-based index)
+    Returns the standard thinking dots indicator.
     
     Returns:
-        str: HTML string for current stage thinking indicator
+        str: HTML string with animated thinking dots
     """
-    stages = create_processing_stages()
-    
-    if 0 <= stage_index < len(stages):
-        return create_thinking_indicator(stages[stage_index])
-    else:
-        return create_thinking_indicator("Processing")
+    return create_thinking_indicator()
 
-# Quick usage examples:
+# Quick usage example:
 if __name__ == "__main__":
-    # Test the functions
-    print("Basic thinking indicator:")
-    print(create_thinking_indicator())
-    print()
-    
-    print("Random thinking indicator:")
-    print(create_random_thinking_indicator())
-    print()
-    
-    print("Query-based thinking indicator:")
-    print(get_thinking_indicator_by_query("What's the derivative of x^2?"))
-    print()
-    
-    print("Custom thinking indicator:")
-    print(create_custom_thinking_indicator("Deep thinking", 5))
+    print("Thinking dots indicator:")
+    print(get_thinking_dots())