Update app.py

app.py

@@ -731,6 +731,13 @@ def get_topics_list() -> List[str]:
         print(f"❌ Error retrieving topics: {str(e)}")
         return ["All Topics", "journal"]
 
+def count_tokens_rough(text: str) -> int:
+    try:
+        return len(text) // 4
+    except Exception as e:
+        print(f"❌ Error counting tokens: {e}")
+        return 0
+
 
 # TWEAK JAWABAN MODEL
 def groq_with_memory(message: str, topic: str = "general", retries: int = 3) -> tuple:
@@ -808,6 +815,85 @@ def groq_with_memory(message: str, topic: str = "general", retries: int = 3) ->
     except Exception as e:
         return f"❌ Error: {str(e)}", ""
 
+def cleanup_old_messages(days: int = 30) -> str:
+    try:
+        cutoff = (datetime.now() - timedelta(days=days)).isoformat()
+        with sqlite3.connect(storage.db.db_path) as conn:
+            conn.execute('DELETE FROM messages WHERE timestamp < ?', (cutoff,))
+            conn.commit()
+        storage.chat_history = [msg for msg in storage.chat_history if msg['timestamp'] >= cutoff]
+        storage.save_history()
+        return f"✅ Cleared messages older than {days} days"
+    except Exception as e:
+        return f"❌ Failed to clean old messages: {str(e)}"
+
+
+def send_message(user_input: str, topic_input: str) -> Tuple[str, str, List[str]]:
+    """Handle sending a user message, getting AI response, and updating topic list."""
+    try:
+        if not user_input.strip():
+            return "❌ Please enter a message", user_input, get_topics_list()
+        if not topic_input.strip():
+            return "❌ Please enter a topic", user_input, get_topics_list()
+        
+        response, error = groq_with_memory(user_input, topic_input)
+        if error:
+            return f"❌ {error}", user_input, get_topics_list()
+        
+        return response, "", get_topics_list()
+    except Exception as e:
+        return f"❌ Error processing message: {str(e)}", user_input, get_topics_list()
+
+def show_current_context() -> str:
+    """Show the current session's conversation context."""
+    try:
+        messages = storage.get_messages_by_session(storage.current_session_id)
+        if not messages:
+            return "No messages in the current session."
+        context = "\n".join([f"{msg['role'].capitalize()}: {msg['content']}" for msg in messages[-5:]])
+        return f"Current Session Context (Last 5 Messages):\n{context}"
+    except Exception as e:
+        return f"❌ Error retrieving context: {str(e)}"
+
+def get_full_history(topic_filter: str = None) -> str:
+    """Display the full chat history, optionally filtered by topic."""
+    try:
+        filters = {'topic': topic_filter} if topic_filter and topic_filter != "All Topics" else None
+        messages = storage.search_messages("", filters)
+        if not messages:
+            return "No messages found in the history."
+        history = "\n".join([f"[{msg['timestamp']}] {msg['role'].capitalize()} (Topic: {msg['topic']}): {msg['content']}" for msg in messages])
+        return f"Chat History:\n{history}"
+    except Exception as e:
+        return f"❌ Error retrieving history: {str(e)}"
+
+def get_chat_summary(topic_filter: str = None) -> str:
+    """Provide a summary of the chat history, optionally filtered by topic."""
+    try:
+        filters = {'topic': topic_filter} if topic_filter and topic_filter != "All Topics" else None
+        messages = storage.search_messages("", filters)
+        if not messages:
+            return "No messages found for summary."
+        
+        total_messages = len(messages)
+        user_messages = len([msg for msg in messages if msg['role'] == 'user'])
+        assistant_messages = len([msg for msg in messages if msg['role'] == 'assistant'])
+        topics = Counter(msg['topic'] for msg in messages)
+        sessions = len(set(msg['session_id'] for msg in messages))
+        
+        summary = (
+            f"Chat Summary:\n"
+            f"Total Messages: {total_messages}\n"
+            f"User Messages: {user_messages}\n"
+            f"Assistant Messages: {assistant_messages}\n"
+            f"Number of Sessions: {sessions}\n"
+            f"Topic Distribution:\n" +
+            "\n".join([f"  - {topic}: {count}" for topic, count in topics.most_common()])
+        )
+        return summary
+    except Exception as e:
+        return f"❌ Error generating summary: {str(e)}"
+        
 
 def clear_all_history() -> Tuple[str, str, str]:
     """Clear all chat history and analytics data."""