Create loading_animations.py

loading_animations.py

@@ -0,0 +1,146 @@
+"""
+Loading animations for Gradio chatbot interface.
+Contains functions to generate animated thinking indicators.
+"""
+
+import random
+
+def create_thinking_indicator(message="Thinking"):
+    """
+    Creates an HTML thinking indicator with animated dots.
+    
+    Args:
+        message (str): The text to show before the dots (default: "Thinking")
+    
+    Returns:
+        str: HTML string with animated thinking indicator
+    """
+    return f'''<div class="thinking-indicator">
+        {message}<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():
+    """
+    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.
+    
+    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
+    """
+    dots = ''.join(['<span class="dot"></span>' for _ in range(dot_count)])
+    
+    return f'''<div class="thinking-indicator">
+        {message}<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):
+    """
+    Creates thinking indicators for multi-stage processing.
+    
+    Args:
+        stage_index (int): Current stage (0-based index)
+    
+    Returns:
+        str: HTML string for current stage thinking indicator
+    """
+    stages = create_processing_stages()
+    
+    if 0 <= stage_index < len(stages):
+        return create_thinking_indicator(stages[stage_index])
+    else:
+        return create_thinking_indicator("Processing")
+
+# Quick usage examples:
+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))